arXiv:2502.17843cs.CVcs.AI2025-02被引 5

用Transformer改进车辆检测,在复杂路况下更准更快。

Automatic Vehicle Detection using DETR: A Transformer-Based Approach for Navigating Treacherous Roads

  • 引入Co-DETR协同分配训练,增强特征与注意力学习
  • 在BadODD数据集上准确率优于传统YOLO和Faster R-CNN
  • 适合自动驾驶车辆检测场景,提升真实道路部署可行性

复杂多变的驾驶环境对自动车辆检测(AVD)提出挑战,包括光照、道路类型和车型差异。传统方法如YOLO和Faster R-CNN难以应对这些复杂性。本文首次将检测变压器(DETR)应用于复杂场景下的车辆检测,提出协作式混合标注训练方案Co-DETR,通过多样化标签分配策略和多并行辅助头增强训练监督,提取正样本坐标以提高训练效率。在BadODD数据集上的大量实验表明,该方法显著提升检测精度,在多种条件下表现更优,具备实际部署潜力。研究推动了自动驾驶导航技术发展,展示了融合CNN与Transformer在挑战性场景中实现鲁棒高效车辆检测的前景。

原文摘要 · Abstract (English)

Automatic Vehicle Detection (AVD) in diverse driving environments presents unique challenges due to varying lighting conditions, road types, and vehicle types. Traditional methods, such as YOLO and Faster R-CNN, often struggle to cope with these complexities. As computer vision evolves, combining Convolutional Neural Networks (CNNs) with Transformer-based approaches offers promising opportunities for improving detection accuracy and efficiency. This study is the first to experiment with Detection Transformer (DETR) for automatic vehicle detection in complex and varied settings. We employ a Collaborative Hybrid Assignments Training scheme, Co-DETR, to enhance feature learning and attention mechanisms in DETR. By leveraging versatile label assignment strategies and introducing multiple parallel auxiliary heads, we provide more effective supervision during training and extract positive coordinates to boost training efficiency. Through extensive experiments on DETR variants and YOLO models, conducted using the BadODD dataset, we demonstrate the advantages of our approach. Our method achieves superior results, and improved accuracy in diverse conditions, making it practical for real-world deployment. This work significantly advances autonomous navigation technology and opens new research avenues in object detection for autonomous vehicles. By integrating the strengths of CNNs and Transformers, we highlight the potential of DETR for robust and efficient vehicle detection in challenging driving environments.

车辆检测Transformer自动驾驶DETR

Thank you to arXiv for use of its open access interoperability. PaperDance 不是 arXiv 官方产品;中文卡片由大模型生成,请以原文为准。