• Laser & Optoelectronics Progress
  • Vol. 61, Issue 8, 0811004 (2024)
Bin Lu1、2, Yawei Liu1、2、*, Yuhang Zhang1、2, and Zhenyu Yang1、2
Author Affiliations
  • 1Department of Computer, North China Electric Power University, Baoding 071003, Hebei , China
  • 2Hebei Key Laboratory of Knowledge Computing for Energy & Power, Baoding 071003, Hebei , China
  • show less
    DOI: 10.3788/LOP231450 Cite this Article Set citation alerts
    Bin Lu, Yawei Liu, Yuhang Zhang, Zhenyu Yang. Point Cloud Segmentation Algorithm Based on Density Awareness and Self-Attention Mechanism[J]. Laser & Optoelectronics Progress, 2024, 61(8): 0811004 Copy Citation Text show less

    Abstract

    We propose a 3D point cloud semantic segmentation algorithm based on density awareness and self-attention mechanism to address the issue of insufficient utilization of inter point density information and spatial location features in existing 3D point cloud semantic segmentation algorithms. First, based on the adaptive K-Nearest Neighbor (KNN) algorithm and local density position encoding, a density awareness convolutional module is constructed to effectively extract key density information between points, enhance the depth of information expression of initial input features, and enhance the algorithm's ability to capture local features. Then, a spatial feature self-attention module is constructed to enhance the correlation between global contextual information and spatial location information based on self-attention and spatial-attention mechanisms. The global and local features are effectively aggregated to extract deeper contextual features, enhancing the segmentation performance of the algorithm. Finally, extensive experiments are conducted on the public S3DIS dataset and ScanNet dataset. The experimental results show that the mean intersection over union of our algorithm reaches 69.11% and 72.52%, respectively, shows significant improvement compared with other algorithms, verifying the proposed algorithm has good segmentation and generalization performances.
    Bin Lu, Yawei Liu, Yuhang Zhang, Zhenyu Yang. Point Cloud Segmentation Algorithm Based on Density Awareness and Self-Attention Mechanism[J]. Laser & Optoelectronics Progress, 2024, 61(8): 0811004
    Download Citation