文章目录 NMS原理介绍 DIoU-NMS效果展示 NMS 耗时对比 YOLOv5 更换方式 YOLOv7 更换方式 NMS原理介绍 在执行目标检测任务时,算法可能对同一目标有多次检测。NMS 是一种让你确保算法只对每个对象得到一个检测框的方法。 在正式使用NMS之前,通常会有一个候选框预清理的工作(简单引入一个置信度阈值),如下图所示: NMS 算法的大致过程:每轮选取置信度最大的Bounding Box(简称BBox) 接着关注所有剩下的BBox中与选取的BBox有着高重叠(IoU)的,它们将在这