期刊文献+

基于CA-YOLOv5的热轧带钢表面缺陷检测方法 被引量:1

Hot-rolled steel strip surface defects detection based on CAYOLOv5
原文传递
导出
摘要 针对当前热轧带钢表面缺陷检测中存在精度低及复杂背景干扰等问题,提出一种基于坐标注意力(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%,而且优于其他先进的检测算法。 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-YOLOv5defects detection method based on coordinate attention(CA).The improvement of CA-YOLOv5are three aspects.First,to make the YOLOv5more suitable for strip defects detection,the data argumentation using the 4or 9mosaic 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 YOLOv5by 6.68%,but also outperformances other state-of-the-art detectors.
作者 杨森泉 丁凡 文昊翔 李璞 胡松喜 YANG Senquan;DING Fan;WEN Haoxiang;LI Pu;HU Songxi(School of Intelligent Engineering,Shaoguan University,Shaoguan,Guangdong 512005,China;Guangdong Provincial Key Laboratory of Technique and Equipment for Macromolecular Advanced Manufacturing,South China University of Technology,Guangzhou,Guangdong 510640,China)
出处 《光电子.激光》 CSCD 北大核心 2024年第1期21-28,共8页 Journal of Optoelectronics·Laser
基金 国家自然科学基金(52105268,62003101) 韶关学院校级自然科学类科研重点项目(SZ2021KJ04)资助项目。
关键词 缺陷检测 YOLOv5 坐标注意力(CA) 解耦检测头 defects detection YOLOv5 coordinate attention(CA) decoupled detection head
  • 相关文献

参考文献6

二级参考文献32

共引文献73

同被引文献10

引证文献1

相关作者

内容加载中请稍等...

相关机构

内容加载中请稍等...

相关主题

内容加载中请稍等...

浏览历史

内容加载中请稍等...
;
使用帮助 返回顶部