如何在Tensorflow中建立预学习模型？

Question

以下文件是通过tf-slim创建的

model.ckpt-68593.data-00000-of-00001
model.ckpt-68593.index
model.ckpt-68593.meta

是否有一种方法可以将此文件集成到由tensorflow提供的单个.ckpt文件中

Answer 1

Tensorflow提供的预训练模型不是单个.ckpt文件，而是通常两个（.ckpt.data，.ckpt.index）或三个（.ckpt.data，.ckpt.meta ，.ckpt.index）个文件。您可以尝试以下操作来简单地加载变量：

with tf.Session() as sess:
    saver = tf.train.Saver()
    saver.restore(sess, "/tmp/model.ckpt")

不是因为有一个名为model.ckpt的文件，而是因为它会从saver.py中自动找到用于还原变量的文件。要恢复图形，可以使用以下方法：

tf.train.import_meta_graph('/tmp/model.ckpt.meta')

您可以参考TensorFlow, why there are 3 files after saving the model?以获得更多信息。