基于ADGCNN的三维点云特征提取方法研究

FEATURE EXTRACTION OF 3D POINT CLOUD BASED ON ADGCNN

  • 摘要: 三维点云数据具有稀疏性和不规则性,在特征提取时,现有方法未考虑特征通道的重要性差异。对此,提出一种结合注意力机制的ADGCNN网络,在EdgeConv结构中引入通道注意力模块,根据特征通道的重要程度分配不同权重,提升网络的表达能力。并且采用最大池化与平均池化拼接的方法处理点云的无序性问题,防止仅使用最大池化造成的信息损失。实验表明,在ModelNet40点云分类数据集上,ADGCNN相比于DGCNN的分类准确率由92.20%提高至93.31%,验证了ADGCNN网络的有效性。

     

    Abstract: The 3D point cloud data is sparse and irregular, and the existing methods do not consider the important difference of feature channels when feature extraction is performed. In this regard, this paper proposes an ADGCNN network combining attention mechanism. The channel attention module was introduced into EdgeConv structure, and different weights were allocated according to the importance of characteristic channels to improve the expression ability of the network. The method of maximum pooling and average pooling was used to deal with the disorder of point cloud, and to prevent the information loss caused by only maximum pooling. Experiments show that the classification accuracy of ADGCNN compared with DGCNN is improved from 92.20% to 93.31% on the ModelNet40 point cloud classification dataset, which verifies the effectiveness of the ADGCNN network.

     

/

返回文章
返回