山东理工大学学报(自然科学版)2026,Vol.40Issue(4):34-42,9.
基于改进YOLOv8n的钢材表面缺陷检测算法
Steel surface defect detection algorithm based on improved YOLOv8n
摘要
Abstract
In order to solve the problems of insufficient feature extraction ability and low detection accura-cy of complex defects such as irregular shape and significant scale variation,a steel surface defect detec-tion algorithm YOLOv8n-OMD based on improved YOLOv8n was proposed.Firstly,in the backbone net-work,the Online Convolutional Re-parameterization(OREPA)is used to reparameterize the complex network structure layer into a single convolutional layer,which has a low computational cost while main-taining the feature extraction ability.Secondly,the Median-Enhanced Channel and Spatial Attention(MECS)was designed and integrated to the end of the backbone network to enhance the extraction of im-portant features and improve the detection accuracy.Finally,the C2f_DCNv4 module was constructed by combining the deformable convolution DCNv4 and introduced into the neck network to increase sampling in the effective receptive field and effective position,thereby capturing the details of complex shape fea-tures more accurately.Experimental results show that the YOLOv8n-OMD algorithm achieves 82.2%mAP@0.5,48.8%mAP@0.5∶0.95,with a computational cost of 7.1×109 FLOPs on the NEU-DET steel surface defect dataset.Compared with the benchmark algorithm YOLOv8n,the mAP@0.5 and mAP@0.5∶0.95 are increased by 3.3%and 1.5%,respectively,and the computational cost is re-duced by 12.3%,which proves the effectiveness and practicability of the YOLOv8n-OMD algorithm for steel surface defect detection.关键词
缺陷检测/YOLOv8n/OREPA/可变形卷积/注意力机制Key words
defect detection/YOLOv8n/online convolutional re-parameterization/deformable convolution/attention mechanism分类
信息技术与安全科学引用本文复制引用
陈辉,李淑婷..基于改进YOLOv8n的钢材表面缺陷检测算法[J].山东理工大学学报(自然科学版),2026,40(4):34-42,9.基金项目
安徽理工大学研究生创新基金项目(2024cx2103) (2024cx2103)