python - Tensorflow : What is the relationship between . 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 文件了。