【AI】Python3使用TF-Hub进行迁移学习（图像分类）

机器环境

• win10
• python3.6

---

安装依赖

• pip install tensorflow==1.7.0
• pip install tensorflow-hub

---

Github地址

• https://github.com/ns2250225/tensorflow-hub-image-classify

---

准备图片素材

• 注意：每一类至少要100张图片以上，不然会报错
• 把图片分好类放到 photos 文件夹里面

---

训练模型（默认是使用Inception V3）

• 先清空output文件夹
• 这里训练20步

python .\retrain.py --image_dir .\photos\ --saved_model_dir .\model\ --bottleneck_dir .\bottleneck\ --how_many_training_steps 20 --output_labels .\output\output_labels.txt --output_graph .\output\retrain.pb

---

测试模型（测试单张图片）

python .\label_image.py --graph .\output\retrain.pb --labels .\output\output_labels.txt --input_layer=Placeholder --output_layer=final_result --image .\test.jpg

---

备注

• 可以使用别的模型来训练
• 在训练模型时加上：–tfhub_module https://tfhub.dev/google/imagenet/mobilenet_v2_100_224/feature_vector/1