Tensorflow 对象检测中的预测数量限制 API

Question

我用 Tensorflow 对象检测 API 训练了 Faster R-CNN 模型，但遇到了一个奇怪的问题。该模型的输出有 最多 100 个预测 ，尽管图像中有更多的对象。这是我测试过的每张图片的案例。

我在 Ten GitHub 上发现了类似的问题，但据我所知，他们在这些方面做得并不多。 https://github.com/tensorflow/tensorflow/issues/30464

也许您过去遇到过类似的问题？知道如何解决这个问题吗？

Answer 1

100 的限制是每个 class，那么您的最大总检测数为 300。

这是 pipeline.config 文件 second_stage_post_processing 部分中设置的网络配置参数。例如当前faster_rcnn_inception_v2_coco.config有：

second_stage_post_processing {
      batch_non_max_suppression {
        score_threshold: 0.0
        iou_threshold: 0.6
        max_detections_per_class: 100
        max_total_detections: 300
      }
      score_converter: SOFTMAX
    }

在训练你的网络之前改变这些值（虽然我不知道改变它们如何影响网络大小and/or在使用预训练检查点进行微调时是否会导致问题）

Tensorflow 对象检测中的预测数量限制 API

Limitation of number of predictions in Tensorflow Object Detection API

python

object-detection

tensorflow

object-detection-api