摘要
针对传统零件识别方法图像特征提取鲁棒性不足,零件识别准确率较低、不能对图像进行实例分割的问题,文章提出了一种基于Mask R-CNN的零件识别方法。该方法利用卷积神经网络对零件图像进行特征提取,选取数据集中标注好的图像微调Mask R-CNN网络,以保证零件识别的准确性,并生成Mask分割掩码,对零件进行实例分割。同时,对数据集进行数据增强和划分K折交叉验证来提高模型的鲁棒性。最后通过搭建实验平台对零件进行识别,证明了该方法的有效性。
Aiming at the problem that the image feature extraction of traditional part recognition method is not robust enough, the accuracy of part recognition is low, and the image cannot be segmented by example. This paper proposes a part recognition method based on Mask R-CNN. The method uses the convolutional neural network to extract the features of the part image, selects the annotated part image from dataset to fine-tuned Mask R-CNN to guarantee the accuracy of the part recognition, and generates the segmentation mask to segment the parts. Data expansion and K-Fold Cross Validation also were used to improve the robustness of the model. Finally, the result of part recognition by building the experimental platform proves the effectiveness of the method.
作者
黄海松
魏中雨
姚立国
HUANG Hai-song;WEI Zhong-yu;YAO Li-guo(Key Laboratory of Advanced Manufacturing Technology,Ministry of Education,Guizhou University,Guiyang 550025,China)
出处
《组合机床与自动化加工技术》
北大核心
2019年第5期122-125,共4页
Modular Machine Tool & Automatic Manufacturing Technique
基金
国家自然科学基金(51865004)
贵州省科技重大专项计划(黔科合重大专项[2017]3004号)
贵州工业攻关重点项目(黔科合GZ字[2015]3009)
贵州工业攻关重点项目(黔科合GZ字[2015]3034)
贵州省教育厅项目(黔教合协同创新字[2015]02)
作者简介
黄海松(1977-),女,彝族,贵州大方人,贵州大学教授,博士生导师,研究方向为制造物联与制造大数据,(E-mail)huang_h_s@126.com;通讯作者:魏中雨(1993-),男,山东巨野人,贵州大学硕士研究生,研究方向为制造物联与制造大数据,(E-mail)weizy0120@qq.com。
