Laser & Optoelectronics Progress, Volume. 60, Issue 24, 2412003(2023)
An Improved YOLOv5 Algorithm for Steel Surface Defect Detection
[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).
[6] Redmon J, Farhadi A. YOLOv3: an incremental improvement[EB/OL]. https://arxiv.org/abs/1804.02767
[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).
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
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)