Laser & Optoelectronics Progress, Volume. 60, Issue 24, 2412003(2023)

An Improved YOLOv5 Algorithm for Steel Surface Defect Detection

Shaoxiong Li1, Zaifeng Shi1,3、*, Fanning Kong1, Ruoqi Wang1, and Tao Luo2
Author Affiliations
  • 1School of Microelectronics, Tianjin University, Tianjin 300072, China
  • 2College of Intelligence and Computing, Tianjin University, Tianjin 300072, China
  • 3Tianjin Key Laboratory of Imaging and Sensing Microelectronic Technology, Tianjin 300072, China
  • show less
    References(27)

    [1] Li W C, Yin B Q. Research progress of automated visual surface defect detection for industrial metal planar materials[J]. Journal of Electronic Measurement and Instrumentation, 35, 1-16(2021).

    [2] Tang B, Kong J Y, Wu S Q. Review of surface defect detection based on machine vision[J]. Journal of Image and Graphics, 22, 1640-1663(2017).

    [3] Ren S Q, He K M, Girshick R et al. Faster R-CNN: towards real-time object detection with region proposal networks[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 39, 1137-1149(2017).

    [4] Liu W, Anguelov D, Erhan D et al. SSD: single shot multibox detector[M]. Leibe B, Matas J, Sebe N, et al. Computer vision-ECCV 2016. Lecture notes in computer science, 9905, 21-37(2016).

    [5] Lin T Y, Goyal P, Girshick R et al. Focal loss for dense object detection[C], 2999-3007(2017).

    [8] Liu R Q, Huang M, Gao Z M et al. MSC-DNet: an efficient detector with multi-scale context for defect detection on strip steel surface[J]. Measurement, 209, 112467(2023).

    [9] Luo H, Li J, Jia C. Rail surface defect detection based on image enhancement and improved cascade R-CNN[J]. Laser & Optoelectronics Progress, 58, 2212001(2021).

    [10] Cheng X, Yu J B. RetinaNet with difference channel attention and adaptively spatial feature fusion for steel surface defect detection[J]. IEEE Transactions on Instrumentation and Measurement, 70, 2503911(2021).

    [11] Sun L S, Wei J X, Zhu D M et al. Surface defect detection algorithm of aluminum profile based on AM-YOLOv3 model[J]. Laser & Optoelectronics Progress, 58, 2415007(2021).

    [12] Yu X Y, Lü W T, Zhou D et al. ES-net: efficient scale-aware network for tiny defect detection[J]. IEEE Transactions on Instrumentation and Measurement, 71, 3511314(2022).

    [13] Cheng S, Yang H G, Xu X Q et al. Improved lightweight X-ray aluminum alloy weld defects detection algorithm based on YOLOv5[J]. Chinese Journal of Lasers, 49, 2104005(2022).

    [14] Dong H W, Song K C, He Y et al. PGA-net: pyramid feature fusion and global context attention network for automated surface defect detection[J]. IEEE Transactions on Industrial Informatics, 16, 7448-7458(2020).

    [15] Liu S T, Huang D, Wang Y H. Receptive field block net for accurate and fast object detection[M]. Ferrari V, Hebert M, Sminchisescu C, et al. Computer vision-ECCV 2018. Lecture notes in computer science, 11215, 404-419(2018).

    [16] Huang Z L, Wei Y C, Wang X G et al. AlignSeg: feature-aligned segmentation networks[J]. IEEE Transactions on Pattern Analysis and Machine Intelligence, 44, 550-557(2022).

    [17] Wang Q L, Wu B G, Zhu P F et al. ECA-net: efficient channel attention for deep convolutional neural networks[C], 11531-11539(2020).

    [19] Wang C Y, Mark Liao H Y, Wu Y H et al. CSPNet: a new backbone that can enhance learning capability of CNN[C], 1571-1580(2020).

    [20] Lin T Y, Dollár P, Girshick R et al. Feature pyramid networks for object detection[C], 936-944(2017).

    [21] Liu S, Qi L, Qin H F et al. Path aggregation network for instance segmentation[C], 8759-8768(2018).

    [22] He K M, Zhang X Y, Ren S Q et al. Deep residual learning for image recognition[C], 770-778(2016).

    [24] Yu F, Koltun V. Multi-scale context aggregation by dilated convolutions[C](2016).

    [25] Song G L, Liu Y, Wang X G. Revisiting the sibling head in object detector[C], 11560-11569(2020).

    [26] He Y, Song K C, Meng Q G et al. An end-to-end steel surface defect detection approach via fusing multiple hierarchical features[J]. IEEE Transactions on Instrumentation and Measurement, 69, 1493-1504(2020).

    Tools

    Get Citation

    Copy Citation Text

    Shaoxiong Li, Zaifeng Shi, Fanning Kong, Ruoqi Wang, Tao Luo. An Improved YOLOv5 Algorithm for Steel Surface Defect Detection[J]. Laser & Optoelectronics Progress, 2023, 60(24): 2412003

    Download Citation

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

    Category: Instrumentation, Measurement and Metrology

    Received: Feb. 27, 2023

    Accepted: Apr. 7, 2023

    Published Online: Nov. 27, 2023

    The Author Email: Shi Zaifeng (shizaifeng@tju.edu.cn)

    DOI:10.3788/LOP230711

    Topics