• 首页
  • 期刊简介
  • 编委会
  • 投稿指南
  • 收录情况
  • 杂志订阅
  • 联系我们
引用本文:吴 闯,于大泳.基于深度卷积神经网络的手机玻璃盖板表面缺陷分类检测研究[J].软件工程,2021,24(12):6-10.【点击复制】
【打印本页】   【下载PDF全文】   【查看/发表评论】  【下载PDF阅读器】  
←前一篇|后一篇→ 过刊浏览
分享到: 微信 更多
基于深度卷积神经网络的手机玻璃盖板表面缺陷分类检测研究
吴 闯,于大泳
(上海理工大学机械工程学院,上海 200093)
wuch19940805@163.com; wy_ydy@163.com
摘 要: 为了实现手机玻璃盖板表面点缺陷、线缺陷和块状缺陷的分类检测,主要研究可以自动提取特征的深度卷积神经网络模型。首先针对数据缺乏问题,提出子图像划分和随机缺陷合成算法,构建了MPGC-DET数据集。为了提高模型的泛化性,以现代成熟的深度卷积神经网络模型为基础,并结合迁移学习和SE模块,搭建手机玻璃盖板表面缺陷分类检测模型。实验表明,最终分类准确率达96.40%,并且使用了Grad-CAM技术进行可视化分析,结果显示模型是根据缺陷所在区域特征进行预测的,说明没有出现过拟合现象。
关键词: 玻璃盖板;深度卷积神经网络;迁移学习;SE模块;分类检测
中图分类号: TP391    文献标识码: A
Research on Classified Detection of Surface Defects of Mobile Phone Glass Cover based on Deep Convolutional Neural Network
WU Chuang, YU Dayong
(School of Mechanical Engineering, University of Shanghai for Science and Technology, Shanghai 200093, China )
wuch19940805@163.com; wy_ydy@163.com
Abstract: In order to realize the classified detection of point defects, line defects and block defects on the glass cover surface of mobile phone, this paper proposes to study deep convolutional neural network (DCNN) model that can automatically extract features. First, to address the problem of data lack, this paper proposes to use sub-image division and random defect synthesis algorithms, and construct the MPGC-DET (Mobile Phone Glass Cover-Detection) data set. In order to improve the generalization of the model, this research takes modern and mature deep convolutional neural network model as the basis and combines transfer learning and SE modules, to build a classified detection model of mobile phone glass cover surface. Experiments show that final classification accuracy rate is 96.40%. Grad-CAM (Gradient-weighted Class Activation Mapping) technology is used for visual analysis. Results show that the proposed model performs prediction based on the characteristics of the area where the defect is located, which indicates that there is no sign of over-fitting.
Keywords: glass cover; deep convolutional neural network; transfer learning; SE module; classified detection


版权所有:软件工程杂志社
地址:辽宁省沈阳市浑南区新秀街2号 邮政编码:110179
电话:0411-84767887 传真:0411-84835089 Email:semagazine@neusoft.edu.cn
备案号:辽ICP备17007376号-1
技术支持:北京勤云科技发展有限公司

用微信扫一扫

用微信扫一扫