» Articles » PMID: 39001117

An MRS-YOLO Model for High-Precision Waste Detection and Classification

Overview
Journal Sensors (Basel)
Publisher MDPI
Specialty Biotechnology
Date 2024 Jul 13
PMID 39001117
Authors
Affiliations
Soon will be listed here.
Abstract

With the advancement in living standards, there has been a significant surge in the quantity and diversity of household waste. To safeguard the environment and optimize resource utilization, there is an urgent demand for effective and cost-efficient intelligent waste classification methodologies. This study presents MRS-YOLO (Multi-Resolution Strategy-YOLO), a waste detection and classification model. The paper introduces the SlideLoss_IOU technique for detecting small objects, integrates RepViT of the Transformer mechanism, and devises a novel feature extraction strategy by amalgamating multi-dimensional and dynamic convolution mechanisms. These enhancements not only elevate the detection accuracy and speed but also bolster the robustness of the current YOLO model. Validation conducted on a dataset comprising 12,072 samples across 10 categories, including recyclable metal and paper, reveals a 3.6% enhancement in mAP50% accuracy compared to YOLOv8, coupled with a 15.09% reduction in volume. Furthermore, the model demonstrates improved accuracy in detecting small targets and exhibits comprehensive detection capabilities across diverse scenarios. For transparency and to facilitate further research, the source code and related datasets used in this study have been made publicly available at GitHub.

References
1.
Ren S, He K, Girshick R, Sun J . Faster R-CNN: Towards Real-Time Object Detection with Region Proposal Networks. IEEE Trans Pattern Anal Mach Intell. 2016; 39(6):1137-1149. DOI: 10.1109/TPAMI.2016.2577031. View

2.
Khan S, Anjum R, Raza S, Bazai N, Ihtisham M . Technologies for municipal solid waste management: Current status, challenges, and future perspectives. Chemosphere. 2021; 288(Pt 1):132403. DOI: 10.1016/j.chemosphere.2021.132403. View

3.
Jin S, Yang Z, Krolczykg G, Liu X, Gardoni P, Li Z . Garbage detection and classification using a new deep learning-based machine vision system as a tool for sustainable waste recycling. Waste Manag. 2023; 162:123-130. DOI: 10.1016/j.wasman.2023.02.014. View

4.
Chen L, Zhu J . Water surface garbage detection based on lightweight YOLOv5. Sci Rep. 2024; 14(1):6133. PMC: 10937728. DOI: 10.1038/s41598-024-55051-3. View

5.
Chen S, Huang J, Xiao T, Gao J, Bai J, Luo W . Carbon emissions under different domestic waste treatment modes induced by garbage classification: Case study in pilot communities in Shanghai, China. Sci Total Environ. 2020; 717:137193. DOI: 10.1016/j.scitotenv.2020.137193. View