我使用 SSD 进行训练TensorFlow 对象检测 API https://research.googleblog.com/2017/06/supercharge-your-computer-vision-models.html如上所述here https://github.com/tensorflow/models/blob/master/object_detection/g3doc/running_locally.md。它生成 ckpt、元和索引文件。为了在我的图像上运行它，我尝试检查演示代码。它要求将模型转换为冻结图。我尝试将模型转换为冻结推理图，如下所述here https://stackoverflow.com/questions/42463896/prediction-using-inceptionv3-in-tensorflow。在该程序中，我必须提供输出节点名称。我无法弄清楚此处必须使用的 SSD 模型中的节点名称。请帮忙。我尝试了 'num_detections:0'、'detection_boxes:0' 等。只是得到错误：
AssertionError：num_detections 不在图中
我们有一个特殊的工具可以在 Tensorflow 对象检测 API 中转换为冻结图 --- 只需运行导出推理图.py https://github.com/tensorflow/models/blob/master/research/object_detection/export_inference_graph.py二进制。使用此工具的说明是here https://github.com/tensorflow/models/blob/master/research/object_detection/g3doc/exporting_models.md.
本文内容由网友自发贡献，版权归原作者所有，本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容，请联系:hwhale#tublm.com(使用前将#替换为@)
