优化策略(十)后处理

非极大值抑制

  1. Nms
  2. Soft Nms
  3. DIou Nms

一、Soft Nms

优化策略(十)后处理

思路:不要粗鲁地删除所有IOU大于阈值的框,而是降低其置信度。

Method

先直接上伪代码,如下图:如文章题目而言,就是用一行代码来替换掉原来的NMS。按照下图整个处理一遍之后,指定一个置信度阈值,然后最后得分大于该阈值的检测框得以保留。

优化策略(十)后处理

原来的NMS可以描述如下:将IOU大于阈值的窗口的得分全部置为0。

优化策略(十)后处理

文章的改进有两种形式,一种是线性加权的:

优化策略(十)后处理

一种是高斯加权的:

优化策略(十)后处理

分析上面的两种改进形式,思想都是:M为当前得分最高框,bi 为待处理框,bi 和M的IOU越大,bi 的得分si 就下降的越厉害。

 

二、DIoU NMS

该nms在DIou loss一文中提出,在nms过程中采用DIoU的计算方式替换了IoU,由于DIoU的计算考虑到了两框中心点位置的信息,故使用DIoU进行评判的nms效果更符合实际,效果更优。
优化策略(十)后处理