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

摘要：

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

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

所属期刊栏目：40

分类号：TP391.4(计算技术、计算机技术)

资助基金：国家自然科学基金资助项目;山东省高等学校科技计划项目;山东省自然科学基金资助项目;山东省高等学校青年创新团队发展计划项目

在线出版日期：2020-06-02（万方平台首次上网日期，不代表论文的发表时间）

页数：共6页

页码：1309-1314