一、概念
在目标检测中,得到多个候选框及其置信度得分。非极大值抑制算法(NMS)对多个候选框,去除重合率大的冗余候选框,得到最具代表性的结果,以加快目标检测的效率。
二、实例
以下图人脸检测为例,目标:消除重合率大且多余的候选框,保留某个区域的一个最优的候选框。
非极大值抑制算法(NMS)抑制冗余的框, 抑制的过程是一个迭代-遍历-消除的过程。
1)将所有框的得分排序,选中最高分及其对应的框
a) 假设图中有A:0.75、B:0.98 、C:0.83、D:0.67、E:0.81
b) 将置信度升序排序为D:0.67、A:0.75、E:0.81、C:0.83、B:0.98
c) 选中得分最高的B:0.98
2)遍历其余的框,如果和当前最高分框的重叠面积(IOU)大于一定阈值,我们就将框删除。
a)由B:0.98对其余A、C、D、E框计算IOU,B与A、C的IOU>阈值,删除A、C框。
b)第一轮得到B:0.98、D:0.67、E:0.81