史豪斌,王少熙,黄睿茜.融合注意力机制与神经网络的三维点云分类算法[J]. 微电子学与计算机,2023,40(5):12-19. doi: 10.19304/J.ISSN1000-7180.2022.0575
引用本文: 史豪斌,王少熙,黄睿茜.融合注意力机制与神经网络的三维点云分类算法[J]. 微电子学与计算机,2023,40(5):12-19. doi: 10.19304/J.ISSN1000-7180.2022.0575
SHI H B,WANG S X,HUANG R X. A 3D point cloud classification algorithm combining attention mechanism and neural network[J]. Microelectronics & Computer,2023,40(5):12-19. doi: 10.19304/J.ISSN1000-7180.2022.0575
Citation: SHI H B,WANG S X,HUANG R X. A 3D point cloud classification algorithm combining attention mechanism and neural network[J]. Microelectronics & Computer,2023,40(5):12-19. doi: 10.19304/J.ISSN1000-7180.2022.0575

融合注意力机制与神经网络的三维点云分类算法

A 3D point cloud classification algorithm combining attention mechanism and neural network

  • 摘要: 为了提高三维点云格式的样本分类准确率,将注意力机制与改进后的Pointnet网络相融合,对提取到的局部特征和全局特征进行加权获得对分类任务更加有效的特征,抑制相对无效的特征. 该模型首先使用Pointnet网络作为基础架构对点云样本中每一个点进行全局特征的获取,使用k近邻方法为均匀采样得到的中心点选取k个相邻点,对该点到其他相邻点的关系进行建模,作为区域的局部特征. 其次,使用 squeeze_excitation_network中的SE_block模块完成特征通道间的权重分配,为改进后的Pointnet网络加入注意力机制,使其能够提取出更加精细且更具有分辨能力的特征. 最后,通过混合池化层进行聚合,混合池化由最大池化和平均池化按照不同的比例融合,文章实验部分对于比例系数的影响进行了展示. 在保证与Pointnet实验环境设置相同的情况下,该模型在 Modelnet40 数据集上的三维物体分类结果相比 Pointnet 取得了4.1%的准确率的提升. 实验表明,文章提出的融合注意力与神经网络结合的模型能够得到更有区分度的样本特征,从而有效的提高了三维点云物体的分类准确率.

     

    Abstract: In order to improve the classification accuracy of 3D point cloud objects, the attention mechanism is integrated with the improved Pointnet network, and the extracted local features and global features are weighted to obtain more effective features for the classification task. Firstly, this model uses Pointnet network as the infrastructure to obtain global features for each point in the sample, uses K-nearest neighbor method to select k neighboring points for each point, and models the relationship between this point and other neighboring points as the local features of the region. Secondly, the SE_block module in the squeeze_excitation_network is used to allocate weights between feature channels, adding an attention mechanism to the modified Pointnet that enables finer and more discriminative feature extraction. Finally, the polymerization was carried out through the mixing pooling layer. The mixing pooling consisted of maximum pooling and average pooling in different proportions. The experimental part of the paper showed the influence of the proportion coefficient.The proposed model achieves 4.1% improvement in object classification accuracy compared with Pointnet on Modelnet40 dataset. Experiments show that the proposed model combining attention and neural network can obtain more discriminative features and effectively improve the classification accuracy of 3D point cloud objects.

     

/

返回文章
返回