tensorflow - 在tensorflow中生成pb文件

标签 tensorflow

我正在尝试使用本教程中给出的方法生成 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/

相关文章:

tensorflow - "Images"选项卡在 Tensorboard 上显示什么?

python - Tensorflow将字符串列转换为多列进行分类

tensorflow - 归一化为 [0,1] 与 [-1,1]

build - 配置 Tensorflow 时,我应该在 "compute capability list"字段中输入什么?

python - tensorflow 错误KeyError : "The name ' import/Mul' refers to an Operation not in the graph.“

python - TFLearn 对每个预测产生相同的结果

python - 尝试导入 Tensorflow 时出现 ModuleNotFoundError

python - TensorFlow 分布式运行时模型并行 CIFAR-10

python - 服务 tensorflow 分类器

javascript - 如何使用 tensorflowjs_converter 转换 TensorFlow 图形 .pb 文件?