arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2304.08069cs.CV

DETR在实时目标检测上超越YOLO

DETRs Beat YOLOs on Real-time Object Detection

Baidu Inc(百度公司);School of Electronic and Computer Engineering, Peking University(北京大学电子与计算机工程学院)

Yian Zhao, Wenyu Lv, Shangliang Xu, Jinman Wei, Guanzhong Wang, Qingqing Dang, Yi Liu, Jie Chen

AI总结:

本文提出首个实时端到端检测器RT-DETR,通过高效混合编码器和不确定性最小化查询选择,在速度和精度上均超越YOLO系列,并支持无需重训的灵活速度调节。

摘要:

YOLO系列因其在速度与精度之间的合理权衡,已成为实时目标检测中最受欢迎的框架。然而,我们观察到YOLO的速度和精度受到NMS的负面影响。近年来,基于Transformer的端到端检测器(DETR)为消除NMS提供了一种替代方案。然而,高昂的计算成本限制了其实用性,并阻碍了它们充分发挥去除NMS的优势。在本文中,我们提出了实时检测Transformer(RT-DETR),据我们所知,这是第一个解决上述困境的实时端到端目标检测器。我们借鉴先进的DETR,分两步构建RT-DETR:首先在提高速度的同时保持精度,然后在保持速度的同时提高精度。具体而言,我们设计了一种高效的混合编码器,通过解耦尺度内交互和跨尺度融合来快速处理多尺度特征,以提高速度。然后,我们提出了不确定性最小化的查询选择,为解码器提供高质量的初始查询,从而提高精度。此外,RT-DETR通过调整解码器层数支持灵活的速度调节,无需重新训练即可适应各种场景。我们的RT-DETR-R50 / R101在COCO上达到53.1% / 54.3%的AP,在T4 GPU上达到108 / 74 FPS,在速度和精度上均优于先前先进的YOLO。我们还开发了扩展的RT-DETR,其性能优于更轻量的YOLO检测器(S和M模型)。此外,RT-DETR-R50在精度上比DINO-R50高出2.2% AP,在FPS上快约21倍。在使用Objects365预训练后,RT-DETR-R50 / R101达到55.3% / 56.2%的AP。项目页面:https://zhao-yian.github.io/RTDETR。

英文摘要:

The YOLO series has become the most popular framework for real-time object detection due to its reasonable trade-off between speed and accuracy. However, we observe that the speed and accuracy of YOLOs are negatively affected by the NMS. Recently, end-to-end Transformer-based detectors (DETRs) have provided an alternative to eliminating NMS. Nevertheless, the high computational cost limits their practicality and hinders them from fully exploiting the advantage of excluding NMS. In this paper, we propose the Real-Time DEtection TRansformer (RT-DETR), the first real-time end-to-end object detector to our best knowledge that addresses the above dilemma. We build RT-DETR in two steps, drawing on the advanced DETR: first we focus on maintaining accuracy while improving speed, followed by maintaining speed while improving accuracy. Specifically, we design an efficient hybrid encoder to expeditiously process multi-scale features by decoupling intra-scale interaction and cross-scale fusion to improve speed. Then, we propose the uncertainty-minimal query selection to provide high-quality initial queries to the decoder, thereby improving accuracy. In addition, RT-DETR supports flexible speed tuning by adjusting the number of decoder layers to adapt to various scenarios without retraining. Our RT-DETR-R50 / R101 achieves 53.1% / 54.3% AP on COCO and 108 / 74 FPS on T4 GPU, outperforming previously advanced YOLOs in both speed and accuracy. We also develop scaled RT-DETRs that outperform the lighter YOLO detectors (S and M models). Furthermore, RT-DETR-R50 outperforms DINO-R50 by 2.2% AP in accuracy and about 21 times in FPS. After pre-training with Objects365, RT-DETR-R50 / R101 achieves 55.3% / 56.2% AP. The project page: https://zhao-yian.github.io/RTDETR.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑