非极大值抑制(non-maximum suppression,NMS)

非极大值抑制(non-maximum suppression,NMS)

在进行目标位置预测时,可能会输出许多相似的具有明显重叠的预测边界框,都围绕着同一目标(如下图),此时只需要保留一个锚框即可。那么根据每个锚框预测出的置信度进行选择,只有置信度最高的被保留,其余非极大锚框,就会被删除

出现多个锚框预测结果的图片:
在这里插入图片描述

通过非极大抑制处理后的物体预测图片:
在这里插入图片描述
图片来源:李沐-动⼿学深度学习


版权声明:本文为qq_43892462原创文章,遵循CC 4.0 BY-SA版权协议,转载请附上原文出处链接和本声明。