Error parsing message

@使用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的错误
Error parsing message_第1张图片

经过检索,问题在于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

你可能感兴趣的:(视觉识别)