1、非极大值抑制步骤
非极大值抑制算法(Non-maximum suppression,NMS)在目标检测中经常用到。我们的检测算法可能对同一目标产生多次检测的结果,非极大值抑制算法可以保证每个目标只检测一次,找到检测效果最好的框。
(1)去除所有预测框置信度于某个阈值的框,这里的阈值选取为0.5。
(2)在剩下的1、2、4、5、6号框中选取置信度最大的,即图中5号框,比较除5号框以外所有框于5号框的交并比,若交并比大于某个阈值(人工给定的值),则抑制这些不是局部极大值预测框输出。
(3)再从剩下的框中选取置信度最大的,重复上面的步骤,直至没有框被舍去时结束。
(4)最后2号和5号框被保留,其他框被抑制。
若检测物体是多个类别的时候,一般情况是对每个类别分别做一次非极大值抑制算法。