2023-TII-SPG

This is the pytorch implementation of the paper.

Training

python train.py --cfg configs/prw.yaml
python train.py --cfg configs/cuhk_sysu.yaml

Evaluation

python train.py --eval --ckpt path_to_ckpt

Citation

@article{song2023learning,
  title={Learning Scene-Pedestrian Graph for End-to-End Person Search},
  author={Song, Zifan and Zhao, Cairong and Hu, Guosheng and Miao, Duoqian},
  journal={IEEE Transactions on Industrial Informatics},
  year={2023},
  publisher={IEEE}
}