tensorflow1.0-解决加载图模型时报错google.protobuf.message.DecodeError: Error parsing message

使用以下代码,加载图模型的时候会报错

#保存图模型
    tf.train.write_graph(sess.graph_def, graph_dir, 'graph.pbtxt',as_text=True)
    #加载图模型
    with tf.gfile.FastGFile("modle_graph/graph.pbtxt","rb") as f:
        graph_def = tf.GraphDef()
        graph_def.ParseFromString(f.read())

graph_def.ParseFromString(f.read())
google.protobuf.message.DecodeError: Error parsing message

解决:

tf.train.write_graph(sess.graph_def, graph_dir, 'graph.pbtxt',as_text=False)

你可能感兴趣的:(机器学习)