我使用提供的脚本here从头开始训练初始模型。 培训的输出是这些文件:
checkpoint
events.out.tfevents.1499334145.fdbf-Dell
model.ckpt-5000.data-00000-of-00001
model.ckpt-5000.index
model.ckpt-5000.meta
...
model.ckpt-25000.data-00000-of-00001
model.ckpt-25000.index
model.ckpt-25000.meta
是否有人使用脚本将这些文件转换为可用于对图像进行分类的内容?我已经尝试修改inception_train.py文件以输出graph.pb,但没有任何反应......
任何帮助将不胜感激,谢谢!
答案 0 :(得分:0)
如何直接使用您的检查点here。
要创建.pb文件,您必须冻结图表,如here所述。
要创建freeze_graph所需的初始.pb(或.pbtxt)文件,您可以使用tf.train.write_graph()