基于半监督深度卷积生成对抗网络的注塑瓶表面缺陷检测模型

• 作者: 张铭 苗玉彬 许凤麟 谢源
• 来源期刊: 计算机科学
• 学科: 工学
• 关键词: 深度卷积生成对抗网络 半监督 小样本 缺陷检测 注塑瓶
• 年，卷（期）: 2020,（7）
• 所属期刊栏目: 计算机图形学&多媒体
• 页码范围: 92-96
• 页数: 5页
• 分类号: TP277
• 字数: 5017字
• 语种: 中文
• DOI: 10.11896/jsjkx.190700093

摘要

注塑瓶表面缺陷检测是注塑成型工艺流程中的重要环节,但生产中存在缺陷的注塑瓶样本数量相对匮乏,使得应用深度学习算法进行缺陷检测时容易产生过拟合现象.针对上述问题,文中提出并构建一种半监督(Semi-supervised)深度卷积生成对抗网络(Deep Convolutional Generative Adversarial Network,DCGAN)模型.该模型首先使用HSV(Hue Saturation Va-lue)颜色空间转换与大津算法(Otsu)对原始注塑瓶图像进行预处理得到训练集;然后组合学习任务,使得DCGAN的无监督判别器与注塑瓶表面缺陷检测的监督分类器共享卷积层参数,同时修改损失函数,在DCGAN模型的Wasserstein距离中加入交叉熵;最后使用Adam优化器进行模型训练.实验结果表明,该模型能够准确分辨具有缺陷的注塑瓶样本,分类准确率达到98.65％.与传统的机器学习算法以及采用数据增强的卷积神经网络模型相比,所提模型的分类准确率更高,且较好地避免了过拟合现象,能满足注塑瓶生产中表面缺陷的自动检测需求.

作者信息

序号	姓名	单位	发文数	被引次数	H指数	G指数
1	苗玉彬	上海交通大学机械与动力工程学院	50	705	14.0	26.0
2	张铭	上海交通大学机械与动力工程学院	28	59	4.0	7.0
3	许凤麟	上海交通大学机械与动力工程学院	3	3	1.0	1.0
4	谢源	上海交通大学机械与动力工程学院	4	29	2.0	4.0