Flexible NMS

来源:互联网 发布:淘宝联盟如何购买省钱 编辑:程序博客网 时间:2024/05/17 08:08
Flexible NMS方法如下:

1)对于置信度最高的边界框,将其与iou> 0.8的所有其他类似框组合,框的位置作为具有置信度权重的边界框位置的加权平均值。

2)对于组合框的置信度,我使用conf = sum(最多N个边界框)/ N

。这对组合多个结果特别有用,我还想惩罚那些只有单个网络找到的一个结果(相比更多网络/边界框预测的置信度更高的结果)。

我使用N == 4 ×预测数的组合。

3)对于与当前框重叠的所有其他框,我按照原始 Soft NMS文章中的所述调整了置信度。

总体来说,这种方法让我从单个模型、默认NMS得分~0.72–0.73,增加到0.77-0.78,后面是使用 Flexible NMS从2个不同模型组合大约6个预测的结果。

我最终提交的结果,训练了3个Faster RCNN模型,对于所有图像的每个预测,翻转和±20%调整大小。

我将所有结果结合,禁用原始NMS,并通过Flexible NMS进行后处理


这个方法扩展了Soft NMS的原始概念,不仅调整置信区间,还要调整边界框的位置。

我在NMS阶段之前检查了Faster-RCNN的结果,发现它经常产生一些具有相似置信区间和位置的封闭边界框。

只保留置信区间最上层结果,抛弃其他所有的结果,听起来有点浪费,尤其是当结合使用多个模型或测试时间扩充以后。


原创粉丝点击