This is the PyTorch version repo for "Exploit the potential of Multi-column architecture for Crowd Counting", which delivered a state-of-the-art, straightforward and end-to-end architecture for crowd counting tasks. We also recommend another work on crowd counting(Deep Density-aware Count Regressor), which is accepted by ECAI2020.

ShanghaiTech Dataset

We strongly recommend Anaconda as the environment.

Python: 3.6

PyTorch: 1.5.0

1、python make_dataset.py # generate the ground truth. the ShanghaiTech dataset should be placed in the "datasets" directory.
2、python train.py # train model
3、python val.py # test model

partA: MAE 55.5 MSE 90.1

partB: MAE 6.8 MSE 10.7