基于YOLOv8-DSG的钢铁表面缺陷检测算法

邹彦艳, 曹衍芬, 张馨月, 李志, 崔世龙

吉林大学学报(信息科学版) ›› 2025, Vol. 43 ›› Issue (01) : 116-125. DOI: 10.19292/j.cnki.jdxxp.20240517.010

基于YOLOv8-DSG的钢铁表面缺陷检测算法

  • 邹彦艳, 曹衍芬, 张馨月, 李志, 崔世龙
作者信息 +
History +

摘要

针对传统图像处理算法对钢铁表面缺陷检测存在识别效率低、漏检误检率高等问题,提出了YOLOv8-DSG(Deformable Convolution Network Squeeze and Excitation Network Generalized Intersection over Union)钢铁表面缺陷检测算法。在传统YOLOv8算法的基础上,首先在Backbone网络的C2f(Convolution to Feature)模块中嵌入了可变形卷积网络DCN(Deformable Convolution Network),增强了模型在复杂背景条件下的特征提取能力;其次,在Neck网络中引入了SE(Squeeze and Excitation Network)注意力模块,突出钢铁表面重要特征信息,提升了特征融合的丰富性;最后,利用GIOU(Generalized Intersection Over Union)损失函数代替原有的CIOU(Complete Intersection Over Union),相比CIOU, GIOU引入了最小包围框面积比率,可更准确衡量框的重合面积。实验结果表明,YOLOv8-DSG算法在NEU-DET数据集上平均精度mAP达到80%,相较于原YOLOv8算法,提高了3.3%,且误检、漏检率低,具有更高的检测精度和运算效率,可在质量检测方面发挥重要作用。

关键词

缺陷检测 / YOLOv8算法 / 可变形卷积 / 注意力机制 / 损失函数

中图分类号

TG115 / TP391.41

引用本文

导出引用
邹彦艳, 曹衍芬, 张馨月, 李志, 崔世龙. 基于YOLOv8-DSG的钢铁表面缺陷检测算法. 吉林大学学报(信息科学版). 2025, 43(01): 116-125 https://doi.org/10.19292/j.cnki.jdxxp.20240517.010

基金

黑龙江省自然科学基金资助项目(LH2023A002)

评论

Accesses

Citation

Detail

段落导航
相关文章

/