Journal of Optoelectronics · Laser, Volume. 35, Issue 1, 21(2024)

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

YANG Senquan1, DING Fan1, WEN Haoxiang1, LI Pu1,2, and HU Songxi1,2、*
Author Affiliations
  • 1[in Chinese]
  • 2[in Chinese]
  • show less

    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.

    Tools

    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

    Download Citation

    EndNote(RIS)BibTexPlain Text
    Save article for my favorites
    Paper Information

    Received: Aug. 12, 2022

    Accepted: --

    Published Online: Sep. 24, 2024

    The Author Email: HU Songxi (kobe1983@foxmail.com)

    DOI:10.16136/j.joel.2024.01.0583

    Topics