基于深度学习模型融合的铸件缺陷自动检测

摘要：

针对目前铸件缺陷检测漏检率高的问题,提出一种基于深度学习模型融合的铸件缺陷检测方法.首先对Faster RCNN网络进行改进,利用特征金字塔结构改进特征提取网络模块,实现多尺度的特征融合,完成铸件缺陷的特征提取;然后,基于ROI Align对网络中的ROI池化层进行改进,将IOU分数引入NMS算法判定过程;再将改进后的网络与Cascade RCNN以及YOLOv3进行融合;最后进行实验研究,验证了融合模型能够有效降低铸件缺陷的漏检率.实验结果表明,将感兴趣区域池化改进后,在Faster RCNN模型中的缺陷召回率提升了 1.73％,在本文网络模型中的缺陷召回率提升了 4.08％;采用模型融合的方法在不考虑分类准确度的情况下,整个模型的缺陷识别率达到95.71％,与单个模型相比,在保证铸件缺陷检测准确率的同时,提高了缺陷检测的召回率,满足了工业应用的要求.

关键词：缺陷检测、深度学习、模型融合、铸件

所属期刊栏目：42

分类号：TH164;TG245;TP391.41

资助基金：重庆市技术创新与应用发展专项;国家重大科学仪器设备开发专项

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

页数：共10页

页码：150-159