基于YOLOv5s的钢铁表面缺陷检测算法  

Steel Surface Defect Detection Algorithm Based on YOLOv5s

作　　者：张瑞芳[1,2] 伏铭强 程小辉[3] ZHANG Rui-fang;FU Ming-qiang;CHEN Xiao-hui(Key Laboratory of Advanced Manufacturing and Automation Technology(Guilin University of Technology),Guilin 541006,China;College of Mechanical and Control Engineering,Guilin University of Technology,Guilin 541006,China;College of Information Science and Engineering,Guilin University of Technology,Guilin 541006,China)

机构地区：[1]广西高校先进制造与自动化技术重点实验室(桂林理工大学),桂林541006 [2]桂林理工大学机械与控制工程学院,桂林541006 [3]桂林理工大学信息科学与工程学院,桂林541006

出　　处：《科学技术与工程》2024年第23期9980-9988,共9页Science Technology and Engineering

基　　金：国家自然科学基金(61662917);广西科技计划重点研发项目(桂科AB17195042);广西中青年教师基础能力提升项目(2018KY0248,2020KY06026);广西建筑新能源与节能重点实验室项(桂科能15-J-21-1)。

摘　　要：为改善因小目标检测效果差而导致的钢铁表面缺陷检测精度差的问题,以YOLOv5s为基础,通过在主干网络添加注意力机制(attention mechanism,SE),将C2f模块代替C3模块,将双向特征金字塔网络(bidirectional feature pyramid network,BIFPN)网络代替颈部网络中的路径聚合网络(path aggregation network,PAN)网络的这三种方法来提升模型对缺陷小目标的检测能力。旨在提升检测精度并达到实时检测要求。结果表明,改进后的YOLOv5s-SCB算法在NEU-DET(northeastern university-detect)上的均值平均精度(mean average precision,mAP)值达到77.9%,在达到实时检测的前提下,相较于YOLOv5s网络提高了3.7%,与其余基于YOLOv5s改进的算法及YOLOv8相比,YOLOv5s-SCB实现了更好的检测效果。可见本文提出的钢铁表面缺陷检测算法YOLOv5S-SCB可以更好地完成钢铁表面缺陷检测。In order to improve the poor accuracy of steel surface defects caused by poor detection of small targets,on the basis of the YOLOv5s,by adding the SE(attention mechanism)in the backbone network mechanism,C2f module instead of C3 module,the BIFPN(bidirectional feature pyramid network)instead of the PAN(path aggregation network)network in the neck,these three methods were used to investigate the improvement of the ability to the defect of small target detection.It aims to improve the detection accuracy and achieve real-time detection.The results show that the mAP(mean average precision)of the improved YOLOv5s-SCB algorithm on NEU-DET(northeastern university-detect)reaches 77.9%,which is 3.7%higher than that of the YOLOv5s network on the premise of real-time detection.Compared with other improved algorithms based on YOLOv5s and YOLOv8,YOLOv5S-SCB achieves better detection effect.It is concluded that the proposed steel surface defect detection algorithm YOLOv5S-SCB can better detect defects on steel surfaces.

关 键 词：钢铁表面缺陷 YOLOv5s 注意力机制 BIFPN C2f 

分 类 号：TP391.4[自动化与计算机技术—计算机应用技术]