/yolov5

YOLOv5 🚀 in PyTorch > ONNX > CoreML > TFLite

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

First run

pip install -r requirements.txt

Next download last.pt from the following link, create weights folder in the repo's home directory, and put the downloaded file in it.

Finally, run the following command

python detect.py --source <0 for webcam, else path/to/dataset_directory>