LW-DETR: A Transformer Replacement to YOLO for Real-Time Detection

CV ODBO ViT
2024年06月05日
本文介绍了一种轻量级的检测变换器LW-DETR,其在实时物体检测方面的表现优于YOLO。该架构是一个简单的堆叠,包括ViT编码器、投影器和浅层DETR解码器。我们的方法利用了最近的先进技术,例如训练有效技术,如改进的损失和预训练,以及交错窗口和全局关注机制来降低ViT编码器的复杂性。我们通过聚合多级特征图和ViT编码器中的中间和最终特征图来改进ViT编码器,形成更丰富的特征图,并引入窗口为主的特征图组织来提高交错注意力计算的效率。实验结果表明,所提出的方法在COCO和其他基准数据集上优于现有的实时检测器,如YOLO及其变体。代码和模型可在(https://github.com/Atten4Vis/LW-DETR)上获得。
In this paper, we present a light-weight detection transformer, LW-DETR, which outperforms YOLOs for real-time object detection. The architecture is a simple stack of a ViT encoder, a projector, and a shallow DETR decoder. Our approach leverages recent advanced techniques, such as training-effective techniques, e.g., improved loss and pretraining, and interleaved window and global attentions for reducing the ViT encoder complexity. We improve the ViT encoder by aggregating multi-level feature maps, and the intermediate and final feature maps in the ViT encoder, forming richer feature maps, and introduce window-major feature map organization for improving the efficiency of interleaved attention computation. Experimental results demonstrate that the proposed approach is superior over existing real-time detectors, e.g., YOLO and its variants, on COCO and other benchmark datasets. Code and models are available at (https://github.com/Atten4Vis/LW-DETR).
许愿