gpt4 book ai didi

Tensorflow:尝试恢复 pb 文件中的模型时出错

转载 作者:行者123 更新时间:2023-11-30 09:43:49 28 4
gpt4 key购买 nike

我正在尝试加载取自 https://github.com/tensorflow/models/tree/master/official/resnet 的已训练模型,但是当我尝试加载 .pb 时,我在 ParseFromString 方法上收到错误:

import tensorflow as tf
from tensorflow.python.platform import gfile
GRAPH_PB_PATH = '../resnet_v2_fp32_savedmodel_NHWC/1538687283/saved_model.pb'
with tf.gfile.FastGFile(GRAPH_PB_PATH, "rb") as f:
graph_def = tf.GraphDef()
graph_def.ParseFromString(f.read())
g_in = tf.import_graph_def(graph_def, name="")
sess = tf.Session(graph=g_in)

DecodeError: Error parsing message

我错了什么?

最佳答案

我遇到了类似的问题,我没有使用 gfile,而是使用 tf.saved_model.loader.load 函数,如这篇文章 https://stackoverflow.com/a/46547595/4637693 中所示。 :

sess = tf.Session(graph=tf.Graph())
model = tf.saved_model.loader.load(sess, [tf.saved_model.tag_constants.SERVING], './model')
graph_def = model.graph_def

关于Tensorflow:尝试恢复 pb 文件中的模型时出错,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/55231461/

28 4 0
Copyright 2021 - 2024 cfsdn All Rights Reserved 蜀ICP备2022000587号
广告合作:1813099741@qq.com 6ren.com