Open Access iconOpen Access

ARTICLE

crossmark

A Lightweight Electronic Water Pump Shell Defect Detection Method Based on Improved YOLOv5s

Qunbiao Wu1, Zhen Wang1,*, Haifeng Fang1, Junji Chen1, Xinfeng Wan2

1 School of Mechanical Engineering, Jiangsu University of Science and Technology, Zhenjiang, 212000, China
2 Suzhou Ditian Robot Co., Ltd., Zhangjiagang, China

* Corresponding Author: Zhen Wang. Email: email

Computer Systems Science and Engineering 2023, 46(1), 961-979. https://doi.org/10.32604/csse.2023.036239

Abstract

For surface defects in electronic water pump shells, the manual detection efficiency is low, prone to misdetection and leak detection, and encounters problems, such as uncertainty. To improve the speed and accuracy of surface defect detection, a lightweight detection method based on an improved YOLOv5s method is proposed to replace the traditional manual detection methods. In this method, the MobileNetV3 module replaces the backbone network of YOLOv5s, depth-separable convolution is introduced, the parameters and calculations are reduced, and CIoU_Loss is used as the loss function of the boundary box regression to improve its detection accuracy. A dataset of electronic pump shell defects is established, and the performance of the improved method is evaluated by comparing it with that of the original method. The results show that the parameters and FLOPs are reduced by 49.83% and 61.59%, respectively, compared with the original YOLOv5s model, and the detection accuracy is improved by 1.74%, which is an indication of the superiority of the improved method. To further verify the universality of the improved method, it is compared with the results using the original method on the PASCALVOC2007 dataset, which verifies that it yields better performance. In summary, the improved lightweight method can be used for the real-time detection of electronic water pump shell defects.

Keywords


Cite This Article

APA Style
Wu, Q., Wang, Z., Fang, H., Chen, J., Wan, X. (2023). A lightweight electronic water pump shell defect detection method based on improved yolov5s. Computer Systems Science and Engineering, 46(1), 961-979. https://doi.org/10.32604/csse.2023.036239
Vancouver Style
Wu Q, Wang Z, Fang H, Chen J, Wan X. A lightweight electronic water pump shell defect detection method based on improved yolov5s. Comput Syst Sci Eng. 2023;46(1):961-979 https://doi.org/10.32604/csse.2023.036239
IEEE Style
Q. Wu, Z. Wang, H. Fang, J. Chen, and X. Wan, “A Lightweight Electronic Water Pump Shell Defect Detection Method Based on Improved YOLOv5s,” Comput. Syst. Sci. Eng., vol. 46, no. 1, pp. 961-979, 2023. https://doi.org/10.32604/csse.2023.036239



cc Copyright © 2023 The Author(s). Published by Tech Science Press.
This work is licensed under a Creative Commons Attribution 4.0 International License , which permits unrestricted use, distribution, and reproduction in any medium, provided the original work is properly cited.
  • 1201

    View

  • 573

    Download

  • 0

    Like

Share Link