NMS 算法:
NMS发生在目标检测算法的最后一步,计算mAP的前一步。
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。