Skip to content
master
Go to file
Code

Latest commit

 

Git stats

Files

Permalink
Failed to load latest commit information.
Type
Name
Latest commit message
Commit time
 
 
 
 
 
 

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

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

About

Camera Style Adaptation for Person Re-identification CVPR 2018

Topics

Resources

Releases

No releases published

Packages

No packages published
You can’t perform that action at this time.