《计算机应用》唯一官方网站

• •    下一篇

基于自适应动态图卷积和无参注意力的点云分类分割方法

李维刚,李歆怡,王永强,赵云涛   

  1. 武汉科技大学
  • 收稿日期:2024-06-25 修回日期:2024-08-30 发布日期:2024-10-29 出版日期:2024-10-29
  • 通讯作者: 李歆怡
  • 基金资助:
    湖北省重点研发计划

Point cloud classification and segmentation method based on adaptive dynamic graph convolution and parameter-free attention

  • Received:2024-06-25 Revised:2024-08-30 Online:2024-10-29 Published:2024-10-29

摘要: 针对传统卷积在处理点云时难以精确提取邻域特征信息和有效融合上下文信息的问题,提出了一种基于自适应动态图卷积和无参注意力的点云分类分割方法 。首先,通过自适应动态图卷积模块(ADGC)学习不同邻域的特征信息,生成自适应卷积核、并更新边缘特征,从而精确提取到点云的局部邻域特征;其次,设计了残差结构学习点云的空间位置信息,以便精确捕获点对之间的几何结构,更好保留和提取细节特征;最后,为了更好关注和提取局部几何特征,将无参注意力模块(PFA)与卷积操作相结合,增强了邻域之间的联系,提升了模型的上下文感知能力。实验结果表明,与PointNet相比,本文方法在多种任务上具有显著优势:分类任务的总体精度(OA)提升了4.6个百分点,部件分割任务的平均实例交并比(mIoU)提升了2.3个百分点,语义分割任务的mIoU提升了24.6个百分点。本文方法进一步提高了对复杂几何结构的理解和表征能力,从而在各种任务中实现更精确的特征提取和实验性能。

关键词: 点云, 分类分割, 自适应, 动态图卷积, 注意力机制

Abstract: To address the challenges of traditional convolution in accurately extracting neighborhood feature information and effectively integrating contextual information in point cloud processing, a Point Cloud Classification and Segmentation Method Based on Adaptive Dynamic Graph Convolution and Parameter-Free Attention was proposed. Firstly, the Adaptive Dynamic Graph Convolution module was used to learn the feature information of different neighborhoods, generate the convolution kernels, and update the edge features, thereby to accurately extract the local neighborhood features of the point cloud. Then, a residual structure was designed to learn the spatial position information of the point cloud, so as to accurately capture the geometric structures between the point pairs, and better retain and extract the detailed features. Finally, in order to better pay attention to and extract local geometric features, the Parameter-Free Attention (PFA) module was combined with convolution operation to enhance the connection between neighbors and improved the context-aware ability of the model. Experimental results show that, compared to the PointNet, the proposed method has significant significant advantages across various tasks: an increase of 4.6 percentage points in Overall Accuracy (OA) for classification tasks, a 2.3 percentage point increase in mean Intersection over Union (mIoU) for part segmentation tasks, and a 24.6 percentage point increase in mIoU for semantic segmentation tasks. The proposed method further improves the understanding and characterization of complex geometries, resulting in more accurate feature extraction and experimental performance in a variety of tasks.

Key words: Keywords: point cloud, classification and segmentation, adaptive, dynamic graph convolution, attention mechanism

中图分类号: