학술논문

3D point cloud object detection algorithm based on Transformer
Document Type
article
Source
Xibei Gongye Daxue Xuebao, Vol 41, Iss 6, Pp 1190-1197 (2023)
Subject
transformer
spatial modulation attention mechanism
heat map initialization
target detection
deep learning
Motor vehicles. Aeronautics. Astronautics
TL1-4050
Language
Chinese
ISSN
1000-2758
2609-7125
Abstract
In response to the difficulty in deploying anchor box based methods in 3D object detection due to the increase in spatial dimensions, this paper studies a point cloud object detection algorithm based on set prediction. This article proposes a Transformer based 3D point cloud object detection algorithm, and combines the characteristics of point clouds in autonomous driving scenarios to propose an improved spatial modulation attention and heat map initialization strategy for training acceleration and query initialization, achieving good detection performance in shallow networks. This article compares it with other algorithms on the KITTI dataset, and the results show that our algorithm has reached an advanced level in performance. We also conducted ablation experiments on the main components of the algorithm to verify the contribution of each module to the detection effect.