Comparative Evaluation of Deep Learning Models for Real-Time Waste Classification
DOI:
https://doi.org/10.29207/resti.v10i4.7455Keywords:
deep learning, EfficientNet, MobileNetV2, Raspberry Pi, single shot multibox detector (SSD), waste management, YOLOv8Abstract
Waste management has become a pressing global challenge due to rapid urbanization and population growth, leading to increased environmental pollution and resource depletion. Automating waste segregation using deep learning can significantly improve recycling efficiency and reduce manual labor. In this study, five deep learning models: YOLOv8, YOLOv5, MobileNetV2, Single Shot Multibox Detector (SSD), and Google’s pre-trained EfficientNet, were evaluated for real-time waste detection and classification. A dataset consisting of 15,150 images across 12 waste categories was used for training and testing, representing common household waste materials and evaluated using real-world objects such as water bottles, glass bottles, cans, and used face masks. Testing was conducted on both a high-performance workstation and a Raspberry Pi 4 edge device to assess detection accuracy, inference speed, and practical deployment feasibility. The results indicate that YOLOv8 achieved the highest average accuracy of 83% with reliable performance across diverse object types, whereas EfficientNet achieved moderate accuracy of approximately 60% with higher inference latency. Lightweight models such as MobileNetV2 and SSD were computationally efficient but exhibited lower accuracy, at 42% and 41%, respectively, particularly when handling irregular or overlapping objects. These findings demonstrate that YOLOv8 provides the most effective balance between accuracy and real-time performance, making it well suited for intelligent waste sorting systems.
Downloads
References
S. Kaza, L. C. Yao, P. Bhada-Tata, and F. Van Woerden, What a Waste 2.0: A Global Snapshot of Solid Waste Management to 2050. Washington, DC: World Bank, 2018. doi: 10.1596/978-1-4648-1329-0.
M. Castro-Bello et al., “Convolutional Neural Network Models in Municipal Solid Waste Classification: Towards Sustainable Management,” Sustainability, vol. 17, no. 8, p. 3523, Apr. 2025, doi: 10.3390/su17083523.
V. Ruiz, Á. Sánchez, J. F. Vélez, and B. Raducanu, “Automatic Image-Based Waste Classification,” 2019, pp. 422–431. doi: 10.1007/978-3-030-19651-6_41.
Q. Zhang et al., “Recyclable waste image recognition based on deep learning,” Resour. Conserv. Recycl., vol. 171, p. 105636, Aug. 2021, doi: 10.1016/j.resconrec.2021.105636.
L. Yong, L. Ma, D. Sun, and L. Du, “Application of MobileNetV2 to waste classification,” PLoS One, vol. 18, no. 3, p. e0282336, Mar. 2023, doi: 10.1371/journal.pone.0282336.
Rima Dias Ramadhani, A. Nur Aziz Thohari, C. Kartiko, A. Junaidi, T. Ginanjar Laksana, and N. Alim Setya Nugraha, “Optimasi Akurasi Metode Convolutional Neural Network untuk Identifikasi Jenis Sampah,” J. RESTI (Rekayasa Sist. dan Teknol. Informasi), vol. 5, no. 2, pp. 312–318, Apr. 2021, doi: 10.29207/resti.v5i2.2754.
Aulia Afifah, E. R. Arumi, M. Maimunah, and Setiya Nugroho, “Comparative Evaluation of Preprocessing Methods for MobileNetV1 and V2 in Waste Classification,” J. RESTI (Rekayasa Sist. dan Teknol. Informasi), vol. 9, no. 3, pp. 444–452, May 2025, doi: 10.29207/resti.v9i3.6211.
B. Wang, Y. Xing, N. Wang, and C. L. P. Chen, “Monitoring Waste From Uncrewed Aerial Vehicles and Satellite Imagery Using Deep Learning Techniques: A Review,” IEEE J. Sel. Top. Appl. Earth Obs. Remote Sens., vol. 17, pp. 20064–20079, 2024, doi: 10.1109/JSTARS.2024.3488056.
R. Risfendra, G. F. Ananda, and H. Setyawan, “Deep Learning-Based Waste Classification with Transfer Learning Using EfficientNet-B0 Model,” J. RESTI (Rekayasa Sist. dan Teknol. Informasi), vol. 8, no. 4, pp. 535–541, Aug. 2024, doi: 10.29207/resti.v8i4.5875.
W. Liu et al., “SSD: Single Shot MultiBox Detector,” 2016, pp. 21–37. doi: 10.1007/978-3-319-46448-0_2.
W. Ma, X. Wang, and J. Yu, “A Lightweight Feature Fusion Single Shot Multibox Detector for Garbage Detection,” IEEE Access, vol. 8, pp. 188577–188586, 2020, doi: 10.1109/ACCESS.2020.3031990.
L. A. Darisang, S. K. Dwisetio, I. S. Edbert, and A. Aulia, “Real-Time Waste Detection Using YOLO, SSD, and Faster R-CNN Integrated with CNN-Based Classification,” in Proceedings of the 1st International Conference on Research and Innovations in Information and Engineering Technology, SCITEPRESS - Science and Technology Publications, 2025, pp. 76–83. doi: 10.5220/0014276000004928.
L. Tan, H. Wu, Z. Xu, and J. Xia, “Multi-object garbage image detection algorithm based on SP-SSD,” Expert Syst. Appl., vol. 263, p. 125773, Mar. 2025, doi: 10.1016/j.eswa.2024.125773.
W. Lin, “YOLO-Green: A Real-Time Classification and Object Detection Model Optimized for Waste Management,” in 2021 IEEE International Conference on Big Data (Big Data), IEEE, Dec. 2021, pp. 51–57. doi: 10.1109/BigData52589.2021.9671821.
G. A. Sampedro, R. G. C. Kim, Y. J. Aruan, D.-S. Kim, and J.-M. Lee, “Smart E-Waste Bin Development Based on YOLOv4 Model,” in 2021 1st International Conference in Information and Computing Research (iCORE), IEEE, Dec. 2021, pp. 125–128. doi: 10.1109/iCORE54267.2021.00041.
A. Aishwarya, P. Wadhwa, O. Owais, and V. Vashisht, “A Waste Management Technique to detect and separate Non-Biodegradable Waste using Machine Learning and YOLO algorithm,” in 2021 11th International Conference on Cloud Computing, Data Science & Engineering (Confluence), IEEE, Jan. 2021, pp. 443–447. doi: 10.1109/Confluence51648.2021.9377163.
R. S. Midigudla, T. Dichpally, U. Vallabhaneni, Y. Wutla, D. M. Sundaram, and S. Jayachandran, “A comparative analysis of deep learning models for waste segregation: YOLOv8, EfficientDet, and Detectron 2,” Multimed. Tools Appl., vol. 84, no. 29, pp. 35941–35964, Jan. 2025, doi: 10.1007/s11042-025-20647-y.
M. M. Abo-Zahhad and M. Abo-Zahhad, “Real time intelligent garbage monitoring and efficient collection using Yolov8 and Yolov5 deep learning models for environmental sustainability,” Sci. Rep., vol. 15, no. 1, p. 16024, May 2025, doi: 10.1038/s41598-025-99885-x.
P. K. Sarswat, R. S. Singh, and S. V. S. H. Pathapati, “Real time electronic-waste classification algorithms using the computer vision based on Convolutional Neural Network (CNN): Enhanced environmental incentives,” Resour. Conserv. Recycl., vol. 207, p. 107651, Aug. 2024, doi: 10.1016/j.resconrec.2024.107651.
R. S. Singh, S. V. S. H. Pathapati, M. L. Free, and P. K. Sarswat, “Identification and Separation of E‐Waste Components Using Modified Image Recognition Model Based on Advanced Deep Learning Tools,” in Technology Innovation for the Circular Economy, Wiley, 2024, pp. 115–127. doi: 10.1002/9781394214297.ch10.
V. Deodhar, R. Bhogaonkar, S. Patankar, and H. Dhabale, “Study of Automated E-Waste Classification Techniques,” 2024, pp. 325–342. doi: 10.1007/978-3-031-61475-0_26.
S. Jebakani, A. S. Robert, K. Harini, J. M., and A. R., "Smart Waste Management System Using Machine Learning," in Proc. 2026 4th International Conference on Knowledge Engineering and Communication Systems (ICKECS), Chickaballapura, India, Apr. 2026, https://doi.org/10.1109/ICKECS70176.2026.11527692.
M. S. Nafiz, S. S. Das, M. K. Morol, A. Al Juabir, and D. Nandi, “ConvoWaste: An Automatic Waste Segregation Machine Using Deep Learning,” in 2023 3rd International Conference on Robotics, Electrical and Signal Processing Techniques (ICREST), IEEE, Jan. 2023, pp. 181–186. doi: 10.1109/ICREST57604.2023.10070078.
J. Gunaseelan, S. Sundaram, and B. Mariyappan, “A Design and Implementation Using an Innovative Deep-Learning Algorithm for Garbage Segregation,” Sensors, vol. 23, no. 18, p. 7963, Sep. 2023, doi: 10.3390/s23187963.
I. Jahan, M. F. Ahmed Rudro, W. R. Parsub, S. I. Swapnil, and A. W. Reza, “Deep Learning-Driven Real-Time Visual Pollution Detection and Multi-Class Waste Classification in Urban and Textile Landscapes,” Procedia Comput. Sci., vol. 252, pp. 529–538, 2025, doi: 10.1016/j.procs.2025.01.012.
M. S. Hossen, A. J. Shoukhin, N. H. Athin, A. Chakma, and M. M. Islam, “Automated Waste Management: A Deep Learning and Cloud Based Food & Plastic Waste Separator Bin,” in 2024 6th International Conference on Sustainable Technologies for Industry 5.0 (STI), IEEE, Dec. 2024, pp. 1–6. doi: 10.1109/STI64222.2024.10951149.
G. Thung and M. Yang, "Classification of Trash for Recyclability Status," Stanford University, CS229 Project Report, 2016. [Online]. Available: https://github.com/garythung/trashnet
Downloads
Published
Issue
Section
License
Copyright in each article belongs to the author
- The author acknowledges that the RESTI Journal (System Engineering and Information Technology) is the first publisher to publish with a license Creative Commons Attribution 4.0 International License.
- Authors can enter writing separately, arrange the non-exclusive distribution of manuscripts that have been published in this journal into other versions (eg sent to the author's institutional repository, publication in a book, etc.), by acknowledging that the manuscript has been published for the first time in the RESTI (Rekayasa Sistem dan Teknologi Informasi) journal ;
