Laser & Optoelectronics Progress, Volume. 60, Issue 4, 0415009(2023)

Steel-Plate Surface-Defect Detection Algorithm Based on Improved YOLOv5s

Yan Zhou1、*, Jiangnan Meng1、**, Jia Wu1,1、">, Zhi Luo2,2、">, and Dongli Wang1,1、">
Author Affiliations
  • 1School of Automation and Electronic Information, Xiangtan University, Xiangtan 411105, Hunan, China
  • 2Hunan Valin Xiangtan Iron and Steel Co., Ltd., Xiangtan 411105, Hunan, China
  • show less

    To solve the problem of the low accuracy and slow speed of traditional methods for detecting surface defects in steel plates, we propose an improved YOLOv5s algorithm. First, the steel datasets were re-clustered using K-means algorithm based on the intersection-over-union (IoU) metric distance, to obtain multiple groups of anchor boxes; a genetic algorithm was used to perform mutation operations and obtain multiple groups of anchor boxes that match the entire ground truth box better. Second, MixUp was fused with the Mosaic data enhancement to avoid over-fitting and improve the generalizability of the model. Then, the network structure was improved, and an attention module was incorporated to improve feature extraction capability of the network further. Finally, Focal loss was incorporated into the loss function to improve the convergence speed and detection accuracy of the network for hard-to-identify samples. Our experimental results show that the mean average precision (mAP) of the improved YOLOv5s algorithm on a test set is 78.4%, which is 3.0 percentage points higher than that of the original algorithm, and the speed is same as the original YOLOv5s. The detection performance of the improved YOLOv5s algorithm is better than that of DDN, Faster R-CNN, and YOLOv3, and it maintains a high detection speed.

    Tools

    Get Citation

    Copy Citation Text

    Yan Zhou, Jiangnan Meng, Jia Wu, Zhi Luo, Dongli Wang. Steel-Plate Surface-Defect Detection Algorithm Based on Improved YOLOv5s[J]. Laser & Optoelectronics Progress, 2023, 60(4): 0415009

    Download Citation

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

    Category: Machine Vision

    Received: Dec. 21, 2021

    Accepted: Mar. 29, 2022

    Published Online: Feb. 14, 2023

    The Author Email: Zhou Yan (yanzhou@xtu.edu.cn), Meng Jiangnan (mjnshizhu@163.com)

    DOI:10.3788/LOP213302

    Topics