Camera Style Adaptation for Person Re-identification CVPR 2018
Switch branches/tags
Nothing to show
Clone or download
Latest commit eb74509 Aug 8, 2018
Permalink
Failed to load latest commit information.
CycleGAN-for-CamStyle Update test.py Aug 8, 2018
reid add reranking Mar 21, 2018
README.md Update README.md Jul 13, 2018
main.py fix bug of camstyle_dataloader May 21, 2018

README.md

Camera Style Adaptation for Person Re-identification

================================================================

Code for Camera Style Adaptation for Person Re-identification (CVPR 2018).

Preparation

Requirements: Python=3.6 and Pytorch>=0.3.0

  1. Install Pytorch

  2. Download dataset

  3. Download CamStyle Images

    • Market-1501-Camstyle [GoogleDriver]

    • DukeMTMC-reID-CamStyle [GoogleDriver]

    • Move them to 'CamStyle/data/market (or duke)/bounding_box_train_camstyle'

CamStyle Generation

You can generate CamStyle imgaes with CycleGAN-for-CamStyle

Training and test re-ID model

  1. IDE
# For Market-1501
python main.py -d market --logs-dir logs/market-ide
# For Duke
python main.py -d duke --logs-dir logs/duke-ide
  1. IDE + CamStyle
# For Market-1501
python main.py -d market --logs-dir logs/market-ide-camstyle --camstyle 46
# For Duke
python main.py -d duke --logs-dir logs/duke-ide--camstyle --camstyle 46
  1. IDE + CamStyle + Random Erasing[4]
# For Market-1501
python main.py -d market --logs-dir logs/market-ide-camstyle-re --camstyle 46 --re 0.5
# For Duke
python main.py -d duke --logs-dir logs/duke-ide--camstyle-re --camstyle 46 --re 0.5
  1. IDE + CamStyle + Random Erasing[4] + re-ranking[3]
# For Market-1501
python main.py -d market --logs-dir logs/market-ide-camstyle-re --camstyle 46 --re 0.5 --rerank
# For Duke
python main.py -d duke --logs-dir logs/duke-ide--camstyle-re --camstyle 46 --re 0.5 --rerank

 

Results

Market-1501 Duke
Methods Rank-1 mAP Rank-1 mAP
IDE 85.6 65.8 72.3 51.8
IDE+CamStyle 88.1 68.7 75.2 53.4
IDE+CamStyle+Random Erasing 89.4 71.5 78.3 57.6

References

  • [1] Our code is conducted based on open-reid

  • [2] Unpaired Image-to-Image Translation using Cycle-Consistent Adversarial Networks, ICCV 2017

  • [3] Re-ranking Person Re-identification with k-reciprocal Encoding. CVPR 2017.

  • [4] Random Erasing Data Augmentation. Arxiv 2017.

Citation

If you find this code useful in your research, please consider citing:

@inproceedings{zhong2018camera,
title={Camera Style Adaptation for Person Re-identification},
author={Zhong, Zhun and Zheng, Liang and Zheng, Zhedong and Li, Shaozi and Yang, Yi},
booktitle={CVPR},
year={2018}
}

Contact me

If you have any questions about this code, please do not hesitate to contact me.

Zhun Zhong