[OneNet]OneNet: Towards End-to-End One-Stage Object Detection笔记

OneNet是一种创新的端到端one-stage目标检测网络,通过Minimum Cost Assignment策略实现One-to-One Label Assignment,避免了NMS后处理。该网络结合了location和classification的成本,提出了一种新的正样本选取方法。实验表明,这种方法提高了检测的准确性和效率,特别是在正样本选取上的改进,使得网络能够更好地聚焦于物体的关键区域。

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


paper: https://arxiv.org/pdf/2012.05780.pdf
code: https://github.com/PeizeSun/OneNet

1. Abstract

文中提出了一个观点,认为以往的one-stage无法去掉NMS实现端到端的主要原因在于正样本的选取。以往的Label assignment任务没有考虑分类的cost,只考虑了位置的cost,从而会得到很多冗余的boxes,造成了后处理中必须使用NMS。
在这里插入图片描述

如下图所示,RetinaNet选取正样本只考虑了box的IoU ,而FCOS只考虑了Point Assignment,本文作者提出的OneNet,将classification和location都考虑来选取正样本并计算cost,提出Minimum Cost Assignment方法实现One-to-One Label Assignment,对于每一个Ground Truth,只有一个正样本与之对应,从而获得最佳Match,消除NMS。
在这里插入图片描述

2. Contribution

① 网络是端到端的全卷积网络,没有ROI以及attention机制。
②正样本的选取基于对于location

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值