MSCNN

A Python 3 and Keras 2 implementation of MSCNN for people countingand provide train method.

Requirement

• Python 3.6
• Keras 2.2.2
• Tensorflow-gpu 1.8.0
• OpenCV 3.4

MSCNN and MSB architectures

MSCNN

MSB

Experiment

data

Mall Dataset crowd counting dataset

Generate density_map from data:

train

run the follow command:

python train.py --size 224 --batch 16 --epochs 100

test in dataset

real count:30
pred count:27

Reference

@article{MSCNN,  
  title={Multi-scale convolutional neural network for crowd counting},  
  author={Lingke Zeng, Xiangmin Xu, Bolun Cai, Suo Qiu, Tong Zhang},
  journal={2017 IEEE International Conference on Image Processing (ICIP)},
  year={2017}
}

Copyright

See LICENSE for details.