冻结" tensorflow物体检测api宠物探测器"图形

时间:2017-07-20 14:57:18

标签: python tensorflow object-detection

我已经按照宠物探测器教程,使用" export_inference_graph.py"导出模型。

然而,当我尝试使用提供的" freeze_graph.py"冻结图表时但现在确定要使用什么--output_node_names。

有没有人知道我应该使用哪个,或者更重要的是我如何在我训练自己的模型时找出用来做什么。

2 个答案:

答案 0 :(得分:0)

要查找output_node_names的用途,只需检查graph.pbtxt文件即可。在这种情况下,它是Softmax

答案 1 :(得分:0)

export_inference_graph.py的输出当前应该已经是一个冻结图形,因此尝试再次冻结它会是多余的。

对于生成的冻结图的输入和输出节点名称,请查看我们提供的jupyter notebook