deepakcrk / yolov5-crowdhuman

Head and Person detection using yolov5. Detection from crowd.

Head & Person Detection Model

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

Download Link: YOLOv5m-crowd-human

Output (Crowd Human Model)

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