Optics and Precision Engineering, Volume. 32, Issue 24, 3658(2024)

Airborne point cloud classification integrating edge convolution and global-local self-attention

Jingmin TU1... Jin YAN1, Li LI2, Jian YAO1,2, Jie LI1,* and Yanfei KANG3 |Show fewer author(s)
Author Affiliations
  • 1Hubei Collaborative Innovation Center for High-efficiency Utilization of Solar Energy, Hubei University of Technology, Wuhan430068, China
  • 2School of Remote Sensing and Information Engineering, Wuhan University, Wuhan430079, China
  • 3Wuhan Survey and Design Co., LTD., Wuhan40020,China
  • show less

    Laser point cloud classification is the basis for 3D scene understanding. In order to solve the problems of insufficient feature expression and unbalanced sample categories in the classification of large scenes of airborne point clouds, this paper proposed an airborne point cloud classification method that integrates edge convolution and global-local self-attention. Firstly, U-net was used as the network framework to integrate Point Transformer and edge convolution modules, so that the model could pay attention to complex ground object boundaries and texture information, and obtain local geometric features with better expressive ability. Secondly, a self-attention mechanism that integrates global context information and local structural features was innovatively proposed, and the global self-attention module tended to the information of the entire input sequence, while the local self-attention module pays more attention to the details of the local region. The combination of the two mechanisms enhanced the capture of long-distance dependence and local structure, and at the same time, the model could take into account the key features of a few categories, reduced the impact of sample category imbalance on the classification accuracy to a certain extent, and helped to improve the classification ability of the model for complex ground object relationships. Finally, the proposed method was verified on the public ISPRS-3D dataset and WHU-Urban3D dataset, and the experimental results show that the classification accuracy of the proposed method on the two datasets is 82.5% and 87.4%, respectively, which is better than that of the classical networks such as PointNet++ and Stratified Transformer and the ISPRS 3D official website competition network, which can effectively improve the classification accuracy of airborne point clouds.

    Keywords
    Tools

    Get Citation

    Copy Citation Text

    Jingmin TU, Jin YAN, Li LI, Jian YAO, Jie LI, Yanfei KANG. Airborne point cloud classification integrating edge convolution and global-local self-attention[J]. Optics and Precision Engineering, 2024, 32(24): 3658

    Download Citation

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

    Category:

    Received: Jul. 8, 2024

    Accepted: --

    Published Online: Mar. 11, 2025

    The Author Email: LI Jie (jielonline@hbut.edu.cn)

    DOI:10.37188/OPE.20243224.3658

    Topics