[TPAMI] Locating and Counting Heads in Crowds With a Depth Prior [Paper]
Download or generate the virtual dataset from the ShanghaiTechRGBDSyn repository.
Download ShanghaiTechRGBD dataset from OneDrive.
By using SG-GAN.
CUDA_VISIBLE_DEVICES=0,1,2,3 python ./tools/train_GTAV_metric.py \
--dataset GTAV --dataroot /group/crowd_counting/GTAV-ours/ \
--cfg_file lib/configs/resnext50_32x4d_GTAV --lr 0.05 --batchsize 8
If you have high capacity GPUs, we recommend training with larger size images.
python ./tools/test_RGBD_metric.py \
--dataset RGBD --dataroot /p300/data/Dataset/SIST_RGBD/RGBDmerge_540P/Part_A/ \
--cfg_file lib/configs/resnext50_32x4d_GTAV \
--load_ckpt checkpoint.pth
This repository borrows partially from VNL and MAML.
If you find this repository useful for your research, please use the following:
@article{lian2021locating,
title={Locating and Counting Heads in Crowds With a Depth Prior},
author={Lian, Dongze and Chen, Xianing and Li, Jing and Luo, Weixin and Gao, Shenghua},
journal={IEEE Transactions on Pattern Analysis and Machine Intelligence},
year={2021},
publisher={IEEE}
}