非极大抑制(Non-max suppression,NMS)算法

非极大抑制(Non-max suppression,NMS)算法
非极大抑制(Non-max suppression,NMS)算法_第1张图片
目标检测算法会生成很多候选框,这时我们要将多余的候选框去除,如上图,这就要用到NMS算法。

以YOLOV1举例,对于YOLOv1而言,网络最终会生成98个候选框。

代码中的做法是:对98个候选框进行筛选,先删除confidence小于0.1的,再对每个候选框分别计算score,score为confidence*最大类别概率,删除score<0.1的,得到剩余候选框列表L后再进行下一步NMS处理

先取L中score最大的候选框,然后删除L中与其IOU>0.5的候选框。再选取L中score次大的候选框,然后删除L中与其IOU>0.5的候选框。以此类推,同样的方法遍历一遍L,最终将剩余的候选框作为NMS算法的结果返回,类别即为该候选框类别概率最大的类。算法中使用到的阈值可进行调整。

你可能感兴趣的:(目标检测)