/yolov5-crowdhuman

Head and Person detection using yolov5. Detection from crowd.

Primary LanguagePythonGNU General Public License v3.0GPL-3.0

Head & Person Detection Model

Download model trained on crowd human using yolov5(m) architeture

Download Link: YOLOv5m-crowd-human


Output (Crowd Human Model)

image


Test

$ python detect.py --weights crowdhuman_yolov5m.pt --source _test/ --view-img

Test (Only Person Class)

python3 detect.py --weights crowdhuman_yolov5m.pt --source _test/ --view-img  --person

Test (Only Heads)

python3 detect.py --weights crowdhuman_yolov5m.pt --source _test/ --view-img  --heads