IEEE Access (Jan 2024)
A Graph Aggregation Convolution and Attention Mechanism Based Semantic Segmentation Method for Sparse Lidar Point Cloud Data
Abstract
In recent years, following the development of sensor and computer techniques, it is favored by many fields, i.e. automatic drive, intelligent home, etc., which the deep learning based semantic segmentation method for point cloud data collected by LiDAR. This type method can automatic extract features of point cloud, helping label semantic categories. However, compared to 2D images, 3D point cloud data is more expensive to acquire. Hence, to save research and production costs, the low-thread LiDAR is a good choice. For one observation scenario, following the decrease of the line, the point cloud becomes sparse, which may cause the information loss. To balance the cost and the segmentation effect, we provide a point cloud completion auxiliary semantic segmentation method. Here, the baseline of the proposed method is Bilateral Augmentation and Adaptive Fusion (BAAF) model. It is the main contribution that a completion module introduction in feature extraction part of BAAF. Under the premise of using low-thread LiDAR sensor to collect data, the semantic segmentation effect of 3D field point cloud is improved as much as possible. It provides theoretical basis for cost saving in practical industrial application. The feature extraction of completion module consists of Graph Aggregation Convolution (GAC) and attention mechanism. Then, we use shuffle transform to upsampling data. In addition, to analyze the effectiveness of the proposed method, we make a new dataset with sparse point cloud data, i.e. Sparse-SemanticKITTI dataset, based on public SemanticKITTI dataset. Furthermore, in experiment part, we prove the research significance. Moreover, we compare the segmentation results between classical methods to the proposed one based on point cloud data in SemanticKITTI, Sparse-SemanticKITTI and Semantic3D dataset, respectively. The effectiveness of the proposed one is obvious. Finally, the model complexity is analyzed. In sum, we provide a sparse point cloud semantic segmentation method to balance the cost and the effect.
Keywords