摘要
传统的直接处理点云的PointNet类深度学习网络大多只考虑了点云的全局特征而忽视了点云局部特征,动态图卷积网络DGCNN通过构建k近邻图完成了对局部特征的弥补。然而现有的DGCNN使用简单的边缘特征作为局部特征的输入,没有对局部特征进行更深入的研究,且仅使用最大池化处理点云无序性问题,这造成了一定的信息损失。提出加权点云分类网络WDGCNN,使用特征拼接思想优化网络结构以实现多层次特征的融合、通过对k近邻图构成的边缘特征设计恰当的加权函数以弱化远点的干扰,相对加强近点的特征、采用最大池化和平均池化相结合的对称函数弥补单独使用最大池化造成的全局信息损失的新方法,实现了模型优化。实验结果表明,在通用点云分类数据集ModelNet40上,WDGCNN相比于DGCNN分类准确率由91.61%达到了93.22%,验证了新方法的有效性。
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 classifica-tion dataset ModelNet40,which verifies the effectiveness of the new method.
作者
孙一珺
胡辉
SUN Yijun;HU Hui(Global Navigation Satellite System Laboratory,School of Information Engineering,East China Jiaotong University,Nanchang 330013,China)
出处
《计算机工程与应用》
CSCD
北大核心
2022年第20期240-246,共7页
Computer Engineering and Applications
基金
国家自然科学基金(61761019)。
关键词
图像处理
三维点云分类
深度学习
图卷积
K近邻
池化
image processing
three-dimensional point cloud classification
deep learning
graph convolution
k-nearest neighbor
pooling