Optoelectronics Letters, Volume. 21, Issue 5, 306(2025)

Steel surface defect detection based on lightweight YOLOv7

Tao SHI, Rongxin WU, Wenxu ZHU, and Qingliang MA
References(24)

[1] [1] QIAO Y, WANG G. Recent status of production, administration policies, and low-carbon technology development of China’s steel industry[J]. Metals, 2024, 14(4): 480-495.

[2] [2] LI J, LI H, HU X, et al. Research progress of surface defect detection technology based on deep learning[J]. Computer integrated manufacturing system, 2024, 30(3): 774-791.

[3] [3] TANG X, ZHOU J, JIAN G, et al. Nondestructive differential eddy current testing for corrosion detection on coated aluminium alloys[J]. Anti-corrosion methods and materials, 2024, 71(6): 855-867.

[4] [4] ZHAO B T, CHEN Y R, JIA X F, et al. Steel surface defect detection algorithm in complex background scenarios[J]. Measurement, 2024, 237: 175-189.

[5] [5] GIRSHICK R, DONAHUE J, DARRELL T, et al. Rich feature hierarchies for accurate object detection and semantic segmentation[C]//2014 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), June 23-28, 2014, Columbus, OH, USA. New York: IEEE, 2014: 580-587.

[6] [6] GIRSHICK R. Fast R-CNN[C]//2015 IEEE International Conference on Computer Vision (ICCV), December 7-13, 2015, Santiago, Chile. New York: IEEE, 2015: 1440-1448.

[7] [7] LIN T Y, DOLLAR P, GIRSHICK R, et al. Feature pyramid networks for object detection[C]//2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), July 21-26, 2017, Honolulu, HI, USA. New York: IEEE, 2017: 936-944.

[8] [8] LIU W, ANGUELOY D, ERHAN D, et al. SSD: single shot MultiBox detector[C]//European Conference on Computer Vision (ECCV), October 11-14, 2016, Amsterdam, The Netherlands. Heidelberg: Springer, 2016, 9905: 21-37.

[9] [9] LIN T Y, GOYAL P, GIRSHICK R, et al. Focal loss for dense object detection[C]//2017 IEEE International Conference on Computer Vision (ICCV), October 22-29, 2017, Venice, Italy. New York: IEEE, 2017: 2999-3007.

[10] [10] REDMON J, DIVVALA S, GIRSHICK R, et al. You only look once: unified, real-time object detection[C]//2016 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), June 27-30, 2016, Las Vegas, NV, USA. New York: IEEE, 2016: 779-788.

[11] [11] XING J F. Surface defect identification and system development of hot-rolled strip based on convolutional neural network[D]. Shenyang: Northeastern University, 2017: 1-54. (in Chinese)

[12] [12] 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, 2020, 69(4P2): 1493-1504.

[13] [13] XU Q, ZHU H J, FAN H H, et al. Research on improved YOLOv3 network on surface defect detection of steel plates[J]. Computer engineering and applications, 2020, 56(16): 265-272.

[14] [14] LIAN J W, HE J H, NIU Y, et al. Fast and accurate detection of surface defect based on improved YOLOv4[J]. Assembly automation, 2022, 42(1): 134-146.

[15] [15] WANG C Y, BOCHKOVSKIY A, LIAO H Y M. YOLOv7: trainable bag-of-freebies sets new state-of-the-art for real-time object detectors[C]//Proceedings of the IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), June 17-24, 2023, Vancouver, Canada. New York: IEEE, 2023: 7464-7475.

[16] [16] WANG C Y, LIAO H, WU Y H, et al. CSPNet: a new backbone that can enhance learning capability of CNN[C]//2020 IEEE/CVF Conference on Computer Vision and Pattern Recognition Workshops (CVPRW), June 13-19, 2020, Seattle, WA, USA. New York: IEEE, 2020: 1571-1580.

[17] [17] HE K M, ZHANG X Y, REN S Q, et al. Spatial pyramid pooling in deep convolutional networks for visual recognition[J]. IEEE transactions on pattern analysis and machine intelligence, 2015, 37(9): 1904-1916.

[18] [18] DING X, ZHANG X, MA N, et al. RepVGG: making VGG-style ConvNets great again[C]//2021 IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), June 20-25, 2021, Nashville, TN, USA. New York: IEEE, 2021: 13733-13742.

[19] [19] HAN K, WANG Y H, TIAN Q, et al. GhostNet: more features from cheap operations[C]//IEEE/CVF Conference on Computer Vision and Pattern Recognition (CVPR), June 13-19, 2020, Seattle, WA, USA. New York: IEEE, 2020: 1577-1586.

[20] [20] WOO S, PARK J, LEE J Y, et al. CBAM: convolutional block attention module[C]//European Conference on Computer Vision (ECCV), September 8-14, 2018, Munich, Germany. Heidelberg: Springer, 2018: 3-19.

[21] [21] DU S, ZHANG B, ZHANG P, et al. An improved bounding box regression loss function based on CIOU loss for multi-scale object detection[C]//2021 IEEE 2nd International Conference on Pattern Recognition and Machine Learning (PRML), July 16-18, 2021, Chengdu, China. New York: IEEE, 2021: 92-98.

[22] [22] ZHANG Y F, REN W Q, ZHANG Z, et al. Focal and efficient IOU loss for accurate bounding box regression[J]. Neurocomputing, 2022, 506: 146-157.

[23] [23] CHEN C Q, FAN Y C, WANG L. Logo detection based on improved mosaic data enhancement and feature fusion[J]. Computer measurement and control, 2022, 30(10): 188-194.

[24] [24] YUN S, HAN D, OH S J, et al. CutMix: regularization strategy to train strong classifiers with localizable features[C]//International Conference on Computer Vision, October 20-26, 2019, Seoul, Korea (South). New York: IEEE, 2019: 659-666.

Tools

Get Citation

Copy Citation Text

SHI Tao, WU Rongxin, ZHU Wenxu, MA Qingliang. Steel surface defect detection based on lightweight YOLOv7[J]. Optoelectronics Letters, 2025, 21(5): 306

Download Citation

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

Received: Jun. 14, 2023

Accepted: Apr. 11, 2025

Published Online: Apr. 11, 2025

The Author Email:

DOI:10.1007/s11801-025-3104-2

Topics