Electro-Optic Technology Application, Volume. 38, Issue 6, 60(2023)

Improve YOLOv5 Steel Surface Defect Detection Technology

LI Dengyue and JIANG Yueqiu
Author Affiliations
  • [in Chinese]
  • show less

    In response to the initial YOLOv5 target detection algorithm's problems of inadequate extraction of defect features in complex targets, imprecise localization, low detection accuracy and high miss detection rate, an improved YOLOv5 steel surface defect detection algorithm is proposed, in which the anchor frame is calculated with the K-Means++ algorithm for anchor frame selection, so that the randomly selected clustering center tends to the global optimal solution as much as possible and the prediction frame is more accurate. The CBAM attention mechanism is added to give higher weights to the defects in complex images and enhance the attention to key information. The results after the experimental comparison show that the improved YOLOv5 algorithm has better detection performance.

    Tools

    Get Citation

    Copy Citation Text

    LI Dengyue, JIANG Yueqiu. Improve YOLOv5 Steel Surface Defect Detection Technology[J]. Electro-Optic Technology Application, 2023, 38(6): 60

    Download Citation

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

    Category:

    Received: Apr. 24, 2023

    Accepted: --

    Published Online: May. 14, 2024

    The Author Email:

    DOI:

    CSTR:32186.14.

    Topics