mPrompt

Crowd Counting via Segmenter-Regressor Mutual Prompt Learning

Training / Testing code of mPrompt is available.

Datasets Preparation

Download the datasets ShanghaiTech A, ShanghaiTech B, UCF-QNRF and NWPU. Then generate the density maps via gen_den_map.py.

Pre-train crowd segmentation model for generating pseudo mask

bash run_seg.sh

Train

bash run_train.sh

Test

Download the pretrained model via Link：https://pan.baidu.com/s/1wv4szaK1Z8UDKHOwi_ZtTA, Extract Password：bked

or

bash download_models.sh

And put the model into folder ./models/

bash run_test.sh