试图将模型保存到 pb 文件，但我没有 .meta 文件

Question

我通过 TensorFlow 对象检测模块训练了一个自定义对象检测器模型，并使用 mobilenetssd 作为我的预训练模型。训练完成后，我有三个文件：

 checkpoint
ckpt-11.data-00000-of-00001
ckpt-11.index

另外我也有这个文件：

pipeline.config

我正在尝试将此模型保存为 pb 文件，并且我想使用此tutorial 中提供的程序。我可以在没有 .meta 文件的情况下运行这个程序吗？我将如何生成 .meta 文件？此外，我还能从哪里获得 output_node_names？

编辑：我也设法使用 chpt-11.index 推断出这个模型。

Answer 1

您只能使用上述文件加载模型。

模型将根据index file ckpt-11.index 和shard file ckpt-11.data-00000-of-00001 加载权重

基本上，shard 文件将包含模型权重，index 文件指示哪些权重存储在哪个分片中。 通常，如果您在单台机器上进行训练，则只有一个分片。 更多关于使用Tensorflow保存和加载模型的细节，可以参考this文档。