Dual attention-based deep learning network for multi-class object semantic segmentation of tunnel point clouds

Abstract

Aiming to automatically segment multi-class objects on the tunnel point cloud, a deep learning network named dual attention-based point cloud network (DAPCNet) is developed in this paper to act on point clouds for segmentation. In the developed model, data normalization and feature aggregation are first processed to eliminate data discrepancies and enhance local features, after which the processed data are input into the built network layers based on the encoder-decoder architecture coupled with an improved 3D dual attention module to extract and learn features. Furthermore, a custom loss function called Facal Cross-Entropy (“FacalCE”) is designed to enhance the model's ability to extract and learn features while addressing imbalanced data distribution. To validate the effectiveness and feasibility of the developed model, a dataset of tunnel point clouds collected from a real engineering project in China is employed. The experimental results indicate that (1) the developed model has excellent performance with Mean Intersection over Union (MIoU) of 0.8597, (2) the improved 3D dual attention module and “FacalCE” contribute to the model performance, respectively, and (3) the developed model is superior to other state-of-the-art methods, such as PointNet and DGCNN. In summary, the DAPCNet model exhibits exceptional performance, offering effective and accurate results for segmenting multi-class objects within tunnel point clouds.