Tensorflow：.ckpt 文件和 .ckpt.meta 和 .ckpt.index 和 .pb 文件有什么关系

Question

我使用saver=tf.train.Saver()保存了我训练的模型，我得到了三种文件，命名为：

• .ckpt.meta
• .ckpt.index
• .ckpt.data

还有一个文件叫：

• 检查点

与 .ckpt 文件有什么关系？

我看到有人保存的模型只有 .ckpt 文件，我不知道如何制作。 如何将模型保存为 .pb 文件？

Answer 1

• .ckpt 文件是saver.save(sess) 的旧版本输出，相当于您的.ckpt-data（见下文）

• “检查点”文件只是在这里告诉一些 TF 函数哪个是最新的检查点文件。

• .ckpt-meta 包含元图，即计算图的结构，没有变量的值（基本上你可以在 tensorboard/graph 中看到）。

• .ckpt-data 包含所有变量的值，没有结构。要在 python 中恢复模型，您通常会使用元数据和数据文件（但您也可以使用.pb 文件）：

  saver = tf.train.import_meta_graph(path_to_ckpt_meta)
  saver.restore(sess, path_to_ckpt_data)
  

• 我不确切知道.ckpt-index，我猜这是正确映射前两个文件的内部需要的某种索引。反正平时没必要，只用.ckpt-meta和.ckpt-data就可以恢复模型。

• .pb 文件可以保存整个图形（元 + 数据）。要在 c++ 中加载和使用（但不训练）图形，您通常会使用它，使用 freeze_graph 创建，它从元数据和数据创建 .pb 文件。小心，（至少在以前的 TF 版本和某些人中）freeze_graph 提供的 py 函数不能正常工作，所以你必须使用脚本版本。 Tensorflow还提供了tf.train.Saver.to_proto()方法，但具体是做什么的我不知道。

这里有很多关于如何保存和恢复图表的问题。例如，请参阅答案here，但请注意，引用的两个教程虽然非常有用，但远非完美，而且很多人似乎仍然难以在 C++ 中导入模型。

编辑： 它看起来像you can also use the .ckpt files in c++ now,，所以我猜你不一定需要 .pb 文件了。