我正在尝试使用本教程中给出的方法生成 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
最佳答案
显然,您正在使用“/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 - 在tensorflow中生成pb文件,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/46846482/