15. 必会--NMS

123 阅读1分钟

NMS 算法:

NMS发生在目标检测算法的最后一步,计算mAP的前一步。

图片.png

NMS 步骤:

(1) 将某一类别的bbx按照confidence从高到低排序。

(2) 记录当前confidence最大的bbx。

(3) 计算最大confidence对应的bbx与剩下所有的bbx的IoU,移除所有大于IoU阈值的bbx(保留小于IoU阈值的bbx索引)。

(4) 对剩下的bbx,循环执行(2)和(3)直到所有的bbx均满足要求(即不能再移除bbx)。

(5) 对每个类依次进行(1)~(4)操作。

(6) 跨种类比较,如果该框在多类别中都是较大confidence,保留最大confidence。