基于改进YOLOv8的带钢表面缺陷检测算法

Enhanced strip steel surface defect detection algorithm based on YOLOv8

  • 摘要: 针对带钢表面缺陷检测用于尺度多变、背景复杂的缺陷容易产生漏检和误检,以及检测精度不佳等问题,提出了一种基于YOLOv8n的带钢表面缺陷检测算法。首先,为了提高主干网络的特征学习能力,将可变形卷积DCNv2引入到主干网络中,并同时融入动态卷积(dynamic convolution)模块,通过扩大感受野有效提高网络的特征提取能力。其次,在特征融合中,采用更加高效的Dysample上采样方法。引入通道注意力机制SE(squeeze and excitation)模块,提高网络对深层特征信息的提取能力。最后,使用DIoU结合inner-IoU作为损失函数,进一步提高算法的检测精度。并在NEU-DET数据上进行大量实验,结果表明,改进后的算法平均检测精度提高了2.2%,达到了79.5%。此外,在GC-10数据集上的实验结果表明该方法具有良好的鲁棒性。

     

    Abstract: Aiming at the problems of strip steel surface defect detection for defects with variable scales and complex backgrounds that are prone to leakage and false detection, as well as poor detection accuracy, a strip steel surface defect detection algorithm based on YOLOv8n is proposed. Firstly, to improve the feature learning ability of the backbone network, deformable convolution DCNv2 is introduced into the backbone network, and at the same time, a dynamic convolution module is incorporated, which effectively improves the feature extraction ability of the network by enlarging the sensory field. Secondly, in the feature fusion, a more efficient Dysample upsampling method is used. The channel attention mechanism module SE (squeeze and excitation) is introduced to improve the network’s ability to extract deep feature information. Finally, DIoU combined with inner-IoU is used as a loss function to further improve the detection accuracy of the algorithm. A large number of experiments are carried out on NEU-DET data, and the experimental results show that the improved algorithm improves the average detection accuracy by 2.2%, reaching 79.5%. In addition, the experimental results on the GC-10 dataset show that the method in this paper has good robustness.

     

/

返回文章
返回