作者:张翠1,杨志清1,周茂杰2(1.桂林理工大学博文管理学院,广西桂林541006;2.桂林理工大学,广西桂林541004)
摘要:利用深度学习方法进行塑料产品缺陷检测时,存在训练样本不足,网络层次增加产生的梯度消失和梯度爆炸等问题,采用残差网络ResNet和压缩与激励网络SENet相结合,构建深度学习模型,解决梯度消失、梯度爆炸和注意力分布的问题,利用工业生产中的产品图像进行缺陷检测试验,经过2种试验结果分析,该算法有效提高了产品缺陷检测的准确率和召回率。
关键词:ResNet;SENet;缺陷检测;塑料产品;深度学习
中图分类号:TG76 文献标识码:B 文章编号:1001-2168(2020)11-0013-05
DOI:10.16787/j.cnki.1001-2168.dmi.2020.11.003
Defect detection based on double channel convolutional neural network method
ZHANG Cui1, YANG Zhi-qing1, ZHOU Mao-jie2 (1.Bowen College of Management, Guilin University of Technology, Guilin, Guangxi 541006, China; 2.Guilin University of Technology, Guilin, Guangxi 541004, China)
Abstract: In the process of plastic product defect detection by using deep learning method, there were some problems, such as insufficient training samples, gradient disappearance
and explosion caused by increasing network level, etc.. Combined ResNet with SENet to build a deep learning model, it solved the problems of gradient disappearance and explosion, and attention distribution. Images in industrial production were used to conduct defect detection experiments. Through the analysis of two experimental results, the algorithm could effectively improve the accuracy rate and recall rate of product defect detection.
Key words: ResNet; SENet; defect detection; plastic products; deep learning