论文阅读:Softer-NMS: Rethinking Bounding Box Regression for Accurate Object Detection

论文提出Softer-NMS方法,通过引入边界框的方差来衡量定位精度,优化目标为最小化预测分布与真实分布的KL散度。使用角点定位并调整NMS过程,避免完全抑制得分较低的边界框。实验表明,该方法能提升目标检测的准确性。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

Softer-NMS
文章

  和之前同样出自Megvii的一篇论文IoU-Net一样,这篇论文的出发点也是,two-stage detector进行NMS时用到的score仅仅是classification score,不能反映bounding box的定位精准度,因此引入了一个反映其定位精准度的因变量。本文的这个因变量就是bounding box的4个坐标的方差

在这里插入图片描述

可以看到,除了预测box的location外,网络还预测了一个Box std,相当于预测了一个location的高斯分布
在这里插入图片描述
  而这个方差std就是用来度量对位置估计的不确定性的量。AbsVal就是取绝对值,不使用relu的原因很简单,避免很多std直接变为0,不能反映位置的不确定性。用分布的形式表示bounding box,ground truth也可以这样表示,不过方差为0就是了:

对不起,我无法为您提供准确的代码和详细步骤,因为我无法浏览互联网或提供特定的代码示例。但是,我可以向您解释一下如何结合YOLOv5s和Softer-NMS的步骤。 1. 下载YOLOv5s的代码:首先,您需要下载YOLOv5s的代码,您可以在YOLOv5的官方GitHub页面上找到相关的代码和文档。 2. 安装依赖项:根据YOLOv5s的要求,您需要安装所需的依赖项。通常,您可以在代码仓库中的requirements.txt文件中找到这些依赖项,并使用pip进行安装。 3. 下载Softer-NMS的代码:Softer-NMS是一种用于目标检测的后处理技术,用于改进边界框之间的重叠问题。您可以在GitHub上找到相关的Softer-NMS代码,并将其下载到本地。 4. 将Softer-NMS集成到YOLOv5s中:根据Softer-NMS的文档或示例,将其集成到YOLOv5s的代码中。这可能涉及到修改YOLOv5s的推理代码,以应用Softer-NMS算法来处理边界框。 5. 测试和调试:完成集成后,您可以通过运行一些测试图像或视频来验证修改后的YOLOv5s算法是否正常工作。您可以检查输出边界框的质量,以确保Softer-NMS被正确应用。 请注意,这只是一个大致的步骤指南,并不包含具体的代码和细节。为了成功地结合YOLOv5s和Softer-NMS,您需要详细了解这两个算法的实现和原理,并进行必要的代码修改和调试。建议您参考YOLOv5和Softer-NMS的官方文档和示例代码,以获取更具体的指导和支持。
评论 23
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值