Question

我使用了这里的flowers_train脚本：flowers_train.py。

在10个新课程上重新训练现有的初始v3模型。 flowers_train脚本生成以下格式的一些检查点文件：

checkpoint                              model.ckpt-1030000.index
events.out.tfevents.1501217995.tron     model.ckpt-1030000.meta
model.ckpt-1020000.data-00000-of-00001  model.ckpt-1035000.data-00000-of-00001
model.ckpt-1020000.index                model.ckpt-1035000.index
model.ckpt-1020000.meta                 model.ckpt-1035000.meta
model.ckpt-1025000.data-00000-of-00001  model.ckpt-1040000.data-00000-of-00001
model.ckpt-1025000.index                model.ckpt-1040000.index
model.ckpt-1025000.meta                 model.ckpt-1040000.meta
model.ckpt-1030000.data-00000-of-00001

找到classify_image.py脚本here。

它需要.pb文件，而不是检查点文件。

过去两周我一直在试着弄清楚如何从检查点文件到.pb文件，所以我可以使用重新训练的模型。

任何想法都会受到赞赏。

Answer 1

任何想法都会受到赞赏。

https://www.tensorflow.org/hub/tutorials/image_retraining上的代码

cd ~
curl -LO http://download.tensorflow.org/example_images/flower_photos.tgz
tar xzf flower_photos.tgz

mkdir ~/example_code
cd ~/example_code
curl -LO https://github.com/tensorflow/hub/raw/master/examples/image_retraining/retrain.py

python retrain.py --image_dir ~/flower_photos

产生文件./output_graph.pb。

使用检查点模型启动v3 flowers_train

1 个答案: