A LiDAR Point Cloud Semantic Segmentation Algorithm Based on Attention Mechanism and Hybrid CNN-LSTM

Shuhuan Wen,Yongzhong Lu,Tao Wang,Artur Babiarz,Mhamed Sayyouri,Huaping Liu
2023-01-01
Abstract:Environment perception is a prerequisite for achieving autonomous driving, and it is of crucial significance to perceive environmental information and interact with dynamic scenes through LiDAR. Therefore, to address the issue of easily neglecting global contextual information when extracting point cloud features, this paper proposes a LiDAR point cloud semantic segmentation algorithm based on attention mechanism and hybrid CNN-LSTM. The proposed method initially utilizes a point embedding module to map points with similar features in the point cloud to nearby positions in a high-dimensional space. Furthermore, an offset feature combining self-attention features and input features is used as the input of the next-level attention module. The attention scores are redistributed into a 2D coordinate system as a pseudo-image, which is then subjected to high-level semantic information extraction and segmentation using encoding and decoding networks. Experiments of the proposed algorithm were performed in SemanticKITTI dataset. Compared to other methods, the proposed algorithm achieved higher semantic segmentation accuracy, and showed a significant improvement for most object classes.
What problem does this paper attempt to address?