This model is inspired by yolov5-crowdhuman and it counts heads and people in images and videos. The model was evaluated in Mall Dataset
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