激光与光电子学进展, 2019, 56 (21): 211004, 网络出版: 2019-11-02
基于图卷积网络的深度学习点云分类模型 下载: 1787次
Deep Learning Model for Point Cloud Classification Based on Graph Convolutional Network
图像处理 三维点云分类 深度学习 图卷积网络 k近邻图 image processing three-dimensional point cloud classification deep learning graph convolutional network k-nearest neighbor graph
摘要
PointNet是三维点云分类中具有代表性的研究成果,该模型开创性地利用深度学习模型对点云进行分类,取得了较好的效果。但是PointNet模型只考虑点云的全局特征而忽略每个点的局部信息,为弥补这个缺陷,提出基于图卷积网络的点云分类模型。在PointNet模型中插入一个kNN graph层,通过在点云空间构造k近邻图,利用图结构有效地获取点云的局部信息,从而提高整体点云分类准确率。分类实验在ModelNet40数据集上进行,对比不同近邻值k对输出精度的影响,结果表明在k取20时,分类准确率最高,达到了93.2%,比PointNet高4.0%。
Abstract
PointNet is one of the representative research results obtained from three-dimensional point cloud classification, which innovatively employs a deep learning model for point cloud classification and achieves good results. However, PointNet does not capture local information of each point, and it considers only the global features of point clouds. Herein, we propose a model for point cloud classification based on graph convolutional networks to solve this problem, in which a k-nearest neighbor (kNN) graph layer is designed and plugged into a PointNet model. The local information of point clouds can be effectively obtained by constructing the kNN graph layer in the point cloud space, which can improve the accuracy of point cloud classification. The point cloud classification experiment is conducted on the ModelNet40 dataset, and the effects of the different neighbor values of k on the output accuracy are compared. The results demonstrate that the highest classification accuracy is achieved when k is 20, reaching 93.2%, which is 4.0% higher than that of PointNet.
王旭娇, 马杰, 王楠楠, 马鹏飞, 杨立闯. 基于图卷积网络的深度学习点云分类模型[J]. 激光与光电子学进展, 2019, 56(21): 211004. Xujiao Wang, Jie Ma, Nannan Wang, Pengfei Ma, Lichaung Yang. Deep Learning Model for Point Cloud Classification Based on Graph Convolutional Network[J]. Laser & Optoelectronics Progress, 2019, 56(21): 211004.