计算机应用 ›› 2020, Vol. 40 ›› Issue (5): 1309-1314.DOI: 10.11772/j.issn.1001-9081.2019101750

• 人工智能 • 上一篇    下一篇

基于胶囊网络的三维模型识别

曹小威, 曲志坚, 徐玲玲, 刘晓红   

  1. 山东理工大学 计算机科学与技术学院,山东 淄博 255000
  • 收稿日期:2019-10-16 修回日期:2019-12-13 出版日期:2020-05-10 发布日期:2020-05-15
  • 通讯作者: 刘晓红(1968—)
  • 作者简介:曹小威(1990—),男,河南项城人,硕士研究生,主要研究方向:三维模型识别; 曲志坚(1980—),男,山东青岛人,副教授,博士,主要研究方向:机器学习; 徐玲玲(1997—),女,山东菏泽人,硕士研究生,主要研究方向:推荐算法; 刘晓红(1968—),女,黑龙江哈尔滨人,教授,硕士,主要研究方向:图像处理。
  • 基金资助:

    国家自然科学基金资助项目(61473179):山东省高等学校科技计划项目(J16LN20); 山东省自然科学基金资助项目(ZR2016FM18,ZR2017LF004);山东省高等学校青年创新团队发展计划项目(2019KJN048)。

3D model recognition based on capsule network

CAO Xiaowei, QU Zhijian, XU Lingling, LIU Xiaohong   

  1. School of Computer Science and Technology, Shandong University of Technology, Zibo Shandong 250000, China
  • Received:2019-10-16 Revised:2019-12-13 Online:2020-05-10 Published:2020-05-15
  • Contact: LIU Xiaohong, born in 1968, M. S., professor. Her research interests include image processing.
  • About author:CAO Xiaowei, born in 1990, M. S. candidate. His research interests include three-dimensional model recognition.QU Zhijian, born in 1980, Ph. D., associate professor. His research interests include machine learning.XU Lingling, born in 1997, M. S. candidate. Her research interests include recommendation algorithm.LIU Xiaohong, born in 1968, M. S., professor. Her research interests include image processing.
  • Supported by:

    This work is partially supported by National Natural Science Foundation of China (61473179), the Shandong Higher Education Science and Technology Program (J16LN20), the Shandong Provincial Natural Science Foundation (ZR2016FM18, ZR2017LF004), the Youth Innovation Team Development Plan of Shandong Province Higher Education (2019KJN048).

摘要:

为解决传统卷积神经网络中大量池化层的引入导致特征信息丢失的问题,依据胶囊网络(CapsNet)使用向量神经元保存特征空间信息的特性,提出了一种用以识别三维模型的网络模型3DSPNCapsNet。使用新的网络结构,提取更具代表性的特征的同时降低了模型复杂度,并提出基于动态路由(DR)算法的DRL算法来优化胶囊权重的迭代计算过程。在ModelNet10上的实验结果表明,相比3DCapsNet以及VoxNet,该网络取得了更好的识别效果,在原始测试集上3DSPNCapsNet的平均识别准确率达到95%,同时验证了该网络对旋转三维模型的识别能力。适当扩展旋转训练集之后,所提网络对各角度旋转模型的平均识别率达到81%。实验结果表明,3DSPNCapsNet对三维模型及其旋转具有良好的识别能力。

关键词: 胶囊网络, 动态路由算法, 池化, 三维模型识别, 旋转

Abstract:

In order to solve the problem of feature information loss caused by the introduction of a large number of pooling layers in traditional convolutional neural networks, based on the feature of Capsule Network (CapsNet)——using vector neurons to save feature space information, a network model 3DSPNCapsNet (3D Small Pooling No dense Capsule Network) was proposed for recognizing 3D models. Using the new network structure, more representative features were extracted while the model complexity was reduced. And based on Dynamic Routing (DR) algorithm, Dynamic Routing-based algorithm with Length information (DRL) algorithm was proposed to optimize the iterative calculation process of capsule weights. Experimental results on ModelNet10 show that compared with 3DCapsNet (3D Capsule Network) and VoxNet, the proposed network achieves better recognition results, and has the average recognition accuracy on the original test set reached 95%. At the same time, the recognition ability of the network for the rotation 3D models was verified. After the rotation training set is appropriately extended, the average recognition rate of the proposed network for rotation models of different angles reaches 81%. The experimental results show that 3DSPNCapsNet has a good ability to recognize 3D models and their rotations.

Key words: Capsule Network (CapsNet), dynamic routing algorithm, pooling, 3D model recognition, rotation

中图分类号: