摘要
为满足多数工业场景下钢板表面缺陷检测的需求,针对钢板表面缺陷检测准确率低及小目标缺陷检测率低等问题,文中提出了一种基于改进YOLOv5(You Only Look Once version 5)的钢板表面缺陷检测算法。在YOLOv5的基础上将CBAM(Convolution Block Attention Module)注意力模块嵌入到主干网络中,提高网络检测精度。加入上下文增强模块,提高了算法对小目标的检测性能。使用NWD(Normalized Wasserstein Distance)度量标准代替原YOLOv5中的IoU(Intersection over Union)度量,提高了网络对裂纹缺陷的识别精确度。实验结果表明,钢板表面缺陷检测算法对裂纹、夹杂、斑块、麻点、压入氧化铁皮、划痕6类缺陷的平均检测精度达到了88.9%,每秒帧数达到110.4 frame·s-1,其中小目标裂纹准确率达到75%。
To meet the requirements of steel plate surface defect detection in most industrial scenarios,a steel plate surface defect detection algorithm based on improved YOLOv5(You Only Look Once version 5)is proposed to solve the problems such as low detection accuracy of steel plate surface defects and failure to detect small target defects.On the basis of YOLOv5,the CBAM(Convolution Block Attention Module)is embedded into the backbone network to improve network detection accuracy.The context enhancement module is added to improve the detection performance of small targets.The NWD(Normalized Wasserstein Distance)metric is used to replace the original IoU(Intersection over Union)metric in YOLOv5,making the network more accurate in identifying crack defects.Experimental results show that the average detection accuracy of the proposed steel plate surface defect detection algorithm for six types of defects,including crack,inclusion,plaque,pitting,pressed iron oxide,scratch,reaches 88.9%,the frame rate reaches 110.4 frame·s-1,and the accuracy of small target crack reaches 75%.
作者
沈庭铅
鲁玉军
辛昊
吴涵超
汪仕男
SHEN Tingqian;LU Yujun;XIN Hao;WU Hanchao;WANG Shinan(School of Mechanical Engineering,Zhejiang Sci-Tech University,Hangzhou 310018,China)
出处
《电子科技》
2025年第6期82-88,共7页
Electronic Science and Technology
基金
浙江省重点研发项目(2022C01242)
浙江理工大学龙港研究院项目(LGYJY2021004)。
