激光与光电子学进展, 2021, 58 (4): 0410016, 网络出版: 2021-02-24   

基于改进生成对抗网络和MobileNetV3的带钢缺陷分类 下载: 1008次

Strip Defect Classification Based on Improved Generative Adversarial Networks and MobileNetV3
作者单位
1 西安工程大学机电工程学院, 陕西 西安 710048
2 绍兴市柯桥区西纺纺织产业创新研究院, 浙江 绍兴 312030
3 西安工程大学计算机科学学院, 陕西 西安 710048
摘要
针对数据集样本数量较少会影响深度学习检测效果的问题,提出了一种基于改进生成对抗网络和MobileNetV3的带钢缺陷分类方法。首先,引入生成对抗网络并对生成器和判别器进行改进,解决了类别错乱问题并实现了带钢缺陷数据集的扩充。然后,对轻量级图像分类网络MobileNetV3进行改进。最后,在扩充后的数据集上训练,实现了带钢缺陷的分类。实验结果表明,改进的生成对抗网络可生成比较真实的带钢缺陷图像,同时解决深度学习中样本不足的问题;且改进的MobileNetV3参数量是原有参数量的1/14左右,准确率为94.67%,比改进前提高了2.62个百分点,可在工业现场对带钢缺陷进行实时准确的分类。
Abstract
Aiming at the problem that the small number of samples in the dataset will affect the effect of deep learning detection, a strip defect classification method based on improved generative adversarial networks and MobileNetV3 is proposed in this paper. First, a generative adversarial network is introduced, and the generator and discriminator are improved to solve the problem of category confusion and realize the expansion of the strip defect data set. Then, the lightweight image classification network MobileNetV3 is improved. Finally, it is trained on the expanded data set to realize the classification of strip defects. Experimental results show that the improved generative adversarial network can generate more real strip steel defect images and solve the problem of insufficient samples in deep learning. And the parameter amount of the improved MobileNetV3 is about 1/14 of that before improvement, and the accuracy is 94.67%, which is 2.62 percentage points higher than that before improvement. It can be used for accurate and real-time classification of strip steel defects in industrial field.

常江, 管声启, 师红宇, 胡璐萍, 倪奕棋. 基于改进生成对抗网络和MobileNetV3的带钢缺陷分类[J]. 激光与光电子学进展, 2021, 58(4): 0410016. Jiang Chang, Shengqi Guan, Hongyu Shi, Luping Hu, Yiqi Ni. Strip Defect Classification Based on Improved Generative Adversarial Networks and MobileNetV3[J]. Laser & Optoelectronics Progress, 2021, 58(4): 0410016.

本文已被 6 篇论文引用
被引统计数据来源于中国光学期刊网
引用该论文: TXT   |   EndNote

相关论文

加载中...

关于本站 Cookie 的使用提示

中国光学期刊网使用基于 cookie 的技术来更好地为您提供各项服务,点击此处了解我们的隐私策略。 如您需继续使用本网站,请您授权我们使用本地 cookie 来保存部分信息。
全站搜索
您最值得信赖的光电行业旗舰网络服务平台!