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 条
  • [1] Point cloud semantic segmentation network based on graph convolution and attention mechanism
    Yang, Nan
    Wang, Yong
    Zhang, Lei
    Jiang, Bin
    Engineering Applications of Artificial Intelligence, 2025, 141
  • [2] Deep Graph Attention Convolution Network for Point Cloud Semantic Segmentation
    Chai Yujing
    Ma Jie
    Liu Hong
    LASER & OPTOELECTRONICS PROGRESS, 2021, 58 (12)
  • [3] Graph Attention Convolution for Point Cloud Semantic Segmentation
    Wang, Lei
    Huang, Yuchun
    Hou, Yaolin
    Zhang, Shenman
    Shan, Jie
    2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019), 2019, : 10288 - 10297
  • [4] Graph Convolution Point Cloud Super-Resolution Network Based on Mixed Attention Mechanism
    Chen, Taoyi
    Qiu, Zifeng
    Zhang, Chunjie
    Bai, Huihui
    ELECTRONICS, 2023, 12 (10)
  • [5] A transfer learning-based network model integrating kernel convolution with graph attention mechanism for point cloud segmentation of livestock
    Shi, Yuanyuan
    Wang, Yuxiao
    Ling, Yin
    Wu, Zhenfang
    Lin, Junyong
    Tian, Xuhong
    Qiong, Huang
    Zhang, Sumin
    Li, Zhiying
    COMPUTERS AND ELECTRONICS IN AGRICULTURE, 2024, 225
  • [6] Dynamic Attention Graph Convolution Neural Network of Point Cloud Segmentation for Defect Detection
    Li, Yumeng
    Zhang, Ruixun
    Li, Huichao
    Shao, Xiuli
    PROCEEDINGS OF 2020 IEEE INTERNATIONAL CONFERENCE ON ARTIFICIAL INTELLIGENCE AND INFORMATION SYSTEMS (ICAIIS), 2020, : 18 - 23
  • [7] Semantic segmentation of point clouds by fusing dual attention mechanism and dynamic graph convolution
    Yang, Jun
    Zhang, Chen
    Beijing Hangkong Hangtian Daxue Xuebao/Journal of Beijing University of Aeronautics and Astronautics, 2024, 50 (10): : 2984 - 2994
  • [8] 3D point cloud classification and segmentation based on dual attention and weighted dynamic graph convolution
    Xiao, Jian
    Wang, Xiaohong
    Li, Wei
    Yang, Yifei
    Luo, Ji
    Guangxue Jingmi Gongcheng/Optics and Precision Engineering, 2024, 32 (18): : 2823 - 2835
  • [9] A Graph Aggregation Convolution and Attention Mechanism Based Semantic Segmentation Method for Sparse Lidar Point Cloud Data
    Zheng, Tong
    Chen, Jialun
    Feng, Wenbin
    Yu, Chongchong
    IEEE ACCESS, 2024, 12 : 10459 - 10469
  • [10] Local Graph Point Attention Network in Point Cloud Segmentation
    Tran, Anh-Thuan
    Le, Hoanh-Su
    Lee, Suk-Hwan
    Kwon, Ki-Ryong
    IEEE ACCESS, 2023, 11 : 33296 - 33312