非极大值抑制,即找到局部极大值,而非最大值,并抑制其领域内的其余值。在目标检测中,对于一个物体可能会预测出多个候选框,那么这时就可以用极大值抑制对一些冗余的框进行滤除。
一般来说,每一个预测框的输出都会带有该框的位置信息以及置信度。
NMS算法流程:
1、首先,对所有的框,通过一个置信度阈值将置信度低的框滤除。
2、接着,选出置信度最高的框,将其保存进输出列表中。
3、依次计算该框与其他剩余的框的IOU值。然后通过一个IOU阈值将和这个置信度最高的框拥有较大IOU的框(即和这个框相近的框)去除。也就是去掉冗余的框咯。
4、 继续对剩余的框进行2,3操作,直到遍历完毕。