README.md

metadata

title: people-counting
app_file: app.py
sdk: gradio
sdk_version: 3.41.2

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