Python 加载pb文件:解码错误:错误解析消息

Python 加载pb文件:解码错误:错误解析消息,python,tensorflow,protocol-buffers,tfx,Python,Tensorflow,Protocol Buffers,Tfx,我正在尝试加载从pusher tfx管道生成的.pb文件。我正在使用以下函数加载文件,但从该函数中得到以下错误。请帮忙 错误: <ipython-input-40-af7ef7ac8a8b> in load_model() 2 with tf.compat.v2.io.gfile.GFile('/home//saved_model.pb', "rb") as f: 3 graph_def = tf.compat.v1.GraphDef(

我正在尝试加载从pusher tfx管道生成的.pb文件。我正在使用以下函数加载文件,但从该函数中得到以下错误。请帮忙

错误:

<ipython-input-40-af7ef7ac8a8b> in load_model()
      2     with tf.compat.v2.io.gfile.GFile('/home//saved_model.pb', "rb") as f:
      3         graph_def = tf.compat.v1.GraphDef()
----> 4         graph_def.ParseFromString(f.read())
      5 
      6     with tf.Graph().as_default() as graph:
DecodeError: Error parsing message

嘿,您可以尝试以下代码来加载tensorflow提供的.pb文件:

import tensorflow as tf
import sys
from tensorflow.python.platform import gfile
from tensorflow.core.protobuf import saved_model_pb2
from tensorflow.python.util import compat

with tf.Session() as sess:
    model_filename ='saved_model.pb'
    with gfile.FastGFile(model_filename, 'rb') as f:
        data = compat.as_bytes(f.read())
        sm = saved_model_pb2.SavedModel()
        sm.ParseFromString(data)
        g_in = tf.import_graph_def(sm.meta_graphs[0].graph_def)

还有更多的错误吗?太模糊了嗨,我已经把错误信息添加到了检查这篇文章中。嗨,湿婆,谢谢你的指导。在不得不将graph_def.ParseFromString(f.read())替换为google.protobuf,text_format.Merge(f.read(),graph_def)之后,我遇到了不同的错误:UnicodeDecodeError:“utf-8”编解码器无法解码位置3中的字节0xb3:无效的start byteTanks!这对我有用。你知道做预测的下一步是什么吗?在Keras中非常简单,我只需加载模型,然后对其调用predict(),但在tensorflow中,我不知道如何实现它。
import tensorflow as tf
import sys
from tensorflow.python.platform import gfile
from tensorflow.core.protobuf import saved_model_pb2
from tensorflow.python.util import compat

with tf.Session() as sess:
    model_filename ='saved_model.pb'
    with gfile.FastGFile(model_filename, 'rb') as f:
        data = compat.as_bytes(f.read())
        sm = saved_model_pb2.SavedModel()
        sm.ParseFromString(data)
        g_in = tf.import_graph_def(sm.meta_graphs[0].graph_def)