Crowd-Counting-YOLOV5

This model is inspired by yolov5-crowdhuman and it counts heads and people in images and videos. The model was evaluated in Mall Dataset

A video from results on Mall Dataset

Test

Please download the network weights from here, then copy it in Crowd-Counting-YOLOV5 directory

Download and extract Mall dataset

cd Crowd-Counting-YOLOV5
mkdir test-img
wget https://personal.ie.cuhk.edu.hk/~ccloy/files/datasets/mall_dataset.zip
unzip -q mall_dataset.zip -d test-img/

Run

cd Crowd-Counting-YOLOV5
python3 detect.py --weights crowdhuman_yolov5m.pt --source test-img/mall_dataset/frames/ --view-img --save-txt --heads

An ipynb file is provided for running this network on google colab