改进YOLOv8n的热轧钢带表面缺陷检测方法

Surface defect detection method for hot-rolled steel strips based on improved YOLOv8n

  • 摘要: 针对热轧钢带表面缺陷形态复杂多变、背景干扰强导致的漏检误检率高,以及模型在边缘设备难部署等问题,基于YOLOv8n提出一种轻量级优化方法. 首先,设计了轻量特征提取模块E-IDC,通过引入Inception深度卷积并与通道注意力相结合,在控制计算复杂度的同时增强模型对细微缺陷的多尺度感知能力;其次,构建C2f-DCNv2模块,引入可变形卷积使模型能根据缺陷形变自适应调整采样点分布,提升对不规则轮廓的拟合精度;最后,采用WIoUv3损失函数,通过动态聚焦机制抑制低质量样本的干扰,提高训练稳定性. 在NEU-DET数据集实验表明,改进模型mAP@0.5提升4.4个百分点,参数量与计算量分别降低7%与16%. 相比主流方法,改进模型在保持较低计算开销的同时,有效提升了复杂缺陷的检测精度.

     

    Abstract: A lightweight YOLOv8n method addresses high missed and false detection rates of complex hot-rolled steel strip defects and edge deployment difficulties. An E-IDC module integrates Inception depthwise convolution and channel attention to enhance subtle defect perception while controlling computational complexity. A C2f-DCNv2 module employs deformable convolution to adaptively adjust sampling distributions, improving irregular contour fitting accuracy. The WIoUv3 loss function suppresses low-quality sample interference via a dynamic focusing mechanism to stabilize training. NEU-DET experiments show a 4.4% mAP@0.5 increase, with 7% fewer parameters and 16% lower computational costs. Compared with mainstream methods, the model effectively enhances complex defect detection accuracy under lower computational overhead.

     

/

返回文章
返回