Skip to main navigation Skip to search Skip to main content

Improved object detection method for unmanned driving based on Transformers

  • Huaqi Zhao
  • , Xiang Peng
  • , Su Wang
  • , Jun Bao Li
  • , Jeng Shyang Pan
  • , Xiaoguang Su
  • , Xiaomin Liu*
  • *Corresponding author for this work
  • Jiamusi University
  • Nanjing University of Information Science & Technology

Research output: Contribution to journalArticlepeer-review

Abstract

The object detection method serves as the core technology within the unmanned driving perception module, extensively employed for detecting vehicles, pedestrians, traffic signs, and various objects. However, existing object detection methods still encounter three challenges in intricate unmanned driving scenarios: unsatisfactory performance in multi-scale object detection, inadequate accuracy in detecting small objects, and occurrences of false positives and missed detections in densely occluded environments. Therefore, this study proposes an improved object detection method for unmanned driving, leveraging Transformer architecture to address these challenges. First, a multi-scale Transformer feature extraction method integrated with channel attention is used to enhance the network's capability in extracting features across different scales. Second, a training method incorporating Query Denoising with Gaussian decay was employed to enhance the network's proficiency in learning representations of small objects. Third, a hybrid matching method combining Optimal Transport and Hungarian algorithms was used to facilitate the matching process between predicted and actual values, thereby enriching the network with more informative positive sample features. Experimental evaluations conducted on datasets including KITTI demonstrate that the proposed method achieves 3% higher mean Average Precision (mAP) than that of the existing methodologies.

Original languageEnglish
Article number1342126
JournalFrontiers in Neurorobotics
Volume18
DOIs
StatePublished - 2024

Keywords

  • Transformer
  • feature extraction
  • object detection
  • optimal transport
  • query denoising

Fingerprint

Dive into the research topics of 'Improved object detection method for unmanned driving based on Transformers'. Together they form a unique fingerprint.

Cite this