-
题名改进YOLOv7的玻璃表面缺陷检测方法研究
- 1
-
-
作者
宋吕明
刘明芹
李祥宾
朱雅
王家超
-
机构
江苏海洋大学海洋工程学院
江苏海洋大学机械工程学院
连云港市拓普科技发展有限公司
-
出处
《机电工程技术》
2024年第6期209-215,共7页
-
基金
江苏省海洋资源开发研究院开放课题(JSIMR201810)。
-
文摘
针对电子产品玻璃边缘表面的电极区域与非电极区域深加工过程中产生的大划伤、小划伤、划痕、异物等缺陷,提出了一种基于改进的YOLOv7的玻璃表面缺陷小样本检测方法。首先,在主干网络中加入卷积注意力模块(Convolutional Block Attention Module, CBAM)提高了通道注意力与空间注意力,解决了玻璃表面缺陷面积较小、在图像中分布差异较大、提高卷积神经网络在缺陷区域学习鲁棒性的特征表示的问题。其次,考虑到工业生产过程中缺陷样本较少、样本量不均衡,采用随机高斯噪声、Mixup、随机填充图像和随机拼接等图像增强方法,将样本进行扩充,并使样本均衡化。最后,将增加一个预检测头用于细长且轻浅的划痕检测,结合其他3个预测头,四预测头结构可以有效缓解过大差异对象带来的尺度方差引起的负面影响。实验结果表明,改进的YOLOv7算法相较于原始算法,平均精度提高了6.15%(mAP),检测效果优于当前YOLOv7网络,在一定程度上提高了工业生产过程中玻璃表面缺陷的小样本检测精度。
-
关键词
YOLOv7
玻璃表面缺陷检测
卷积注意力模块
图像增强
四预测头结构
-
Keywords
YOLOv7
glass surface defect detection
convolutional attention module
image enhancement
four prediction head structure
-
分类号
TP391.4
[自动化与计算机技术—计算机应用技术]
-