兵工自动化2024，Vol.43Issue(12)：30-34,5.DOI:10.7690/bgzdh.2024.12.008
基于改进YOLOv5的钢材表面缺陷检测
Steel Surface Defect Detection Based on Improved YOLOv5
摘要
Abstract
Aiming at the problems of unsatisfactory defect detection effect and insufficient feature extraction of small and medium targets in steel surface defect detection,a YOLOv5s-ADW algorithm based on YOLOv5 algorithm was proposed.a mixed model of self-attention and convolution(ACmix)module is integrated into the backbone network layer to enhance the feature sensitivity of the model.The deformable large kernel attention(D-LKA)mechanism is added to the feature fusion layer to enhance the ability of the model to capture irregular defects in images.The original loss function was replaced by Wise-IoU loss function to reduce the influence of low-quality examples on the model detection effect and improve the detection ability of small target defects,and the experimental verification was carried out on NEU-DET.Experimental results show that the mean average precision(mAP)of the YOLOv5s-ADW algorithm reaches 88.3%,which is 14.4%higher than that of the original model.The average precision(AP)of small target defects and defects with high missed detection rate is also greatly improved,which can better solve the above problems compared with other mainstream algorithms.关键词
YOLOv5/钢材表面缺陷检测/小目标缺陷检测/特征提取Key words
YOLOv5/Steel surface defect detection/Small target defect detection/feature extraction分类
信息技术与安全科学引用本文复制引用
刘祉燊,张晓玲,刘珂宇,刘晓军,刘晓静..基于改进YOLOv5的钢材表面缺陷检测[J].兵工自动化,2024,43(12):30-34,5.基金项目
山东省高等学校科技计划项目(J18KA359) （J18KA359）
