The original YOLOv5 model was used to detect the person and garbage. Thereafter the culprit was detected based on their distance from garbage.
Python 3.7 or later with all requirements.txt
dependencies installed, including torch >= 1.5
. To install run:
$ pip install -U -r requirements.txt