RT-DETR:DETRs Beat YOLOs on Real-time Object Detection(CVPR2024)



原文
哇撒

Abstract

作者观察到NMS对YOLOs的速度和准确性有负面影响。(NMS进行后处理,不仅降低了推理速度,而且引入了导致速度和精度不稳定的超参数)。并且不同的场景对召回率和准确率的重视程度不同,需要仔细选择合适的NMS阈值,这阻碍了实时检测器的发展
DETRs为消除NMS提供了另一种选择,然而,高昂的计算成本限制了它们的实用性,阻碍了它们充分发挥排除NMS的优势

在本文中,作者提出了实时检测转换器(RT-DETR)解决NMS带来的影响,这是所知的第一个实时端到端对象检测器

RT-DETR包括两个关键增强:快速处理多尺度特征的高效混合编码器,以及提高初始对象查询质量的最小不确定性查询选择
并且RT-DETR支持灵活的速度调整,无需重新训练,消除了两个NMS阈值带来的不便,促进了其实际应用

RT-DETR-R50 /R101在COCO上实现53.1% / 54.3%的AP,在T4 GPU上实现108 / 74 FPS,在速度和精度方面都优于以前先进的yolo。此外,RT-DETR-R50在精度上比DINO-R50高出2.2%,在FPS上高出约21倍。使用Objects365进行预训练后,RT- der - r50 / R101的AP达到55.3% / 56.2%

在小目标上表现并不良好

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值