基于改进YOLOv8n的钢材表面缺陷检测
DOI:
作者:
作者单位:

1.沈阳化工大学;2.辽宁省化工过程工业智能化技术重点实验室

作者简介:

通讯作者:

中图分类号:

TP391.4

基金项目:

辽宁省教育厅基本科研项目面上项目(LJKMZ20220782)


Steel surface defect detection algorithm based on improved YOLOv8n
Author:
Affiliation:

Fund Project:

  • 摘要
  • |
  • 图/表
  • |
  • 访问统计
  • |
  • 参考文献
  • |
  • 相似文献
  • |
  • 引证文献
  • |
  • 资源附件
  • |
  • 文章评论
    摘要:

    针对钢材表面缺陷类型繁多、尺寸差异显著,且现有模型结构复杂、检测精度不高等问题,提出一种改进YOLOv8n的钢材表面缺陷检测算法。首先,在特征提取模块引入全维动态卷积(ODConv),实现对多维度特征的高效捕捉, 减少信息损失;其次,在特征融合模块嵌入渐进特征金字塔网络(AFPN)结构,促进非相邻层级特征间的直接交互,缓解语义断层,提高特征融合质量;最后,在损失函数部分应用动态非单调聚焦机制的Wise-IoUv3替代原有的CIoU损失,以优化边界框回归,加快网络收敛的同时提高检测精度。在NEU-DET数据集上进行多组实验,结果表明,改进后的YOLOv8-ODAW网络模型相比原网络模型mAP50%提升了7.3%、GFLOPs下降了21.95%,对钢材表面缺陷有更好的定位能力和检测精度,检测速度能够满足工业检测的要求。

    Abstract:

    Addressing the issues of diverse defect types, significant size variations on steel surfaces, as well as the complexity and limited accuracy of existing models, an improved steel surface defect detection algorithm based on YOLOv8n was proposed. Initially, Omni-dimensional Dynamic Convolution (ODConv) was introduced in the feature extraction module to efficiently capture multi-dimensional features, thereby reducing information loss. Subsequently, an Asymptotic Feature Pyramid Network (AFPN) structure was embedded in the feature fusion module to facilitate direct interaction between non-adjacent level features, alleviating semantic gaps and enhancing the quality of feature integration. Lastly, the original CIoU loss was supplanted by Wise-IoUv3, which employs a dynamic non-monotonic focusing mechanism in the loss function, to optimize bounding box regression. This adjustment expedited network convergence while boosting detection precision. Extensive experiments conducted on the NEU-DET dataset demonstrated that the refined YOLOv8-ODAW network model achieved a 7.3% increase in mAP50%, a 21.95% decrease in GFLOPs compared to the original model, exhibiting superior localization capability and detection accuracy for steel surface defects. The detection speed met the requirements of industrial inspection standards.

    参考文献
    相似文献
    引证文献
引用本文
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2024-05-21
  • 最后修改日期:2024-07-15
  • 录用日期:2024-07-17
  • 在线发布日期:
  • 出版日期: