Abstract
Aiming at the problems of low detection accuracy and slow detection speed in the traditional method of detecting strip steel surface defects, this paper proposes an improved yolov5 algorithm for detecting strip steel surface defects. Firstly, the data set of strip surface defects was constructed, and the K-means algorithm was used to cluster the defect samples, and the prior box parameters of different sizes were obtained. Secondly, the attention-yolov5 algorithm is proposed, which draws on the item-based Attention mechanism, adds channel Attention and spatial Attention mechanism to the feature extraction network, and uses the filtered weighted feature vector to replace the original feature vector for residual fusion. Finally, In order to improve the ability of defect feature extraction, the convolution layer is added after the main feature is extracted from different feature layers of the network output and after the pooling structure of spatial pyramid. The experimental results show that the mAP value of the improved yolov5 algorithm on the test set is as high as 87.3%, which is 5% higher than the original yolov5 algorithm. The average detection time of a single image is 0.0219s, which is basically the same as the original algorithm, and the detection performance is also better than the Faster RCNN and yolov3.
This is a preview of subscription content, log in via an institution to check access.
Access this chapter
Tax calculation will be finalised at checkout
Purchases are for personal use only
References
Li, X.: Study on surface quality defects of cold rolling hood furnace annealing strip, 4(05), 11–12 (2021)
Zheng, Y., Li, G., Li, Y.: A review on the application of deep learning in image recognition. Comput. Eng. Appl. 55(12), 20–36 (2019)
Girshick, R., Donahue, J., Darrell, T., et al.: Rich feature hierarchies for accurate object detection and semantic segmentation. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 580–587. IEEE Computer Society Press, LosAlamitos (2014)
Girshick, R.: Fast RCNN. In: Proceedings of the IEEE International Conference on Computer Vision, pp. 1440–1448. IEEE Computer Society Press, Los Alamitos (2015)
Ren, S.Q., He, K.M., Girshick, R., et al.: Faster RCNN: towards real-time object detection with region proposal networks. IEEE Trans. Pattern Anal. Mach. Intell. 39(6), 1137–1149 (2017)
He, K.M., Gkioxari, G., Dollár, P., et al.: Mask RCNN. In: Proceedings of the IEEE International Conference on Computer Vision, pp. 2961–2969. IEEE Computer Society Press, Los Alamitos (2017)
Sermanet, P., Eigen, D., Zhang, X., et al.: OverFeat: integrated recognition, localization and detection using convolutional neworks[OL]. https://arxiv.org/abs/1312.6229. 03 July 2019
Redmon, J., Divvala, S., Girshick, R., et al.: You only look once: unified, real-time object detection. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 779–788. IEEE Computer Society Press, Los Alamitos (2016)
Redmon, J., Farhadi, A.: yolo9000: better, faster, stronger. In: Proceedings of the IEEE Conference on Computer Vision and Pattern Recognition, pp. 7263–7271. IEEE Computer Society Press, Los Alamitos (2017)
Redmon, J., Farhadi, A.: yolov3: an incremental improvement[OL]. https://arxiv.org/abs/1804.02767. 03 July 2019
Liu, W., Anguelov, D., Erhan, D., Szegedy, C., Reed, S., Fu, C.-Y., Berg, A.C.: SSD: single shot multiBox detector. In: Leibe, B., Matas, J., Sebe, N., Welling, M. (eds.) ECCV 2016. LNCS, vol. 9905, pp. 21–37. Springer, Cham (2016). https://doi.org/10.1007/978-3-319-46448-0_2
Li, B., Wang, C., Wu, J., Liu, J., Tong, L., Guo, Z.: An Improved yoloV4 algorithm for surface defect detection of AEROENGINE parts [J/OL]. Laser Optoelectron. Prog. {3}, {4}, {5}, 1–17. http://kns.cnki.net/KCMS/detail/31.1690. 18 July 2021. TN. 20210105.1048.006. HTML
Chang, J., Guan, S., Shi, H., Hu, L., Ni, Y.: Defect classification of strip steel based on improved generative adversation network and mobilenetv3. Adv. Laser Optoelectron. 201, 58(04), 221–226
Ma, X., Zhu, D., Jin, C., Tong, X.: Detection of bullet appearance defect based on improved RCNN. Laser Optoelectron. Prog. 56(15), 117–124 (2019)
Cheng, S., Dai, J., Yang, H., Chen, Y.: Weld image detection and recognition based on improved yolov4 [J/OL]. Laser Optoelectron. {3}, {4} {5}, 1–12. http://kns.cnki.net/kcms/detail/31.1690. 18 July 2021 The tn. 20210712.1728.088. HTML
Weng, Y., Xiao, J., Xia, Y.: Improved mask RCNN algorithm for surface defect detection of strip steel [J/OL]. Comput. Eng. Use. http://kns.cnki.net/kcms/detail/11.2127.TP.20210420.0937.014.html
Author information
Authors and Affiliations
Corresponding author
Editor information
Editors and Affiliations
Rights and permissions
Copyright information
© 2022 The Author(s), under exclusive license to Springer Nature Singapore Pte Ltd.
About this paper
Cite this paper
Zeqiang, S., Bingcai, C. (2022). Improved Yolov5 Algorithm for Surface Defect Detection of Strip Steel. In: Liang, Q., Wang, W., Mu, J., Liu, X., Na, Z. (eds) Artificial Intelligence in China. Lecture Notes in Electrical Engineering, vol 854. Springer, Singapore. https://doi.org/10.1007/978-981-16-9423-3_56
Download citation
DOI: https://doi.org/10.1007/978-981-16-9423-3_56
Published:
Publisher Name: Springer, Singapore
Print ISBN: 978-981-16-9422-6
Online ISBN: 978-981-16-9423-3
eBook Packages: Computer ScienceComputer Science (R0)