Error parsing message

版权声明:本文为博主原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。
本文链接:https://blog.csdn.net/nanchifeng3190/article/details/87642780
@使用jupyter notebook运行object_detection_tutorial.ipynb,运用的自己训练的pb出现错误。

detection_graph = tf.Graph()
with detection_graph.as_default():
  od_graph_def = tf.GraphDef()
  with tf.gfile.GFile(PATH_TO_FROZEN_GRAPH, 'rb') as fid:
    serialized_graph = fid.read()
    od_graph_def.ParseFromString(serialized_graph)
    tf.import_graph_def(od_graph_def, name='')

其中od_graph_def.ParseFromString(serialized_graph)出现Error parsing message的错误


经过检索,问题在于tensorflow生成的物体生成模型有两个。
路径分别为
/home/jitao/raccoon_dataset-master/train/saved_model/saved_model.pb
/home/jitao/raccoon_dataset-master/train/frozen_inference_graph.pb

应选择第二个路径,我错误的选择了第一个,导致了错误发生。
修改为第二个路径之后,问题解决。

参考网站:https://github.com/tensorflow/models/issues/2133
————————————————
版权声明:本文为CSDN博主「软件安装好难啊」的原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接及本声明。
原文链接:https://blog.csdn.net/nanchifeng3190/article/details/87642780

发布了296 篇原创文章 · 获赞 219 · 访问量 54万+

猜你喜欢

转载自blog.csdn.net/qq_36387683/article/details/103344848
今日推荐