高级检索
白静, 司庆龙, 秦飞巍. 基于卷积神经网络和投票机制的三维模型分类与检索[J]. 计算机辅助设计与图形学学报, 2019, 31(2): 303-314. DOI: 10.3724/SP.J.1089.2019.17160
引用本文: 白静, 司庆龙, 秦飞巍. 基于卷积神经网络和投票机制的三维模型分类与检索[J]. 计算机辅助设计与图形学学报, 2019, 31(2): 303-314. DOI: 10.3724/SP.J.1089.2019.17160
Bai Jing, Si Qinglong, Qin Feiwei. 3D Model Classification and Retrieval Based on CNN and Voting Scheme[J]. Journal of Computer-Aided Design & Computer Graphics, 2019, 31(2): 303-314. DOI: 10.3724/SP.J.1089.2019.17160
Citation: Bai Jing, Si Qinglong, Qin Feiwei. 3D Model Classification and Retrieval Based on CNN and Voting Scheme[J]. Journal of Computer-Aided Design & Computer Graphics, 2019, 31(2): 303-314. DOI: 10.3724/SP.J.1089.2019.17160

基于卷积神经网络和投票机制的三维模型分类与检索

3D Model Classification and Retrieval Based on CNN and Voting Scheme

  • 摘要: 针对现有基于深度学习的三维模型多视图分类算法利用最大池化、平均池化等像素级运算完成视图信息的融合,可能造成模型有益信息淹没和混淆的问题,提出一种基于卷积神经网络和投票机制的三维模型分类检索算法.首先将三维模型转化为一组二维视图,然后基于丰富的数字图像库ImageNet和成熟的图像深度学习模型CaffeNet完成二维视图的分类,最后利用加权投票的方式完成三维模型的分类;同时基于投票机制,提出4种三维模型距离度量算法,支持三维模型的检索.将文中算法应用于刚性三维模型库ModelNet10,ModelNet40,非刚性三维模型库SHREC10, SHREC11和SHREC15中,分类准确率分别为93.18%, 93.07%, 99.5%, 99.5%和99.4%,检索性能突出;并通过实验验证该算法的有效性.

     

    Abstract: The existing deep learning algorithms for view-based 3D model classification use pixel-level operations,such as maximum pooling and average pooling,to fuse the views’information,which may lose or overwrite the useful information of 3D models.Aiming at the problem,a 3D model classification and retrieval algorithm based on convolutional neural network and voting scheme is proposed.Firstly,each 3D model is converted to a set of 2D views.Then,those 2D views are classified based on deep learning model CaffeNet with rich digital image library ImageNet.Finally,the 3D model is classified by weighted voting.Furthermore,based on the voting scheme,four distance measurement algorithms are proposed to retrieve 3D model.Experiments on the rigid 3D model libraries ModelNet10,ModelNet40,and the non-rigid 3D model libraries SHREC10,SHREC11 and SHREC15 demonstrate the effectiveness of the proposed algorithm.The classification accuracy rates for above five libraries are 93.18%,93.07%,99.5%,99.5%and 99.4%respectively,and the retrieval performance is on par or better than state-of-the-art methods.

     

/

返回文章
返回