NMS—非极大值抑制算法的理解

非极大值抑制(NMS)在目标检测中用于去除重合度高的候选框,提高检测效率。通过排序得分、计算IOU并设定阈值来迭代删除冗余框,最终保留最具代表性的结果。例如在人脸检测中,经过NMS处理,可以有效消除多余框,如B:0.98和E:0.81被保留。
摘要由CSDN通过智能技术生成

一、概念

在目标检测中,得到多个候选框及其置信度得分。非极大值抑制算法(NMS)对多个候选框,去除重合率大的冗余候选框,得到最具代表性的结果,以加快目标检测的效率。

二、实例

以下图人脸检测为例,目标:消除重合率大且多余的候选框,保留某个区域的一个最优的候选框。

非极大值抑制算法(NMS)抑制冗余的框, 抑制的过程是一个迭代-遍历-消除的过程。

这里写图片描述

1)将所有框的得分排序,选中最高分及其对应的框

a) 假设图中有A:0.75、B:0.98 、C:0.83、D:0.67、E:0.81

b) 将置信度升序排序为D:0.67、A:0.75、E:0.81、C:0.83、B:0.98

c) 选中得分最高的B:0.98

2)遍历其余的框,如果和当前最高分框的重叠面积(IOU)大于一定阈值,我们就将框删除。

a)由B:0.98对其余A、C、D、E框计算IOU,B与A、C的IOU>阈值,删除A、C框。

b)第一轮得到B:0.98、D:0.67、E:0.81

这里写图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值