tensorflow冻结图和graph.pbtxt不一致(?)

时间:2019-02-25 15:14:07

标签: python tensorflow

我正在自定义数据集上训练来自模型动物园的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。

0 个答案:

没有答案