我已经按照宠物探测器教程,使用" export_inference_graph.py"导出模型。
然而,当我尝试使用提供的" freeze_graph.py"冻结图表时但现在确定要使用什么--output_node_names。
有没有人知道我应该使用哪个,或者更重要的是我如何在我训练自己的模型时找出用来做什么。
答案 0 :(得分:0)
要查找output_node_names的用途,只需检查graph.pbtxt文件即可。在这种情况下,它是Softmax
答案 1 :(得分:0)
export_inference_graph.py的输出当前应该已经是一个冻结图形,因此尝试再次冻结它会是多余的。
对于生成的冻结图的输入和输出节点名称,请查看我们提供的jupyter notebook。