非极大值抑制

从一堆重叠的检测框里只留下分数最高那个的后处理步骤。

目标检测器常对同一个物体输出多个互相重叠的候选框。非极大值抑制按置信度排序，先保留分数最高的框，再删掉与它交并比（IoU，两框重叠面积占并集的比例）超过阈值的其他框，重复直到处理完。它是 YOLO、Faster R-CNN 等检测器的标准后处理，也用于抓取位姿检测中去掉重复的抓取候选。阈值太高会留下重复框，太低会误删挨得很近的真实物体；DETR 等端到端检测器设计上不需要 NMS。

例子检测器对同一个杯子输出 5 个重叠框，NMS 保留分数最高的一个，删掉与它 IoU 超过 0.5 的其余框。

也叫

NMS 后处理

相关

目标检测、检测框（边界框）、交并比、YOLO、DETR、抓取位姿检测

来源

torchvision.ops.nms - PyTorch Documentation