Two-stage 3D Object Detection Method Based on Self-attention Mechanism

Ying PENG , Sheng-gen ZHANG , Jun-fu HUANG , Qiang ZHANG

Science Technology and Engineering ›› 2024, Vol. 24 ›› Issue (25) : 10825 -10831.

PDF
Science Technology and Engineering ›› 2024, Vol. 24 ›› Issue (25) : 10825-10831 DOI: 10.12404/j.issn.1671-1815.2400232
research-article
Two-stage 3D Object Detection Method Based on Self-attention Mechanism
Author information +
History +
PDF

Abstract

In order to accurately identify the target obstacle in the traffic scene, considering the complexity of the real road scene and the importance of road safety, taking the sparsely embedded convolutional detection(SECOND) model as the basic model, the point cloud representation ability was enhanced by using the self-attention mechanism to obtain global semantic information, and the region of interest (RoI) detection head was used to optimize the 3D suggestion box generated by the candidate region to improve its detection accuracy. Then, two-stage 3D object detection method based on self-attention mechanism named SAR-SECOND was proposed. The results show that compared with the existing advanced 3D object detection methods, SAR-SECOND overall detection accuracy of car is 82.28%, the overall accuracy of pedestrians is 51.45%, and the overall accuracy of the rider is 72.41%. It can be seen that the effectiveness of this method is effective.

Keywords

autonomous vehicles / 3D object detection / attention mechanisms / sparse convolution

Cite this article

Download citation ▾
Ying PENG, Sheng-gen ZHANG, Jun-fu HUANG, Qiang ZHANG. Two-stage 3D Object Detection Method Based on Self-attention Mechanism. Science Technology and Engineering, 2024, 24(25): 10825-10831 DOI:10.12404/j.issn.1671-1815.2400232

登录浏览全文

4963

注册一个新账户 忘记密码

References

Funding

PDF (6365KB)

18

Accesses

0

Citation

Detail

Sections
Recommended

/

〈 〉