• 中国计算机学会会刊
  • 中国科技核心期刊
  • 中文核心期刊

Computer Engineering & Science

Previous Articles     Next Articles

A 3D vehicle detection algorithm
based on attention mechanism

WAN Si-yu   

  1. (School of Electronic Information and Electrical Engineering,Shanghai Jiao Tong University,Shanghai 200240,China)
  • Received:2019-07-20 Revised:2019-08-30 Online:2020-01-25 Published:2020-01-25

Abstract:

3D vehicle detection is a key problem in automatic driving scene, which involves 3D object detection and 3D object classification. Current 3D detection and classification networks treat all input point cloud data equally. However, in the actual detection process, the importance of different points in the point cloud for detection may not be the same. In order to get better detection results, attention mechanism is introduced to get the weights of the features of different points, so that the features of some points can get more attentions in regression. Experiments show that the model has higher accuracy than the existing methods while maintaining real-time efficiency.
 

Key words: vehicle detection, attention mechanism, deep learning