目标检测 — NMS

Posted 2021-01-04 eilearn

1、非极大值抑制步骤

　　非极大值抑制算法（Non-maximum suppression，NMS）在目标检测中经常用到。我们的检测算法可能对同一目标产生多次检测的结果，非极大值抑制算法可以保证每个目标只检测一次,找到检测效果最好的框。

（1）去除所有预测框置信度于某个阈值的框，这里的阈值选取为0.5。

（2）在剩下的1、2、4、5、6号框中选取置信度最大的，即图中5号框,比较除5号框以外所有框于5号框的交并比，若交并比大于某个阈值（人工给定的值），则抑制这些不是局部极大值预测框输出。

（3）再从剩下的框中选取置信度最大的，重复上面的步骤，直至没有框被舍去时结束。

（4）最后2号和5号框被保留，其他框被抑制。

　　若检测物体是多个类别的时候，一般情况是对每个类别分别做一次非极大值抑制算法。