目标检测中的非极大值抑制(Non-Maximum Suppression, NMS)及其变体是用于后处理的关键技术,用于消除重复的检测框,确保每个目标仅保留一个最优的预测结果。以下是对NMS及其主要变体的详细介绍:
1. 标准NMS(Non-Maximum Suppression)
原理:
NMS通过保留置信度最高的检测框,并抑制与其重叠度(IoU, Intersection over Union)较高的其他框,从而消除冗余检测。
步骤:
- 将所有检测框按置信度从高到低排序。
- 选择最高置信度的框,计算其与剩余框的IoU。
- 若某框与当前最高分框的IoU超过预设阈值(如0.5),则将其剔除。
- 重复上述过程,直到所有框被处理。
局限性:
- 遮挡问题:当两个目标靠得很近时,可能误删正确检测框。
- 硬阈值依赖:固定的IoU阈值对密集目标的适应性差。
- 分类与定位不一致:置