ACD-YOLO: Improved YOLOv5-based method for steel surface defects detection

Jiacheng Fan,Min Wang,Baolei Li,Mingxue Liu,Dingcai Shen

Published 2023 in IET Image Processing

ABSTRACT

Since the quality of steel is of paramount importance in modern production, the defects detection of steel surface is significantly crucial. In this field, two‐stage detection algorithms have encountered issues about low detection speed, while one‐stage detection algorithms have room for improvement in detection accuracy. How to trade‐off between accuracy and speed of detection to better meet the demands of industrial production remains a challenge. To address this problem, this paper proposes a You Only Look Once version 5 (YOLOv5)‐based improved method ACD‐YOLO. ACD‐YOLO model incorporates anchors optimization, context augmentation module, and efficient convolution operators. In anchor optimization, the boundaries of anchors are optimized using an improved genetic algorithm. Moreover, to improve detection accuracy, a context augmentation module is incorporated into both the head and the backbone end of the network. Additionally, efficient convolution operators are adopted to address the increase of computation complexity caused by the context augmentation module. Experimental results show that ACD‐YOLO achieves mean average precision of 79.3%, with frames per second of 72. Compared to reference methods, ACD‐YOLO achieves the best balance between accuracy and speed of detection, and is more suitable for practice industrial production.

PUBLICATION RECORD

  • Publication year

    2023

  • Venue

    IET Image Processing

  • Publication date

    2023-11-11

  • Fields of study

    Materials Science, Computer Science, Engineering

  • Identifiers
  • External record

    Open on Semantic Scholar

  • Source metadata

    Semantic Scholar

CITATION MAP

EXTRACTION MAP

CLAIMS

  • No claims are published for this paper.

CONCEPTS

  • No concepts are published for this paper.

REFERENCES

Showing 1-19 of 19 references · Page 1 of 1

CITED BY

Showing 1-21 of 21 citing papers · Page 1 of 1