SoftNMS的配置

参考知乎:

https://zhuanlan.zhihu.com/p/41046620

1.置信度阈值:

假如还检测出了3号框,而我们的最终目标是检测出1号和2号框,并且剔除3号框,原始的nms只会检测出一个1号框并剔除2号框和3号框,而softnms算法可以对1、2、3号检测狂进行置信度排序,可以知道这三个框的置信度从大到小的顺序依次为:1-》2-》3(由于是使用了惩罚,IoU越大,得分越低,所有可以获得这种大小关系),如果我们再选择了合适的置信度阈值,就可以保留1号和2号,同时剔除3号,实现我们的功能。

但是,这里也有一个问题就是置信度的阈值如何选择,作者在这里依然使用手工设置的值,依然存在很大的局限性,所以该算法依然存在改进的空间。

2. 衰减f(iou(M,bi))权重函数的形式:

原来的NMS可以描述如下:将IoU大于阈值的窗口的得分全部置为0。

Soft-NMS的改进有两种形式,一种是线性加权的:

一种是高斯加权的:

实际应用测试结果:

置信度阈值=0.0001

soft-NMS在人体检测上FPPI为1时提升1个点,FPPI为0.05没有结果不变

置信度阈值=0.1

soft-NMS在人体检测上FPPI为1时提升1,FPPI为0.05掉点。受到测试集合不干净影响。


NMS方式

FPPI@0.05

FPPI@1

备注

precisionrecallprecisionrecall 
原始NMS0.99715 0.583340.962130.84840 

softNMS-线性加权

0.99713 0.580050.962490.85694置信阈值0.1
softNMS-高斯加权 0.99624 0.441680.962760.86333置信阈值0.1
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值