Steel Surface Defect Detection Algorithm based on Improved YOLOv8

Authors

  • Shuai Yang

DOI:

https://doi.org/10.62051/ijmee.v3n3.01

Keywords:

YOLOv8, Steel Surface Defects, SimAM, Wise-IoU

Abstract

Aiming at the problem of steel surface defects, a defect detection algorithm based on YOLOv8 is constructed. Firstly, SimAM is added to the head to improve the expression ability of the features to enhance the detection ability of the model for tiny defects or small targets. Then the CIoU loss function is replaced with Wise-IoU to enhance the detection accuracy of the model. The experimental results show that the constructed improved models P, mAP0.5 and mAP0.5:0.95 reach 74.6%, 75.6% and 43.4%, respectively, which are improved by 5.9%, 0.5% and 0.6%, respectively, compared with the original YOLOv8n model. The detection accuracy was effectively improved.

References

[1] Changlu Xu, Linsheng Li, Jiwei Li. “Defect detection method for lithium battery pole piece based on the fusion of Gamma correction and LOG algorithm”, Mechanical Design and Research, Vol. 36, p. 105-109, 2020.

[2] W. W. Li, Q. Li, C. Gao. “An unmanned vehicle target detection algorithm based on multi-scale feature fusion”, Journal of Beijing University of Information Science and Technology (Natural Science Edition), Vol. 36, p. 19-23, 2021.

[3] Jigao Xu, Jiangyi Chen, Dongchen Qin, et al. “Research on steel surface defect detection based on CNN and CBAM”, Mechanical Design and Manufacturing, p. 1-5, 2024. https://doi.org/10.19356/j.cnki.1001-3997.20240725.016.

[4] Zhao W, Chen F, Huang H, et al. “A new steel defect detection algorithm based on deep learning”, Computational Intelligence and Neuroscience, p. 1-13, 2021.

[5] Kou X, Liu S, Cheng K, et al. “Development of a YOLO-V3-based model for detecting defects on steel strip surface”, Measurement, Vol. 182, p.109454, 2021. https://doi.org/10.1016/j.measurement.2021.109454.

[6] Wang L, Liu X, Ma J, et al. “Real-Time Steel Surface Defect Detection with Improved Multi-Scale YOLO-v5”, Processes, Vol. 11, p. 1357, 2023.

[7] ZHENG Z,WANG P,LIU W, et al. “Distance-IoU loss: Faster and better learning for bounding box regression”, arXiv, 2019.DOI:10.1609/aaai. v34i07.6999.

[8] Li Y T, Fan Q S, Huang H S, et al. “A modified YOLOv8 detection network for UAV aerial image recognition”, Drones, Vol. 7, p. 304, 2023.

Downloads

Published

31-10-2024

Issue

Section

Articles