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 条
  • [31] A Graph-Voxel Joint Convolution Neural Network for ALS Point Cloud Segmentation
    Zhang, Jinming
    Hu, Xiangyun
    Dai, Hengming
    IEEE ACCESS, 2020, 8 : 139781 - 139791
  • [32] Point Cloud Scene Segmentation Based on Dual Attention Mechanism and Multi-Scale Features
    Yu Lili
    Yu Haiyang
    He Zixin
    Chen Liangxuan
    LASER & OPTOELECTRONICS PROGRESS, 2021, 58 (24)
  • [33] Point-attention Net: a graph attention convolution network for point cloudsegmentation
    Suting Chen
    Zelin Miao
    Huaixin Chen
    Mithun Mukherjee
    Yanyan Zhang
    Applied Intelligence, 2023, 53 : 11344 - 11356
  • [34] Point-attention Net: a graph attention convolution network for point cloudsegmentation
    Chen, Suting
    Miao, Zelin
    Chen, Huaixin
    Mukherjee, Mithun
    Zhang, Yanyan
    APPLIED INTELLIGENCE, 2023, 53 (09) : 11344 - 11356
  • [35] 3D Point Cloud Classification Method Based on Multiple Attention Mechanism and Dynamic Graph Convolution
    Zhang, Yu
    Wang, Zilong
    Zhu, Yongjian
    INFORMATION TECHNOLOGY AND CONTROL, 2023, 52 (03): : 605 - 616
  • [36] Point attention network for point cloud semantic segmentation
    Dayong REN
    Zhengyi WU
    Jiawei LI
    Piaopiao YU
    Jie GUO
    Mingqiang WEI
    Yanwen GUO
    Science China(Information Sciences), 2022, 65 (09) : 99 - 112
  • [37] Point attention network for point cloud semantic segmentation
    Ren, Dayong
    Wu, Zhengyi
    Li, Jiawei
    Yu, Piaopiao
    Guo, Jie
    Wei, Mingqiang
    Guo, Yanwen
    SCIENCE CHINA-INFORMATION SCIENCES, 2022, 65 (09)
  • [38] Point attention network for point cloud semantic segmentation
    Dayong Ren
    Zhengyi Wu
    Jiawei Li
    Piaopiao Yu
    Jie Guo
    Mingqiang Wei
    Yanwen Guo
    Science China Information Sciences, 2022, 65
  • [39] Differential Graph Convolution Network for point cloud understanding
    Bai, Yun
    Li, Guanlin
    Yang, Chaozhi
    Li, Yachuan
    Xiao, Qian
    Li, Zongmin
    NEUROCOMPUTING, 2024, 597
  • [40] Associative graph convolution network for point cloud analysis
    Yang, Xi
    Yin, Xingyilang
    Wang, Nannan
    Gao, Xinbo
    PATTERN RECOGNITION, 2025, 159