waste_detection

The file has been saved in the model branches.

Prepare

1 Create a virtual environment with Python

2 Install pytorch ; torchvision

3 Install all dependencies

pip install -r requirements.txt

run

python detect_model.py --source [yourfile] --name [filename] --save-crop

Licence

please follow the LICENCE of YOLOv5!