Computer Engineering and Applications ›› 2022, Vol. 58 ›› Issue (20): 240-246.DOI: 10.3778/j.issn.1002-8331.2103-0395

• Graphics and Image Processing • Previous Articles     Next Articles

Weighted Point Cloud Classification Network Based on Dynamic Graph Convolution

SUN Yijun, HU Hui   

  1. Global Navigation Satellite System Laboratory, School of Information Engineering, East China Jiaotong University, Nanchang 330013, China
  • Online:2022-10-15 Published:2022-10-15

基于动态图卷积的加权点云分类网络

孙一珺,胡辉   

  1. 华东交通大学 信息工程学院 GNSS实验室,南昌 330013

Abstract: Most of the traditional deep learning networks which directly deal with point clouds only consider the global features of point clouds and ignore the local features of point clouds. Dynamic graph convolution network(DGCNN) completes the local feature compensation by constructing [k]-nearest neighbor graph. However, the existing DGCNN uses simple edge features as the input of local features, and does not conduct further research on local features, In this paper, weighted dynamic graph convolution network(WDGCNN), a weighted point cloud classification network is proposed, which uses the idea of feature mosaicking to optimize the network structure to achieve multi-level feature fusion, and designs an appropriate weighting function for the edge features of [k]-nearest neighbor graph to weaken the interference of remote points and relatively strengthen the features of near points. A new pooling method is adopted to compensate for the global information loss caused by the use of maximum pooling alone. Experimental results show that compared with DGCNN, the classification accuracy of WDGCNN increases from 91.61% to 93.22% on the general point cloud classification dataset ModelNet40, which verifies the effectiveness of the new method.

Key words: image processing, three-dimensional point cloud classification, deep learning, graph convolution, [k]-nearest neighbor, pooling

摘要: 传统的直接处理点云的PointNet类深度学习网络大多只考虑了点云的全局特征而忽视了点云局部特征,动态图卷积网络DGCNN通过构建[k]近邻图完成了对局部特征的弥补。然而现有的DGCNN使用简单的边缘特征作为局部特征的输入,没有对局部特征进行更深入的研究,且仅使用最大池化处理点云无序性问题,这造成了一定的信息损失。提出加权点云分类网络WDGCNN,使用特征拼接思想优化网络结构以实现多层次特征的融合、通过对[k]近邻图构成的边缘特征设计恰当的加权函数以弱化远点的干扰,相对加强近点的特征、采用最大池化和平均池化相结合的对称函数弥补单独使用最大池化造成的全局信息损失的新方法,实现了模型优化。实验结果表明,在通用点云分类数据集ModelNet40上,WDGCNN相比于DGCNN分类准确率由91.61%达到了93.22%,验证了新方法的有效性。

关键词: 图像处理, 三维点云分类, 深度学习, 图卷积, [k]近邻, 池化