Tensorflow Inception训练并创建GraphDef

时间:2017-01-07 08:08:54

标签: tensorflow

我很擅长使用Tensorflow。我正在这个回购中使用Inception-V3模型进行图像识别:

https://github.com/tensorflow/models/tree/master/inception#how-to-train-from-scratch

它提供二进制文件来帮助您训练自己的模型。但是在关于图像识别的Tensorflow文档中,

https://www.tensorflow.org/tutorials/image_recognition/

它们允许您下载和使用经过ImageNet数据培训的预训练的Inception-V3模型。您下载的模型采用.pb GraphDef的格式,而您在tensorflow/models/inception repo中生成的模型是.ckpt“检查点”格式的文件。

这两种类型的文件有什么区别?如何将生成的初始模型转换为GraphDef文件?

1 个答案:

答案 0 :(得分:0)

documentation on metagraphs应该会有所帮助。它显示了保存和加载这些包含足够信息的结构来重现模型。