基于VGG-CapsNet的零件手绘草图识别模型

作者：杨钟亮; 黄瑞红; 陈育苗; 张凇; 毛新华

来源：计算机辅助设计与图形学学报, 2021, 33(11): 1677-1687.

• 手绘草图
• 零件识别
• 胶囊网络
• 深度学习

针对概念设计阶段现有CAD系统难以通过手绘草图准确匹配对应的零件的问题,提出预训练网络(VGG)与胶囊网络(CapsNet)相结合的零件手绘草图识别模型(VGG-CapsNet).招募了5名设计师绘制零件草图,构建包含标准件和非标件在内的23类零件手绘草图数据集;设计了组间实验与组内实验,分别构建基于VGG-CapsNet的零件手绘草图识别模型,并与rVGG-13模型、rCNN-13模型的识别结果进行比较.实验结果表明, VGG-CapsNet模型在组间和组内实验中的平均准确率均高于其他2种模型,为零件设计知识的检索与重用提供技术支持.

• 单位
  北京中丽制机工程技术有限公司; 华东理工大学; 东华大学