Academic Journal of Computing & Information Science, 2023, 6(3); doi: 10.25236/AJCIS.2023.060312.
Shulin Li, Hong Zhao, Yuxi Huang
College of Mechanic & Control Engineer, Guilin University of Technology, Guilin, China
Defect detection is of great importance to ensure the quality of biscuit production. An improved YOLOv5 biscuit detection algorithm is proposed for the problems of poor real-time and low accuracy of biscuit defect detection methods. First, the number of C3s in the backbone network is reduced, and then the depth-separable convolution is used instead of the normal convolution in the network to reduce the model parameters and computation and improve the detection speed. Secondly, the SE attention module is added to the feature extraction layer to enhance the feature extraction capability of the backbone network and improve the accuracy of biscuit defect detection. Finally, the EIOU loss function is introduced to accelerate the model convergence and accurate target localization. The improved algorithm is tested on the self-built biscuit dataset, and the experimental results show that: the detection accuracy of the proposed algorithm can reach 99.2%, and the detection speed is 67 frames/s, which can meet the actual production requirements.
biscuit defect, YOLOv5, lightweight, attention mechanism
Shulin Li, Hong Zhao, Yuxi Huang. A biscit defect detection method based on improved YOLOv5. Academic Journal of Computing & Information Science (2023), Vol. 6, Issue 3: 92-101. https://doi.org/10.25236/AJCIS.2023.060312.
[1] Cheng JF, Fang GWS, Gao HF. Research progress of machine vision technology for surface defect detection [J/OL]. Application Research of Computers. 1-13[2023-02-26]. https: // doi. org/ 10. 19734/ j. issn. 1001-3695.2022.08.0426.
[2] Ren Shaoqing, He Kaiming, Girshick R, et al. Faster R-CNN: towards real-time object detection with region proposal networks [J]. IEEE Trans on Pattern Analysis and Machine Intelligence, 2017, 39 (6): 1137-1149.
[3] Liu Wei, Anguelov D, Erhan D, et al. SSD: single shot multibox detector [C]// European Conference on Computer Vision, [s.L.]: Springer, Cham, 2016: 21-37.
[4] Redmon J, Farhadi A. YOLO9000: better, faster, stronger [J]. 2017 IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2017: 6517-6525.
[5] Wan Guang et al. Ceramic tile surface defect detection based on deep learning [J]. Ceramics International, 2022, 48(8): 11085-11093.
[6] Ma Zhuxi, Li Yibo, Huang Minghui, et al. A lightweight detector based on attention mechanism for aluminum strip surface defect detection [J]. Computers in Industry, 2022, 136: 103585.
[7] Su Z, Han K, Song W, et al. Railway fastener defect detection based on improved YOLOv5 algorithm[C]// 2022 IEEE 6th Advanced Information Technology, Electronic and Automation Control Conference (IAEAC ). 0.
[8] Zheng Liaomo, Wang Xiaojie, Wang Qi, et al. A fabric defect detection method based on improved YOLOv5 [C]// 2021 the 7th International Conference on Computer and Communications (ICCC), Piscataway, NJ: IEEE, 2021: 620-624.
[9] Ultralytics. YOLOv5 [EB/OL]. [2022-4-10]. https:// github.co m/ultralytics/yolov5.
[10] HOWARD A G, ZHU M, CHEN B, et al. Mobilenets: Efficient convolutional neural networks for mobile vision applications [J]. International Journal of Computer Vision, 2017, 5(8): 122-131.
[11] HU J, SHEN L, SUN G. Squeeze-and-excitation networks[C]//Proceedings of the IEEE conference on computer vision and pattern recognition. 2018: 7132-7141.
[12] WOO S, PARK J, LEE J Y, et al. Cbam: Convolutional block attention module[C]//Proceedings of the European conference on computer vision (ECCV). 2018: 3-19.
[13] HOU Q, ZHOU D, FENG J. Coordinate attention for efficient mobile network design[C] // Proceedings of the IEEE/CVF conference on computer vision and pattern recognition. 2021: 13713-13722.
[14] ZHANG Y F, REN W, ZHANG Z, et al. Focal and efficient IOU loss for accurate bounding box regression [J]. Neurocomputing, 2022, 506: 146-157.
[15] Redmon J, Farhadi A. YOLOv3: an incremental improvement [EB/OL] [2022-5-25]. https://arxiv. org/pdf/1804.02767.pdf.
[16] Jiang X, Hu H, Liu X, et al. A smoking behavior detection method based on the YOLOv5 network[J]. 2022.