3D point cloud object detection algorithm based on Transformer

LIU Mingyang,YANG Qiming,HU Guanhua,GUO Yan,Jiandong ZHANG,Mingyang LIU,Qiming YANG,Guanhua HU,Yan GUO
DOI: https://doi.org/10.1051/jnwpu/20234161190
2023-12-01
Xibei Gongye Daxue Xuebao/Journal of Northwestern Polytechnical University
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.
What problem does this paper attempt to address?