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 （见下文）

• “checkpoint”文件仅用于告知某些TF函数，这是最新的检查点文件。

• .ckpt-meta包含.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()方法，但我不知道它究竟做了什么。

这里有很多关于如何保存和恢复图表的问题。 例如，请看这里的答案，但要注意两个引用的教程虽然真的很有用，但远非完美，很多人似乎仍然很难用c ++导入模型。

编辑 ：看起来你现在也可以在c ++中使用.ckpt文件，所以我猜你不一定需要.pb文件了。