Official code for Motional Foreground Attention-based (MFA) Video Crowd Counting.
- Bus: BaiduNetDisk, GoogleDrive,
- Canteen: BaiduNetDisk, GoogleDrive,
- Classroom: BaiduNetDisk, GoogleDrive.
- Download the dataset.
- Change the path to the path which includes 'train.py'.
- Run 'pre_train.py' to conduct stage-1 training.
python pre_train.py (dataset path) (save path) amb (gpu id) (task name) -m (roi file path)
- Run 'train.py' to conduct stage-2 training.
python train.py (dataset path) (save path) (stage-1 model path) PreTrain (gpu id) (task name) -m (roi file path)
- Download the dataset.
- Change the path to the path which includes 'test.py'.
- Run 'test.py' to conduct testing.
python test.py (dataset path) (model weight path) (gpu id) -m (roi file path)