MATNet: Semantic segmentation of 3D point clouds with multiscale adaptive transformer

被引:1
|
作者
Zheng, Yufei [1 ]
Lu, Jian [1 ]
Chen, Xiaogai [1 ]
Zhang, Kaibing [1 ]
Zhou, Jian [1 ]
机构
[1] Xian Polytech Univ, Sch Elect & Informat, Xian 710600, Peoples R China
基金
中国国家自然科学基金;
关键词
3D point cloud; Semantic segmentation; Multiscale; Selfattention; Transformer;
D O I
10.1016/j.compeleceng.2024.109526
中图分类号
TP3 [计算技术、计算机技术];
学科分类号
0812 ;
摘要
In recent years, the Transformer model has made significant progress in semantic segmentation tasks. However, existing self-attention mechanisms perform well in capturing remote dependencies and global features, but ignore local area information in point cloud data and have limitations in dealing with multi-scale features. To address this problem, this paper introduces a multiscale self-attention fusion (MSA) module, which adaptively fuses features within different scale neighborhoods and learns global contextual features by connecting local neighborhoods. Then, the multiscale channel aggregation module (MCA)is used to perform deep point-by-point and point-by-point convolution of the point cloud channel, aggregating channel features at multiple scales to extract more accurate local feature information. Finally, in this study, the multiscale adaptive fusion (MSA) module and the multiscale channel aggregation (MCA) module form a sequential network structure that adaptively and dynamically adjusts different scales of point cloud objects to enhance the perception of objects of different sizes for better segmentation performance. By testing and validating the model on the publicly available S3DIS Area 5 dataset and the ScantNetV2 dataset, the model achieves mIoU index values of 71.9% and 72.3%, respectively, which demonstrates the effectiveness and superiority of the proposed method. Code will be made publicly available at https://github.com/Cocoyufei/MAT/tree/master.
引用
收藏
页数:15
相关论文
共 50 条
  • [21] AUTOMATED TRAINING DATA CREATION FOR SEMANTIC SEGMENTATION OF 3D POINT CLOUDS
    De Geyter, Sam
    Bassier, Maarten
    Vergauwen, Maarten
    MEASUREMENT, VISUALISATION AND PROCESSING IN BIM FOR DESIGN AND CONSTRUCTION MANAGEMENT II, 2022, 46-5 (W1): : 59 - 67
  • [22] MHNet: Multiscale Hierarchical Network for 3D Point Cloud Semantic Segmentation
    Liang, Xiaoli
    Fu, Zhongliang
    IEEE ACCESS, 2019, 7 : 173999 - 174012
  • [23] DPRNet: Deep 3D Point Based Residual Network for Semantic Segmentation and Classification of 3D Point Clouds
    Arshad, Saira
    Shahzad, Muhammad
    Riaz, Qaiser
    Fraz, Muhammad Moazam
    IEEE ACCESS, 2019, 7 : 68892 - 68904
  • [24] Joint Semantic and Instance Segmentation in 3D Point Cloud Based on Transformer
    Liu, Suyi
    Wu, Chengdong
    Xu, Fang
    Wang, Juxiang
    Chi, Jianning
    Yu, Xiaosheng
    2023 35TH CHINESE CONTROL AND DECISION CONFERENCE, CCDC, 2023, : 4074 - 4080
  • [25] On the Segmentation of 3D LIDAR Point Clouds
    Douillard, B.
    Underwood, J.
    Kuntz, N.
    Vlaskine, V.
    Quadros, A.
    Morton, P.
    Frenkel, A.
    2011 IEEE INTERNATIONAL CONFERENCE ON ROBOTICS AND AUTOMATION (ICRA), 2011,
  • [26] Defect segmentation with local embedding in industrial 3D point clouds based on transformer
    Jing, Junfeng
    Wang, Huaqing
    MEASUREMENT SCIENCE AND TECHNOLOGY, 2024, 35 (03)
  • [27] Semantic Segmentation Networks of 3D Point Clouds for RGB-D Indoor Scenes
    Wang, Ya
    Zell, Andreas
    TWELFTH INTERNATIONAL CONFERENCE ON MACHINE VISION (ICMV 2019), 2020, 11433
  • [28] SEMANTIC ENRICHMENT OF 3D POINT CLOUDS USING 2D IMAGE SEGMENTATION
    Rai, A.
    Srivastava, N.
    Khoshelham, K.
    Jain, K.
    GEOSPATIAL WEEK 2023, VOL. 48-1, 2023, : 1659 - 1666
  • [29] Convolutional point transformer for semantic segmentation of sewer sonar point clouds
    Li, Chen
    Li, Hanlin
    Chen, Ke
    ENGINEERING APPLICATIONS OF ARTIFICIAL INTELLIGENCE, 2024, 138
  • [30] SAT3D: Slot Attention Transformer for 3D Point Cloud Semantic Segmentation
    Ibrahim, Muhammad
    Akhtar, Naveed
    Anwar, Saeed
    Mian, Ajmal
    IEEE TRANSACTIONS ON INTELLIGENT TRANSPORTATION SYSTEMS, 2023, 24 (05) : 5456 - 5466