Question

我正在尝试使用本教程中给出的方法生成一个pb文件，

http://cv-tricks.com/how-to/freeze-tensorflow-models/

import tensorflow as tf
saver = tf.train.import_meta_graph('/Users/pr/tensorflow/dogs-cats-model.meta', clear_devices=True)
graph = tf.get_default_graph()
input_graph_def = graph.as_graph_def()
sess = tf.Session()
saver.restore(sess, "./dogs-cats-model")

当我尝试运行此代码时，我收到此错误 -

DataLossError (see above for traceback): Unable to open table file ./dogs-cats-model: Data loss: file is too short to be an sstable: perhaps your file is in a different file format and you need to use a different restore operator?

当我用Google搜索此错误时，大多数人建议使用版本2格式生成元文件？这是正确的做法吗？

使用的Tensorflow版本 -

1.3.0

Answer 1

显然，你正在使用'/Users/pr/tensorflow/dogs-cats-model.meta'和'./dogs-cats-model.meta'。你确定他们指向同一个文件吗？

以下代码在我的机器上运行良好：

import tensorflow as tf
saver = tf.train.import_meta_graph('./dogs-cats-model.meta', clear_devices=True)
graph = tf.get_default_graph()
input_graph_def = graph.as_graph_def()
sess = tf.Session()
saver.restore(sess, "./dogs-cats-model")

在tensorflow中生成pb文件

1 个答案: