开放图像预训练模型的图形文件在哪里?

时间:2017-07-03 13:29:40

标签: tensorflow pre-trained-model

我需要预训练的openimages数据集的图形文件(.pb文件)。下载模型后,只有3个文件

labelmap.txt    
model.ckpt  
model.ckpt.meta

我在哪里可以找到.pb文件?

1 个答案:

答案 0 :(得分:0)

model.ckpt.meta包含图形和权重。你必须使用import_metagraph功能导入它。 那么你应该能够以pb格式导出图形defn,如果你真的需要它。 但是由于模型中定义了特定于Google的操作,我无法加载ckpt.meta。如果你能搞清楚,请告诉我。