This paper details the design and implementation of a garbage detection system utilizing YOLOv5, a cutting-edge, real-time object detection neural network. The system employs the pre-trained YOLOv5 Nano model, which has been adapted to a dataset specifically tailored for garbage detection. A significant challenge encountered during the study was the issue of model convergence, attributed to data inconsistencies in determining what constitutes garbage. To address this, a series of custom datasets were utilized in a trial-and-error approach, enabling the model to attain stability and achieve reliable performance metrics. The study further incorporates an MLOps pipeline, streamlining the processes of data collection, processing, model training, and testing. This pipeline optimizes the garbage detection model for deployment on edge devices, enabling cameras to capture video stream metadata for more efficient trash collection. By integrating AI-driven object detection with video metadata, the system promotes environmental sustainability through improved waste management practices.

错误:搜索内容不能为空,请输入英文关键词
错误:关键词超出字数限制,请精简
高级检索

Development of an AI-Powered Garbage Detection System for Environmental Sustainability Via YOLOv5

  • Lakshmi Narasimha Raju Mudunuri,
  • Venu Madhav Aragani,
  • Praveen Kumar Maroju

摘要

This paper details the design and implementation of a garbage detection system utilizing YOLOv5, a cutting-edge, real-time object detection neural network. The system employs the pre-trained YOLOv5 Nano model, which has been adapted to a dataset specifically tailored for garbage detection. A significant challenge encountered during the study was the issue of model convergence, attributed to data inconsistencies in determining what constitutes garbage. To address this, a series of custom datasets were utilized in a trial-and-error approach, enabling the model to attain stability and achieve reliable performance metrics. The study further incorporates an MLOps pipeline, streamlining the processes of data collection, processing, model training, and testing. This pipeline optimizes the garbage detection model for deployment on edge devices, enabling cameras to capture video stream metadata for more efficient trash collection. By integrating AI-driven object detection with video metadata, the system promotes environmental sustainability through improved waste management practices.