基于改进YOLOv5的陶瓷表面缺陷检测算法  

Ceramic Surface Defect Detection Algorithm Based on Improved YOLOv5

作　　者：潘金晶 曾成 张晶 李再勇 耿雪娜 PAN Jinjing;ZENG Cheng;ZHANG Jing;LI Zaiyong;GENG Xuena(CYG Vision Technology(Zhuhai)Co.,Ltd.,Zhuhai 519085,China;School of Computer Science and Technology,Changchun University of Science and Technology,Changchun 130013,China)

机构地区：[1]长园视觉科技(珠海)有限公司,广东珠海519085 [2]长春理工大学计算机科学技术学院,吉林长春130013

出　　处：《现代信息科技》2024年第13期70-75,共6页Modern Information Technology

基　　金：国家自然科学基金青年科学基金(62206028)。

摘　　要：提出一种陶瓷表面缺陷检测算法YOLOv5-G。该算法在YOLOv5框架的基础上,将全局注意力机制(GAM)引入主干和颈部网络中,该机制能够在减少信息弥散的情况下放大全局交互特征,增强了网络的特征表达能力;使用α-CIoU作为改进算法的边界框回归损失函数,自适应地向上加权高IoU对象的损失和梯度,使得模型可以更加关注IoU高的目标,从而帮助提高定位精度。在工业相机成像的陶瓷表面缺陷数据集上进行测试,结果表明,与YOLOv5模型相比,基于α-CIoU的YOLOv5模型的平均精度均值(mAP50)和召回率(Recall)分别提升了2.3%、4.6%;改进算法的平均精度均值(mAP50)、精确率(Precision)、召回率(Recall)分别提升了3.9%、1.7%、5.1%。It propose a ceramic surface defect detection algorithm YOLOv5-G.This algorithm introduces the Global Attention Mechanism(GAM)into the backbone and neck networks based on the YOLOv5 framework.This mechanism can amplify the global interaction features while reducing the information dispersion,and enhances the feature expression capability of the network.Takingα-CIoU as the bounding box regression loss function of the improved algorithm,the loss and gradient of high IoU objects are adaptively weighted upward,so that the model can pay more attention to the targets with high IoU,thus helping to improve the positioning accuracy.The test is carried out on the ceramic surface defect data set of the industrial camera imaging,and the results show that compared with the YOLOv5 model,the Mean Average Precision(mAP50)and Recall rate of the YOLOv5-G model based onα-CIoU are increased by 2.3%and 4.6%respectively.The Mean Average Precision(mAP50),Precision and Recall of the improved algorithm are all significantly improved by 3.9%,1.7%and 5.1%respectively.

关 键 词：陶瓷表面缺陷 全局注意力机制 α-IoU YOLOv5 

分 类 号：TP183[自动化与计算机技术—控制理论与控制工程] TP391.4[自动化与计算机技术—控制科学与工程]