版权声明:本文为博主原创文章,遵循 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