基于YOLO-AMM的轻量化带钢表面缺陷检测算法
DOI:
CSTR:
作者:
作者单位:

东华大学机械工程学院 上海 201620

作者简介:

通讯作者:

中图分类号:

TN911;TP391.4

基金项目:


Lightweight strip steel surface defect detection algorithm based on YOLO-AMM
Author:
Affiliation:

College of Mechanical Engineering, Donghua University,Shanghai 201620, China

Fund Project:

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

    针对现有带钢表面缺陷检测算法参数量大、检测精度不足及在嵌入式设备部署难等问题,提出了一种基于改进YOLOv11n的轻量化检测算法YOLO-AMM。首先,引入ADown模块,解决传统跨步卷积下采样导致的微小缺陷特征丢失问题以提升精度,减少参数量以提升实时性;其次,构建C3k2_MDSB模块,增强多尺度特征提取能力和计算效率;最后,在特征融合网络中引入混合局部通道注意力(MLCA)机制,提升对不同尺度缺陷的特征提取能力。实验表明,YOLO-AMM算法的mAP50较YOLOv11n提升3.6%,召回率提升2.3%,精确率提升2.9%,同时参数减少18.0%,计算量减少14.1%,模型大小下降16.5%,优于多种经典轻量级与同类改进算法,适合部署于资源受限的嵌入式设备中。

    Abstract:

    To address the issues of large parameter counts, insufficient detection accuracy, and difficulty in deployment on embedded devices in existing strip steel surface defect detection algorithms, a lightweight detection algorithm named YOLO-AMM based on improved YOLOv11n is proposed. Firstly, the ADown module is introduced to solve the problem of tiny defect feature loss caused by traditional strided convolution downsampling, thereby improving accuracy and to reduce the number of parameters for enhanced real-time performance. Secondly, the C3k2_MDSB module is constructed to strengthen multi-scale feature extraction capability and computational efficiency. Finally, the mixed local channel attention (MLCA) attention mechanism is incorporated into the feature fusion network to improve the feature extraction ability for defects of different scales. Experimental results show that compared with YOLOv11n, the mAP50 of the YOLO-AMM algorithm is increased by 3.6%, the recall rate by 2.3%, and the precision by 2.9%. Meanwhile, the number of parameters is reduced by 18.0%, the computational load by 14.1%, and the model size by 16.5%. The proposed algorithm outperforms a variety of classic lightweight algorithms and similar improved algorithms and is suitable for deployment on resource-constrained embedded devices.

    参考文献
    相似文献
    引证文献
引用本文

孙艳龙,陈广锋.基于YOLO-AMM的轻量化带钢表面缺陷检测算法[J].电子测量技术,2026,49(11):245-253

复制
分享
相关视频

文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:
  • 最后修改日期:
  • 录用日期:
  • 在线发布日期: 2026-09-03
  • 出版日期:
文章二维码

重要通知公告

①《电子测量技术》期刊收款账户变更公告