Point Cloud Segmentation Network Based on Attention Mechanism and Dual Graph Convolution

被引:1
|
作者
Yang, Xiaowen [1 ,2 ,3 ]
Wen, Yanghui [1 ,2 ,3 ]
Jiao, Shichao [1 ,2 ,3 ]
Zhao, Rong [1 ,2 ,3 ]
Han, Xie [1 ,2 ,3 ]
He, Ligang [4 ]
机构
[1] North Univ China, Sch Comp Sci & Technol, Taiyuan 030051, Peoples R China
[2] Shanxi Key Lab Machine Vis & Virtual Real, Taiyuan 030051, Peoples R China
[3] Shanxi Prov Vis Informat Proc & Intelligent Robot, Taiyuan 030051, Peoples R China
[4] Univ Warwick, Dept Comp Sci, Coventry CV4 7AL, England
基金
中国国家自然科学基金;
关键词
point cloud; semantic segmentation; dynamic graph convolution; dual graph convolution; attention mechanism; 3D SEMANTIC SEGMENTATION;
D O I
10.3390/electronics12244991
中图分类号
TP [自动化技术、计算机技术];
学科分类号
0812 ;
摘要
To overcome the limitations of inadequate local feature representation and the underutilization of global information in dynamic graph convolutions, we propose a network that combines attention mechanisms with dual graph convolutions. Firstly, we construct a static graph based on the dynamic graph using the K-nearest neighbors algorithm and geometric distances of point clouds. This integration of dynamic and static graphs forms a dual graph structure, compensating for the underutilization of geometric positional relationships in the dynamic graph. Next, edge convolutions are applied to extract edge features from the dual graph structure. To further enhance the capturing ability of local features, we employ attention pooling, which combines max pooling and average pooling operations. Secondly, we introduce channel attention modules and spatial self-attention modules to improve the representation ability of global features and enhance semantic segmentation accuracy in our network. Experimental results on the S3DIS dataset demonstrate that compared to dynamic graph convolution alone, our proposed approach effectively utilizes both semantic and geometric relationships between point clouds using dual graph convolutions while addressing limitations related to insufficient local feature extraction. The introduction of attention mechanisms helps mitigate underutilization issues with global information, resulting in significant improvements in model performance.
引用
收藏
页数:15
相关论文
共 50 条
  • [41] Text Classification Based on Graph Convolution Neural Network and Attention Mechanism
    Zhai, Sheping
    Zhang, Wenqing
    Cheng, Dabao
    Bai, Xiaoxia
    ACM International Conference Proceeding Series, 2022, : 137 - 142
  • [42] Point Cloud Classification and Segmentation Model Based on Graph Convolution and 3D Direction Convolution
    Lan, Hong
    Chen, Hao
    Zhang, Pufen
    Computer Engineering and Applications, 2023, 59 (08) : 182 - 191
  • [43] Warship Image Segmentation Network based on Void Convolution and Attention Mechanism
    Chen, Hongyan
    Yin, Donghui
    Qin, Zhen
    Wu, Qiong
    2024 9TH INTERNATIONAL CONFERENCE ON SIGNAL AND IMAGE PROCESSING, ICSIP, 2024, : 785 - 789
  • [44] DPPCN: density and position-based point convolution network for point cloud segmentation
    Li, Yaqian
    Zhang, Ze
    Li, Haibin
    Zhang, Wenming
    PATTERN ANALYSIS AND APPLICATIONS, 2025, 28 (02)
  • [45] Medical nucleus image segmentation network based on convolution and attention mechanism
    Zhi P.
    Deng J.
    Zhong Z.
    Shengwu Yixue Gongchengxue Zazhi/Journal of Biomedical Engineering, 2022, 39 (04): : 730 - 739
  • [46] Graph Regulation Network for Point Cloud Segmentation
    Du, Zijin
    Liang, Jianqing
    Liang, Jiye
    Yao, Kaixuan
    Cao, Feilong
    IEEE TRANSACTIONS ON PATTERN ANALYSIS AND MACHINE INTELLIGENCE, 2024, 46 (12) : 7940 - 7955
  • [47] A Dual Attention Neural Network for Airborne LiDAR Point Cloud Semantic Segmentation
    Zhang, Ka
    Ye, Longjie
    Xiao, Wen
    Sheng, Yehua
    Zhang, Shan
    Tao, Xia
    Zhou, Yaqin
    IEEE TRANSACTIONS ON GEOSCIENCE AND REMOTE SENSING, 2022, 60
  • [48] DGPolarNet: Dynamic Graph Convolution Network for LiDAR Point Cloud Semantic Segmentation on Polar BEV
    Song, Wei
    Liu, Zhen
    Guo, Ying
    Sun, Su
    Zu, Guidong
    Li, Maozhen
    REMOTE SENSING, 2022, 14 (15)
  • [49] Point Cloud Upsampling Network Incorporating Dynamic Graph Convolution and Multi-Head Attention
    Yang, Xiaoping
    Chen, Fei
    Li, Zhenhua
    Liu, Guanghui
    INFORMATION TECHNOLOGY AND CONTROL, 2024, 53 (04):
  • [50] Lightweight Fundus Image Segmentation Network Combining Structured Convolution and Dual Attention Mechanism
    Wang, Huadeng
    Liu, Jin
    Li, Bingbing
    Pan, Xipeng
    Liu, Zhenbing
    Lan, Rushi
    Luo, Xiaonan
    Jisuanji Fuzhu Sheji Yu Tuxingxue Xuebao/Journal of Computer-Aided Design and Computer Graphics, 2024, 36 (05): : 760 - 774