@使用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
应选择第二个路径,我错误的选择了第一个,导致了错误发生。
修改为第二个路径之后,问题解决。