Research on lightweight waste-surface detection algorithm based on improved YOLOv5s

Junyi Chen; Bin Yuan; Lingpeng Chen

doi:10.1117/12.3037813

12 September 2024 Research on lightweight waste-surface detection algorithm based on improved YOLOv5s

Junyi Chen, Bin Yuan, Lingpeng Chen

Proceedings Volume 13256, Fourth International Conference on Computer Vision and Pattern Analysis (ICCPA 2024); 1325624 (2024) https://doi.org/10.1117/12.3037813
Event: Fourth International Conference on Computer Vision and Pattern Analysis (ICCPA 2024), 2024, Anshan, China

Abstract

In the context of the embedded system for water-surface garbage collection devices, the existing model is insufficiently lightweight, resulting in suboptimal real-time detection of water-surface garbage. This study introduces a V5-MBCE algorithm specifically designed for water-surface garbage recognition. By substituting the YOLOv5s backbone network with the MobileNetV3 network, the model achieves a lightweight improvement. Moreover, the activation function within the MobileNetV3 network is modified to GELU, thereby enhancing the robustness of model training. To augment the feature fusion capacity of water-surface garbage at various scales, the feature fusion network is altered to incorporate a BiFPN structure. Additionally, the inclusion of the CBAM attention mechanism bolsters the model’s focus on detecting targets. The loss function has been refined to EIoU, allowing for more precise border positioning of the prediction box. Experimental results demonstrate that the proposed algorithm attains a detection accuracy of 94.4%,representing a 1.2 percentage point increase compared to the original model. Furthermore, the model size is reduced to 7.1MB, approximately 54% smaller than the original model. Finally, combining the v5-MEBC model and binocular ranging algorithm, the detection and collection of water-surface garbage can be realized in the collection device.

(2024) Published by SPIE. Downloading of the abstract is permitted for personal use only.

Citation Download Citation

Junyi Chen, Bin Yuan, and Lingpeng Chen "Research on lightweight waste-surface detection algorithm based on improved YOLOv5s", Proc. SPIE 13256, Fourth International Conference on Computer Vision and Pattern Analysis (ICCPA 2024), 1325624 (12 September 2024); https://doi.org/10.1117/12.3037813

ACCESS THE FULL ARTICLE

INSTITUTIONAL
Select your institution to access the SPIE Digital Library.

SELECT YOUR INSTITUTION

PERSONAL
Sign in with your SPIE account to access your personal subscriptions or to use specific features such as save to my library, sign up for alerts, save searches, etc.

PERSONAL SIGN IN

No SPIE Account? Create one

PURCHASE THIS CONTENT

SUBSCRIBE TO DIGITAL LIBRARY

50 downloads per 1-year subscription

Members: $195

Non-members: $335 ADD TO CART

25 downloads per 1 - year subscription

Members: $145

Non-members: $250 ADD TO CART

PURCHASE SINGLE ARTICLE

Includes PDF, HTML & Video, when available

Members: $17.00

Non-members: $21.00 ADD TO CART

PROCEEDINGS
11 PAGES

DOWNLOAD PAPER SAVE TO MY LIBRARY

GET CITATION

RIGHTS & PERMISSIONS

Get copyright permission Get copyright permission on Copyright Marketplace

KEYWORDS

Convolution

Detection and tracking algorithms

Object detection

Feature fusion

Education and training

Target detection

Performance modeling

Show All Keywords

Keywords/Phrases

Search In:

Publication Years