目标检测 - 非极大值抑制(Non Maximum Suppression)

非极大值抑制(Non Maximum Suppression)

目标检测中，NMS被用于后期的物体边界框去除中.

NMS 对检测得到的全部 boxes 进行局部的最大搜索，以搜索某邻域范围内的最大值，从而滤出一部分 boxes，提升最终的检测精度.

NMS ：

• 输入：

  检测到的Boxes(同一个物体可能被检测到很多Boxes，每个box均有分类score)

• 输出：

  最优的Box.

• 过程：

  去除冗余的重叠 Boxes，对全部的 Boxes 进行迭代-遍历-消除.

  1.将所有框的得分排序，选中最高分及其对应的框;

  2. 遍历其余的框，如果和当前最高分框的重叠面积(IOU)大于一定阈值，则将框删除;

  3. 从未处理的框中继续选一个得分最高的，重复上述过程.

  ​

  例如：

  假设某物体检测到 4 个 Boxes，每个 Box 分别对应一个类别 Score，根据 Score 从小到大排列依次为，(B1, S1), (B2, S2), (B3, S3), (B4, S4). S4 > S3 > S2 > S1.

  • Step 1. 根据Scor