Laser & Optoelectronics Progress, Volume. 58, Issue 14, 1410014(2021)

Recognition Algorithm of Strip Steel Surface Defects Based on Attention Model

Yanuo Lu1, Bingcai Chen1,2、*, Degang Chen1, Shixiang Yan1, and Shunping Li1
Author Affiliations
  • 1School of Computer Science and Technology, Xinjiang Normal University, Urumqi, Xinjiang 830054, China
  • 2College of Computer Science and Technology, Dalian University of Technology, Dalian, Liaoning 116024, China
  • show less

    In order to improve the quality and output of industrial strip steels and address the problems of traditional manual identification such as identification difficulty, low efficiency and lack of objectivity, we propose a method for identifying strip steel surface defects based on the soft attention mechanism and improve the traditional deep residual network ResNet model. Moreover, we use the pseudo-color image enhancement technique to process images and obtain new training sets. The experimental results show that compared with the traditional models, the improved models of A-ResNet50 and A-ResNet101 can both accurately identify different types of strip steel surface defect images under different signal-to-noise ratios. The accuracies on the test set are 98.61% and 98.05%, and the unit inference time is 0.078 s and 0.130 s, respectively. Thus the feasibility and reliability of these two models in the identification of surface defects on strip steels are confirmed. The proposed method possesses a high identification accuracy, which can be used to realize the intelligent identification of surface defects on strip steels and meet the demands of industrial identification.

    Tools

    Get Citation

    Copy Citation Text

    Yanuo Lu, Bingcai Chen, Degang Chen, Shixiang Yan, Shunping Li. Recognition Algorithm of Strip Steel Surface Defects Based on Attention Model[J]. Laser & Optoelectronics Progress, 2021, 58(14): 1410014

    Download Citation

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

    Category: Image Processing

    Received: Sep. 30, 2020

    Accepted: Nov. 14, 2020

    Published Online: Jun. 30, 2021

    The Author Email: Chen Bingcai (china@dlut.edu.cn)

    DOI:10.3788/LOP202158.1410014

    Topics