MDBNet

This is the PyTorch version repo for Static Crowd Scene Analysis via Deep Network with Multi-branch Dilated Convolution Blocks

Prerequisites

Python: 2 or 3
PyTorch: 0.4.1

Ground Truth

Please follow the make_dataset.ipynb to generate the ground truth (In the python2). It shall take some time to generate the dynamic ground truth. Note you need to generate your own json file.

Train

Try python train.py train.json val.json 0 0 to start training process.(python3)

Test

Follow the RunTest.py.(python3)