如何将 tensorflow .meta .data .index 转换为 .ckpt 文件？

Question

众所周知，在使用tensorflow保存检查点时，我们有 3 个文件，例如：

model.ckpt.data-00000-of-00001

模型.ckpt.index

模型.ckpt.meta

我检查了更快的rcn n，发现他们有一个evaluation.py 脚本可以帮助评估预训练模型，但该脚本只接受 .ckpt 文件（因为他们提供了上面的一些预训练模型）。

我已经从他们的预训练模型中进行了一些微调

然后我想知道是否有一种方法可以将所有.data-00000-of-00001 、 .index 和 .meta 转换为一个 .ckpt 文件以在检查点上运行评估.py 脚本？

（我还注意到他们在 repo 中提供的预训练模型确实只有 1 个.ckpt文件，当 save-checkpoint 函数生成 3 个文件时，他们如何做到这一点？）

Answer 1

这些

{ model.ckpt.data-00000-of-00001

模型.ckpt.index

模型.ckpt.meta }

是最近的检查点格式

尽管

{模型.ckpt}

是以前的检查点格式

它将与将 Nintendo Switch 转换为 NES 的概念相同......或者将 3 片 CD 包转换为单个 ROM 盒式磁带......

Answer 2

你不需要转换，你可以使用保存网络中的变量

saver = tf.train.Saver() 
saver.save(sess, 'path of save/fileName.ckpt')

要恢复网络以供稍后或在另一个脚本中重用，请使用：

saver = tf.train.Saver()
saver.restore(sess, tf.train.latest_checkpoint('path of save/')
sess.run(....) 

要点：

1. 第一次和以后的运行之间的 sess 必须相同（连贯结构）。

2. saver.restore 需要保存文件的文件夹路径，而不是单个文件路径。