An Improved YOLOv8n Algorithm for Steel Surface Defect Detection

Dongmei Ma; Na Zhang; Xuelong Lv

doi:10.62051/ijcsit.v4n2.17

Authors

Dongmei Ma
Na Zhang
Xuelong Lv

DOI:

https://doi.org/10.62051/ijcsit.v4n2.17

Keywords:

YOLOv8n, Context Guided Block, ASFF, Shape-IoU

Abstract

Due to the manufacturing process and other aspects, steel surface defects vary in shape and size, which is a difficult point for defect detection. For this reason, this paper proposes a steel surface defect detection algorithm based on YOLOv8n. The algorithm improves the C2f module of Backbone in this network by adding Context Guided Block (CG block), which can effectively extract the local features, surrounding context and global context, and integrate this information to improve the precision and accuracy of detection. Meanwhile, we also change the detection header of YOLOv8 to Detect_ASFF by introducing the Adaptive Spatial Feature Fusion (ASFF) technique. This improvement effectively filters out the conflicting information, which can allow the model to have a better adaptability at different scales. In this paper, Shape-IoU is adopted as the bounding box loss function to make up for the traditional IoU loss that ignores the intrinsic properties of the bounding box in the calculation process. After optimisation, mAP@0.5 are improved to 67.9% and mAP@0.5~0.95 are improved to 35.1%, which is 2.8% and 1.1% respectively. The improved algorithm in this paper has improved in detection accuracy, which proves the practicality and effectiveness of the algorithm in this paper.

Downloads

Download data is not yet available.

References

[1] Su Hu, Zhang Jiabin, Zhang Bohao et al. A review of surface defect detection based on visual perception [J]. Computer Integrated Manufacturing Systems, 2023, 29(01): 169-191. SU H, ZHANG J B, ZHANG B H, et al. Review of surface defect inspection based on visual perception [J]. Computer Integrated Manufacturing Systems, 2023, 29(01): 169-191.

[2] Girshick R, Donahue J, Darrell T, et al. Rich feature hierarchies for accurate object detection and semantic segmentation[C]//Proceedings of the IEEE conference on computer vision and pattern recognition. 2014: 580-587.

[3] Girshick R. Fast r-cnn[C]//Proceedings of the IEEE international conference on computer vision. 2015: 1440-1448.

[4] Ren S, He K, Girshick R, et al. Faster r-cnn: Towards real-time object detection with region proposal networks [J]. Advances in neural information processing systems, 2015, 28.

[5] Liu, W. et al. SSD: Single Shot MultiBox Detector [C]. ECCV. 2016, 28: 2137.

[6] ] Lin T Y, Goyal P, Girshick R, et al. Focal loss for dense object detection[C]//Pro ceedings of the IEEE international conference on computer vision. 2017: 2980-298 8.

[7] REDMON J, DIVVALA S, GIRSHICK R, et al. You only look once: unified, real-time object detection[C]//Proceedings of 2016 IEEE Conference on Computer Vision and Pattern Recognition, June 26-July 1, 2016, Las Vegas, NV, USA. New York: IEEE, 2016:779-788.

[8] Redmon J, Farhadi A. YOLO9000: better, faster, stronger[C]//Proceedings of the IEEE conference on computer vision and pattern recognition. 2017: 7263-7271.

[9] Redmon J, Farhadi A. Yolov3: An incremental improvement [J]. arXiv preprint arXiv:1804.02767, 2018.

[10] Bochkovskiy A, Wang C Y, Liao H Y M. Yolov4: Optimal speed and accuracy of object detection [J]. arXiv preprint arXiv:2004.10934, 2020.

[11] Li C, Li L, Jiang H, et al. YOLOv6: A single-stage object detection framework for industrial applications [J]. arXiv preprint arXiv:2209.02976, 2022.

[12] Li M J, Wang H, Wan Z B. Surface defect detection of steel strips based on improved YOLOv4 [J]. Computers and Electrical Engineering, 2022, 102: 108208.

[13] Yi C C, Xu B, Chen J, et al. An improved yolox model for detecting strip surface defects [J]. steel research international, 2022, 93(11): 2200505.

[14] Wang Z, Liu W S. Surface defect detection algorithm for strip steel based on improved YOLOv7 model [J]. IAENG International Journal of Computer Science, 2024, 51(3):308-316. yolo

[15] Wu T, Tang S, Zhang R, et al. Cgnet: A light-weight context guided network for semantic segmentation [J]. IEEE Transactions on Image Processing, 2020, 30: 1169-1179. (CGB)

[16] Shen, Chunhua, et al. "Adaptively spatial feature fusion f or object detection." Pattern Recognition Letters 137 (20 19): 27-37. ASFF

[17] Lin T Y, Dollár P, Girshick R, et al. Feature pyramid networks for object detection [C]. Proceedings of the IEEE conference on computer vision and pattern recognition. 2017: 2117-2125.shapeiou

[18] He Yun arty. Research on small target detection algorithm based on deep learning [D]. Jilin University, 2024. DOI:10.27162/d.cnki.gjlin.2024.004707.