Tensorflow学习:训练自己的简单图像识别模型(图片分类模型)

Tensorflow学习:训练自己的简单图像识别模型(图片分类模型)

1、利用inception的模型做基础,加一层输出层图片分类,只训练最后一层就好,前面的利用他们训练好的权值。

2、google提供的训练文件,自己提供训练的图片训练集。
利用官方提供的retrain.py进行训练,通过Windows的一个批处理文件retrain.bat执行官方提供的retrain.py程序。

python D:/workspace/PyCharm/venv/cuiyongling/retrain.py ^
--bottleneck_dir bottleneck ^
--how_many_training_steps 200 ^
--model_dir D:/workspace/PyCharm/venv/cuiyongling/inception_model/ ^
--output_graph output_graph.pb ^
--output_labels output_labels.txt ^
--image_dir D:/workspace/PyCharm/venv/cuiyongling/images/data/ 
pause

版权声明:本文为cui1004原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。