基于YOLOv8-NFMC的带钢表面缺陷检测算法
DOI:
作者:
作者单位:

安徽理工大学电气与信息工程学院 淮南 232001

作者简介:

通讯作者:

中图分类号:

TP391.41

基金项目:

国家自然科学基金项目(面上项目,重点项目,重大项目)


Surface Defect Detection Algorithm for Strip Steel Based on YOLOv8-NFMC
Author:
Affiliation:

Fund Project:

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

    针对YOLOv8算法在应用于带钢表面缺陷检测时存在漏检和错检等问题,提出了一种改进 YOLOv8 算法。针对数据集中的小目标的标签,在原损失 CIOU 的基础上面加入标准化高斯瓦瑟斯坦距离(Normalized Gaussian Wasserstein Distance,NWD),提升模型对小目标缺陷的检测能力;采用聚焦调制(Focal Modulation)替换 YOLOv8模型的空间池化金字塔(spatial pyramid pooling-fast,SPPF),在轻量化的同时,提高多尺度特征的表达能力;采用移动翻转瓶颈卷积(Mobile Inverted Bottleneck Conv,MBConv)替换 C2f中的 Conv 构建新模块C2f-MB,同时使用C2f-MB替换原有的 C2f 模块,增强特征表达能力和多尺度特征融合能力;在主干部分加入卷积块注意力机制(Convolutional Block Attention Module,CBAM)来抑制背景干扰,能更好捕获全局信息,提升了主干部分的特征提取能力。实验表明:改进后的 YOLOv8 算法在计算量下降的同时,mAP@50 提升了3个百分点,对漏检和错检等问题有明显改善。

    Abstract:

    Aiming at the problems of leakage and wrong detection when the YOLOv8 algorithm is applied to the surface defect detection of strip steel, an improved YOLOv8 algorithm is proposed. For the labels of small targets in the dataset, Normalized Gaussian Wasserstein Distance (NWD) is added on top of the original lossy CIOU, which improves the model's ability to detect defects of small targets; Focal Modulation is used to replace the spatial pooling pyramid of the YOLOv8 model, which improves the expression ability of multi-scale features while lightweighting; Mobile Inverted Bottleneck Conv (MBConv) is used to replace the Conv in C2f to construct a new module C2f-MB, and at the same time replace the original C2f-MB with C2f-MB. MB to replace the original C2f module with C2f-MB, which enhances the feature expression ability and multi-scale feature fusion ability; the Convolutional Block Attention Module (CBAM) is added in the backbone part to suppress the background interference, which can better capture the global information and improve the feature extraction ability of the backbone part. Experiments show that the improved YOLOv8 algorithm improves mAP@50 by 3 percentage points while decreasing the computation amount, which significantly improves the problems of missed detection and wrong detection.

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