YOLO-Green: A Real-Time Classification and Object Detection Model Optimized for Waste Management
Wesley Lin
Abstract
Deep neural networks (DNNs) play an important role in our daily lives, from aiding us in menial tasks to solving world issues such as cancer cell detection. However, few pieces of research have been conducted using DNNs and deep learning models as a medium to help classify and detect trash, in efforts to solve our global waste crisis. This is because current DNNs struggle to be both efficient and accurate while detecting indistinct objects such as waste. To address this issue, this work focuses on YOLO-Green, a novel real-time object detection model designed specifically for trash detection. The model is trained on a dataset gathered from real-world trash divided into seven of the most common types of solid waste. With only 100 epochs of training, YOLO-Green achieves an outstanding mAP of 78.04%, FPS of 2.72, while retaining a model size of only 117 MB. Based on the original object detection of YOLOv4, YOLO-Green exceeds YOLOv4 and other popular deep learning models in both its accuracy and efficiency, while maintaining a relatively small model size. Ultimately, this study sheds a positive light on the potential of using deep learning models as an alternative to manual waste management.