一、非极大值抑制
非极大值抑制(Non-Maximum Suppression,NMS)是目标检测领域中常用的一种后处理技术,用于在检测到的候选目标框中去除冗余和重叠的框,从而得到最终的检测结果。NMS的主要目标是筛选出具有最高置信度的目标框,并去除与其高度重叠的其他框。
在目标检测任务中,一个目标可能会被多个不同的检测器检测出来,导致多个相交或者重叠的候选框。NMS的主要思想是通过一系列步骤来过滤掉置信度较低或与更高置信度框重叠较多的候选框,保留具有高置信度且不会过多重叠的目标框。
NMS的基本步骤如下:
-
按照置信度排序: 首先,将所有候选框按照其置信度从高到低进行排序。
-
选择最高置信度框: 选取置信度最高的候选框,作为保留的检测结果之一,并将该框从候选框列表中移除。
-
计算IoU: 对于剩余的候选框,计算它们与已选框之间的交并比(IoU,Intersection over