/NAE4PS

Norm-Aware Embedding for Efficient Person Search (CVPR'20)

Primary LanguagePythonMIT LicenseMIT

Norm-Aware Embedding for Efficient Person Search

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

Preparation

  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

Experiments

  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

Citation

@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}
    }