Journal of Optoelectronics · Laser, Volume. 35, Issue 1, 21(2024)
Hot-rolled steel strip surface defects detection based on CA-YOLOv5
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.
Get Citation
Copy Citation Text
YANG Senquan, DING Fan, WEN Haoxiang, LI Pu, HU Songxi. Hot-rolled steel strip surface defects detection based on CA-YOLOv5[J]. Journal of Optoelectronics · Laser, 2024, 35(1): 21
Received: Aug. 12, 2022
Accepted: --
Published Online: Sep. 24, 2024
The Author Email: HU Songxi (kobe1983@foxmail.com)