This is unofficial and very simple implementation of the CVPR 2016 paper on crowd counting "Single-image crowd counting via multi-column convolutional neural network". Full code is developed on Tensorflow platform.
- Install Tensorflow
- Clone this repository
git clone https://github.com/aditya-vora/crowd_counting_tensorflow.git
-
Download the dataset and keep it in the $ROOT/data folder
-
The dataset can be downloaded from:
Dropbox: https://www.dropbox.com/s/fipgjqxl7uj8hd5/ShanghaiTech.zip?dl=0
Baidu Disk: http://pan.baidu.com/s/1nuAYslz
-
Train the model using $ROOT/train.py
-
Test the model
Note: More details about the results and some other code files for data parsing will be updated soon.
Citations:
[1] Zhang, Yingying, et al. "Single-image crowd counting via multi-column convolutional neural network." Proceedings of the IEEE conference on computer vision and pattern recognition. 2016.