标签:enter 完成后 利用 mamicode 实现 step 输出 put use
data目录:需要训练的数据放在该目录: inception_model:把下载inception模型放在该目录下:
test_images:把训练完成之后,测试图片所放目录
windows:
运行批处理文件retrain.bat
python retrain.py ^ --bottleneck_dir bottleneck ^ #自行生成每个训练图片的数据 --how_many_training_steps 200 ^ #步数 --model_dir inception_model ^ #模型 --output_graph output_graph.pb ^ #训练后输出模型 --output_labels output_labels.txt ^ #训练后输出label --image_dir data pause
Ubuntu:
运行批处理文件retrain.sh
步骤2:
训练完成后,测试如下
Python predict.py
测试结果如下:
结论:如果测试物体不相似,识别率很高,但是如果物体形状相似,识别率则不高
源码:https://pan.baidu.com/s/1qdRmnQsRv5k3QesZxRC9QA 提取码:jipt
计算机视觉(3):用inception-v3模型重新训练自己的数据模型
标签:enter 完成后 利用 mamicode 实现 step 输出 put use
原文地址:https://www.cnblogs.com/zhml/p/12552809.html