Laser & Optoelectronics Progress, Volume. 61, Issue 8, 0811004(2024)

Point Cloud Segmentation Algorithm Based on Density Awareness and Self-Attention Mechanism

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

    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.

    Tools

    Get Citation

    Copy Citation Text

    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

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Category: Imaging Systems

    Received: Jun. 2, 2023

    Accepted: Jul. 31, 2023

    Published Online: Mar. 13, 2024

    The Author Email: Liu Yawei (1425655356@qq.com)

    DOI:10.3788/LOP231450

    Topics