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

Question

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

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

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

Answer 1

100 的限制是每个类，那么您的最大总检测数为 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
    }

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