DS-Trans: A 3D Object Detection Method Based on a Deformable Spatiotemporal Transformer for Autonomous Vehicles

• Published in: Remote sensing (Basel, Switzerland) Vol. 16; no. 9; p. 1621
• Main Authors: Zhu, Yuan, Xu, Ruidong, Tao, Chongben, An, Hao, Wang, Huaide, Sun, Zhipeng, Lu, Ke
• Format: Journal Article
• Language: English
• Published: Basel MDPI AG 01.05.2024
• Subjects: 3D object detection; Algorithms; Artificial intelligence; autonomous vehicle; Comparative analysis; data collection; Deformation; Feature extraction; Formability; Information processing; Machine learning; Methods; Misalignment; Modules; Neural networks; Object recognition; Object recognition (Computers); Pattern recognition; point clouds; Sensors; Spatial data; Transformer; Transformers; Vehicles; Weather; China
• ISSN: 2072-4292; 2072-4292
• DOI: 10.3390/rs16091621

Facing the significant challenge of 3D object detection in complex weather conditions and road environments, existing algorithms based on single-frame point cloud data struggle to achieve desirable results. These methods typically focus on spatial relationships within a single frame, overlooking the semantic correlations and spatiotemporal continuity between consecutive frames. This leads to discontinuities and abrupt changes in the detection outcomes. To address this issue, this paper proposes a multi-frame 3D object detection algorithm based on a deformable spatiotemporal Transformer. Specifically, a deformable cross-scale Transformer module is devised, incorporating a multi-scale offset mechanism that non-uniformly samples features at different scales, enhancing the spatial information aggregation capability of the output features. Simultaneously, to address the issue of feature misalignment during multi-frame feature fusion, a deformable cross-frame Transformer module is proposed. This module incorporates independently learnable offset parameters for different frame features, enabling the model to adaptively correlate dynamic features across multiple frames and improve the temporal information utilization of the model. A proposal-aware sampling algorithm is introduced to significantly increase the foreground point recall, further optimizing the efficiency of feature extraction. The obtained multi-scale and multi-frame voxel features are subjected to an adaptive fusion weight extraction module, referred to as the proposed mixed voxel set extraction module. This module allows the model to adaptively obtain mixed features containing both spatial and temporal information. The effectiveness of the proposed algorithm is validated on the KITTI, nuScenes, and self-collected urban datasets. The proposed algorithm achieves an average precision improvement of 2.1% over the latest multi-frame-based algorithms.