我正在自定义数据集上训练来自模型动物园的ssd_resnet50。通过从zoo-zoo提供的ssd_resnet目录中研究原始冻结图,我可以看到输出节点是“ num_detections,detection_classes,detection_boxes,detection_scores”。
在目标目录中训练后,有model.ckpt文件和graph.pbtxt文件。 我尝试通过``冻结图''脚本冻结图像(freeze_graph --input_graph graph.pbtxt --input_binary False --input_checkpoint model.ckpt-xxxx --out_graph outdir / graph-frozen.pb --output_node_names = num_detections, detection_classes,detection_boxes,detection_scores)。 我收到一条错误消息,提示“ AssertionError:num_detections不在图中”。
我做错了什么? graph.pbtxt和原始冻结的图形不应该包含相同的图形吗?
PS :(我知道export_nference_graph.py脚本,但是我想了解它的工作原理。export_inference_graph在exporter.py中隐藏了很多东西……)
谢谢您的答复。 问候, k。