/YoloV5_GarbageLitterDetection

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

YoloV5_GarbageLitterDetection

Detecting the garbage litter in public places

The original YOLOv5 model was used to detect the person and garbage. Thereafter the culprit was detected based on their distance from garbage.


Requirements

Python 3.7 or later with all requirements.txt dependencies installed, including torch >= 1.5. To install run:

$ pip install -U -r requirements.txt

Demonstration video

Garbage_Litter_Detection