This repository hosts our code for our paper Norm-Aware Embedding for Efficient Person Search.

1. Clone this repo

   https://github.com/DeanChan/NAE4PS.git && cd NAE4PS

2. Build environment with conda

   conda env create --prefix <your_conda_env_path> -f environment.yml

1. Download CUHK-SYSU and PRW to data/

2. Download the trained model

   mkdir logs && wget https://github.com/DeanChan/NAE4PS/releases/download/v0.0.0/logs.tar.gz -P logs/ 
   cd logs && tar xvzf logs.tar.gz

3. Test

   CUDA_VISIBLE_DEVICES=0 python scripts/test_NAE.py -p logs/cuhk_sysu/

4. Train

   CUDA_VISIBLE_DEVICES=0 python scripts/train_NAE.py --debug --lr_warm_up -p ./logs/<your_logging_path>/ --batch_size 5 --nw 5 --w_RCNN_loss_bbox 10.0 --epochs 22 --lr 0.003

@inproceedings{chen2020norm,
  title={Norm-Aware Embedding for Efficient Person Search},
  author={Chen, Di and Zhang, Shanshan and Yang, Jian and Schiele, Bernt},
  booktitle={CVPR},
  year={2020}
}

@inproceedings{chen2020hoim,
      title={Hierarchical Online Instance Matching for Person Search},
      author={Chen, Di and Zhang, Shanshan and Ouyang, Wanli and Yang, Jian and Schiele, Bernt},
      booktitle={AAAI},
      year={2020}
    }