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