基于CA-YOLOv5的热轧带钢表面缺陷检测方法
DOI:
CSTR:
作者:
作者单位:

(1.韶关学院 智能工程学院,广东 韶关 512005; 2.华南理工大学 广东省高分子先进制造技术及装备重点实验室,广东 广州 510640)

作者简介:

胡松喜 (1983-),男,博士,副教授,主要从事装备设计及其过程控制研究。

通讯作者:

中图分类号:

基金项目:

国家自然科学基金(52105268, 62003101)和韶关学院校级自然科学类科研重点项目(SZ2021KJ04)资助项目


Hot-rolled steel strip surface defects detection based on CA-YOLOv5
Author:
Affiliation:

(1.School of Intelligent Engineering, Shaoguan University, Shaoguan, Guangdong 512005, China;2.Guangdong Provincial Key Laboratory of Technique and Equipment for Macromolecular Advanced Manufacturing, South China University of Technology, Guangzhou, Guangdong 510640, China)

Fund Project:

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

    针对当前热轧带钢表面缺陷检测中存在精度低及复杂背景干扰等问题,提出一种基于坐标注意力(coordinate attention, CA)的CA-YOLOv5缺陷检测方法。主要对YOLOv5的输入端、外加模块和检测端3个方面进行改进:在输入端,采用随机拼接4张或9张图片的方法对训练数据进行增广,并利用遗传算法(genertic algorithm,GA)对网络超参数进行寻优,使得YOLOv5更适用于带钢缺陷检测;在主干网络和外加模块之间引入CA机制,加强网络对缺陷深层特征的提取能力;最后,在检测端,对每一检测分支进行解耦,将检测的分类和位置回归两类任务分开,提升网络对缺陷的检测能力。在NEU-DET热轧带钢表面缺陷数据集上进行了验证实验,实验结果证明,CA-YOLOv5的均值平均精度(mean average precision,mAP)达到84.36%,不仅较原YOLOv5算法提升6.68%,而且优于其他先进的检测算法。

    Abstract:

    The hot-rolled steel strip surface defects detection methods have low detection accuracy and suffer background clutters.To solve the issues mentioned above,this paper proposes a CA-YOLOv5 defects detection method based on coordinate attention (CA).The improvement of CA-YOLOv5 are three aspects.First,to make the YOLOv5 more suitable for strip defects detection,the data argumentation using the 4 or 9 mosaic images randomly and hyperparameter optimization based on the genetic algorithm (GA) are employed in the input of the network.Second,the CA mechanism is introduced between the backbone and the neck model to improve the feature extracting ability of the network.Third,each detecting head is decoupled to separate the classification and regression tasks,which could further boost the accuracy of defects detection.Comparative experiments on NEU-DET dataset show that the proposed network obtains 84.36% mean average precision (mAP),which not only exceeds YOLOv5 by 6.68%,but also outperformances other state-of-the-art detectors.

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

杨森泉,丁凡,文昊翔,李璞,胡松喜.基于CA-YOLOv5的热轧带钢表面缺陷检测方法[J].光电子激光,2024,35(1):21~28

复制
分享
文章指标
  • 点击次数:
  • 下载次数:
  • HTML阅读次数:
  • 引用次数:
历史
  • 收稿日期:2022-08-12
  • 最后修改日期:2022-11-16
  • 录用日期:
  • 在线发布日期: 2024-01-03
  • 出版日期:
文章二维码