NMS算法原理及代码验证

NMS算法在图像目标检测中用于去除重复的检测框,保持最高置信度的框。通过设置IOU阈值,按置信度排序并移除与最高置信度框IOU超过阈值的框,最终每个物体仅保留一个最佳检测框。NMS算法的IOU阈值调整是优化目标检测的关键。
摘要由CSDN通过智能技术生成

NMS算法常常在图像目标检测的各种算法中使用。原始图像中,单个检测目标往往被识别出多个位置、大小都不相同的目标检测框,这些检测框可以是类似于这样的名字,bbox(bounding box), region proposal, anchors等等。可以在诸如faster rcnn或YOLO算法中看到。

举例来讲,原始图中包含两个检测物体,每个检测物体会存在许多检测框。因为是同一个物体,因此这些框的IOU值都很高(相比较不同物体的检测框来说)。针对同一个检测目标,有必要只留下置信度(score)最高的那个检测框作为该物体的检测框,其它的都应该过滤掉。

该算法具体的步骤如下:

1、假定有6个带置信率的region proposals,并预设一个IOU的阈值如0.7。

2、按置信率大小对6个框排序: 0.95, 0.9, 0.9, 0.8, 0.7, 0.7。

3、设定置信率为0.95的region proposals为一个物体框;

4、在剩下5个region proposals中,去掉与0.95物体框IOU大于0.7的。

5、重复2~4的步骤,直到没有region proposals为止。

6、每次获取到的最大置信率的region proposals就是我们筛选出来的目标。

这个操作的直观结果,就是图片上每个待检查物体,仅仅保留一个置信率最大region proposals。

凡是与这个置信率最大的框相交面积大于某个阈值的其它框,统统去掉。

因此,调

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值