Journal of Computer Applications ›› 2025, Vol. 45 ›› Issue (6): 1980-1986.DOI: 10.11772/j.issn.1001-9081.2024060878

• Multimedia computing and computer simulation • Previous Articles    

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

Weigang LI1,2, Xinyi LI1(), Yongqiang WANG1, Yuntao ZHAO1   

  1. 1.School of Information Science and Engineering,Wuhan University of Science and Technology,Wuhan Hubei 430081,China
    2.Engineering Research Center for Metallurgical Automation and Measurement Technology of Ministry of Education (Wuhan University of Science and Technology),Wuhan Hubei 430081,China
  • Received:2024-06-27 Revised:2024-08-30 Accepted:2024-09-06 Online:2024-10-29 Published:2025-06-10
  • Contact: Xinyi LI
  • About author:LI Weigang, born in 1977, Ph. D., professor. His research interests include industrial process control, artificial intelligence, machine learning algorithms.
    LI Xinyi, born in 2000, M. S. candidate. Her research interests include point cloud data processing, deep learning, pattern recognition.
    WANG Yongqiang, born in 1994, Ph. D. candidate. His research interests include intelligent processing of point cloud.
    ZHAO Yuntao, born in 1982, Ph. D., professor. His research interests include robot dynamics, 3D machine vision.
  • Supported by:
    Hubei Provincial Science and Technology Talent Service Enterprise Project(202400288)

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

李维刚1,2, 李歆怡1(), 王永强1, 赵云涛1   

  1. 1.武汉科技大学 信息科学与工程学院,武汉 430081
    2.冶金自动化与检测技术教育部工程研究中心(武汉科技大学),武汉 430081
  • 通讯作者: 李歆怡
  • 作者简介:李维刚(1977—),男,湖北咸宁人,教授,博士,主要研究方向:工业过程控制、人工智能、机器学习算法
    李歆怡(2000 —),女,湖北武汉人,硕士研究生,主要研究方向:点云数据处理、深度学习、模式识别 lixinyi_myself@163.com
    王永强(1994—),男,河南许昌人,博士研究生,主要研究方向:点云智能处理
    赵云涛(1982—),男,内蒙古赤峰人,教授,博士,主要研究方向:机器人动力学、三维机器视觉。
  • 基金资助:
    湖北省科技人才服务企业项目(202400288)

Abstract:

To address the challenges of traditional convolution in extracting neighborhood feature information accurately and integrating contextual information effectively 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 (ADGC) was used to learn feature information of different neighborhoods, generate the adaptive convolution kernels, and update the edge features, thereby extracting local neighborhood features of the point cloud accurately. Then, a residual structure was designed to learn spatial position information of the point cloud, so as to capture geometric structure between the point pairs accurately, and better retain and extract the detailed features. Finally, in order to better pay attention to and extract the local geometric features, the Parameter-Free Attention module (PFA) was combined with convolution operation to enhance connection among the neighbors and improve context-aware ability of the model. Experimental results show that compared to PointNet, the proposed method has significant advantages across various tasks. In specific, the proposed method has an increase of 4.6 percentage points in Overall Accuracy (OA) for classification tasks, an increase of 2.3 percentage points in mean Intersection over Union (mIoU) for part segmentation tasks, and an increase of 24.6 percentage points in mIoU for semantic segmentation tasks. It can be seen that the proposed method further improves the understanding and representation abilities of complex geometries, resulting in more accurate feature extraction and experimental performance in a variety of tasks.

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

摘要:

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

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

CLC Number: