【问题标题】:freezing the "tensorflow object detection api pet detector" graph冻结“tensorflow对象检测api宠物检测器”图
【发布时间】:2017-07-20 14:57:17
【问题描述】:

我遵循了宠物检测器教程,我使用“export_inference_graph.py”导出了模型。

但是,当我尝试使用提供的“freeze_graph.py”冻结图形时,现在确定使用什么 --output_node_names。

有谁知道我应该使用哪个,或者更重要的是,当我训练自己的模型时,我如何知道要使用什么。

【问题讨论】:

    标签: python tensorflow object-detection


    【解决方案1】:

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

    【讨论】:

      【解决方案2】:

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

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

      【讨论】:

      • 感谢您解决这个问题。我已经用我的冻结图浏览了提供的记事本,但我目前收到此错误:KeyError:“名称'image_tensor:0'指的是不存在的张量。操作'image_tensor'不存在于图表。”
      • 我从 detection_graph.get_operations() pastebin.com/3XvMZrLE987654322@导出的所有操作名称列表
      • 我假设因为我没有类似的操作,某处出现了严重错误。我按照教程进行操作,使用我自己的图像和 labelImg 对它们进行注释。
      • 好的,所以我已经完成了整个过程,我的图像大小调整为原始大小的 20%,而且效果好多了,我的 eval 运行顺利。这次我使用了 ssd_inception_v2 特征提取器。这次当我查看我的 graph.pbtxt 时,我不确定应该使用哪个节点来导出? Softmax 不存在,ssd_inception_v2 是否有不同的输出节点?我想一个更好的问题是我如何知道从 graph.pbtxt 中选择哪个节点。我认为这是教程中唯一缺少的东西。
      猜你喜欢
      • 2021-09-24
      • 1970-01-01
      • 2019-08-05
      • 2021-12-18
      • 2017-12-02
      • 1970-01-01
      • 1970-01-01
      • 2018-02-27
      • 1970-01-01
      相关资源
      最近更新 更多