Home

Awesome

DDAM-PS: Diligent Domain Adaptive Mixer for Person Search -- WACV2024

This is the official implementation for our DDAM-PS: Diligent Domain Adaptive Mixer for Person Search accepted in WACV2024.

Paper (arXiv)

Highlights


Framework

framework

Performance :

SourceTargetmAPTop-1CKPT
PRWCUHK-SYSU79.581.3ckpt
CUHK-SYSUPRW36.781.2ckpt

Installation

run python setup.py develop to enable SPCL

Install Nvidia Apex

Run pip install -r requirements.txt in the root directory of the project.

Data Preparation

  1. Download CUHK-SYSU and PRW datasets, and unzip them.
  2. Or can be found at PS Datasets
  3. Modify configs/prw_da.yaml and configs/cuhk_sysu_da.yaml to change the dataset store place to your own path.

Testing

  1. Following the link in the above table, download our pretrained model to anywhere you like

  2. Evaluate its performance by specifying the paths of checkpoint and corresponding configuration file.

PRW as the target domain:

python train_da_dy_cluster.py --cfg configs/cuhk_sysu_da.yaml --eval --ckpt $MODEL_PATH

CUHK-SYSU as the target domain:

python train_da_dy_cluster.py --cfg configs/prw_da.yaml --eval --ckpt $MODEL_PATH

Training

PRW as the target domain:

python train_da_dy_cluster.py --cfg configs/cuhk_sysu_da.yaml

CUHK-SYSU as the target domain:

python train_da_dy_cluster.py --cfg configs/prw_da.yaml

Contact

If you have any questions, please create an issue on this repository or contact us at mustansar.fiaz@ibm.com

References

Our code is based on DAPS repositories. We thank them for releasing their baseline code.

Citation

@inproceedings{fiaz2024ddam_ps,
  title={DDAM-PS: Diligent Domain Adaptive Mixer for Person Search},
  author={Almansoori, Mohammed Khaleed  and Fiaz, Mustansar and Cholakkal, Hisham},
  booktitle={Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision},
  year={2024}
}
@inproceedings{fiaz2023sat,
  title={SAT: Scale-Augmented Transformer for Person Search},
  author={Fiaz, Mustansar and Cholakkal, Hisham and Anwar, Rao Muhammad and Khan, Fahad Shahbaz},
  booktitle={Proceedings of the IEEE/CVF Winter Conference on Applications of Computer Vision},
  year={2023}
}
@inproceedings{fiaz2022psarm,
  title={PS-ARM: An End-to-End Attention-aware Relation Mixer Network for Person Search},
  author={Fiaz, Mustansar and Cholakkal, Hisham and Narayan, Sanath and Anwar, Rao Muhammad and Khan, Fahad Shahbaz},
  booktitle={Proceedings of the ACCV Asian Conference on Computer Vision}, 
  year={2022}
}