【发布时间】:2017-07-20 14:57:17
【问题描述】:
我遵循了宠物检测器教程,我使用“export_inference_graph.py”导出了模型。
但是,当我尝试使用提供的“freeze_graph.py”冻结图形时,现在确定使用什么 --output_node_names。
有谁知道我应该使用哪个,或者更重要的是,当我训练自己的模型时,我如何知道要使用什么。
【问题讨论】:
标签: python tensorflow object-detection
我遵循了宠物检测器教程,我使用“export_inference_graph.py”导出了模型。
但是,当我尝试使用提供的“freeze_graph.py”冻结图形时,现在确定使用什么 --output_node_names。
有谁知道我应该使用哪个,或者更重要的是,当我训练自己的模型时,我如何知道要使用什么。
【问题讨论】:
标签: python tensorflow object-detection
要查找用于 output_node_names 的内容,只需检查 graph.pbtxt 文件。在这种情况下,它是 Softmax
【讨论】:
export_inference_graph.py 的输出当前应该已经是一个冻结图,因此再次尝试冻结它是多余的。
对于生成的冻结图的输入和输出节点名称,请查看我们提供的 jupyter notebook。
【讨论】: