Open Access iconOpen Access

ARTICLE

crossmark

Lightweight Surface Litter Detection Algorithm Based on Improved YOLOv5s

Zunliang Chen1,2, Chengxu Huang1,2, Lucheng Duan1,2, Baohua Tan1,2,*

1 College of Science (College of Chip Industry), Hubei University of Technology, Wuhan, 430068, China
2 National “111 Research Center” Microelectronics and Integrated Circuits, Hubei University of Technology, Wuhan, 430068, China

* Corresponding Author: Baohua Tan. Email: email

Computers, Materials & Continua 2023, 76(1), 1085-1102. https://doi.org/10.32604/cmc.2023.039451

Abstract

In response to the problem of the high cost and low efficiency of traditional water surface litter cleanup through manpower, a lightweight water surface litter detection algorithm based on improved YOLOv5s is proposed to provide core technical support for real-time water surface litter detection by water surface litter cleanup vessels. The method reduces network parameters by introducing the deep separable convolution GhostConv in the lightweight network GhostNet to substitute the ordinary convolution in the original YOLOv5s feature extraction and fusion network; introducing the C3Ghost module to substitute the C3 module in the original backbone and neck networks to further reduce computational effort. Using a Convolutional Block Attention Mechanism (CBAM) module in the backbone network to strengthen the network’s ability to extract significant target features from images. Finally, the loss function is optimized using the Focal-EIoU loss function to improve the convergence speed and model accuracy. The experimental results illustrate that the improved algorithm outperforms the original Yolov5s in all aspects of the homemade water surface litter dataset and has certain advantages over some current mainstream algorithms in terms of model size, detection accuracy, and speed, which can deal with the problems of real-time detection of water surface litter in real life.

Keywords


Cite This Article

APA Style
Chen, Z., Huang, C., Duan, L., Tan, B. (2023). Lightweight surface litter detection algorithm based on improved yolov5s. Computers, Materials & Continua, 76(1), 1085-1102. https://doi.org/10.32604/cmc.2023.039451
Vancouver Style
Chen Z, Huang C, Duan L, Tan B. Lightweight surface litter detection algorithm based on improved yolov5s. Comput Mater Contin. 2023;76(1):1085-1102 https://doi.org/10.32604/cmc.2023.039451
IEEE Style
Z. Chen, C. Huang, L. Duan, and B. Tan "Lightweight Surface Litter Detection Algorithm Based on Improved YOLOv5s," Comput. Mater. Contin., vol. 76, no. 1, pp. 1085-1102. 2023. https://doi.org/10.32604/cmc.2023.039451



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.
  • 813

    View

  • 533

    Download

  • 0

    Like

Share Link