CVPR 2018 Exploit the Unknown Gradually: One-Shot Video-Based Person Re-Identification by Stepwise Learning
Switch branches/tags
Nothing to show
Clone or download
Latest commit d5d29c2 Jul 23, 2018
Permalink
Type Name Latest commit message Commit time
Failed to load latest commit information.
data Update README.md May 4, 2018
examples Update README.md May 5, 2018
logs add the training code Apr 13, 2018
reid Update README.md May 6, 2018
.gitignore add the training code Apr 13, 2018
LICENSE add the training code Apr 13, 2018
README.md Update README.md Jul 23, 2018
run.py Update README.md May 1, 2018
run.sh Update README.md May 5, 2018

README.md

Exploit the Unknown Gradually: One-Shot Video-Based Person Re-Identification by Stepwise Learning

Pytorch implementation for our paper [Link]. This code is based on the Open-ReID library.

Preparation

Dependencies

  • Python 3.6
  • PyTorch (version >= 0.2.0)
  • h5py, scikit-learn, metric-learn, tqdm

Download datasets

Train

For the DukeMTMC-VideoReID dataset:

python3 run.py --dataset DukeMTMC-VideoReID --logs_dir logs/DukeMTMC-VideoReID_EF_10/ --EF 10 --mode Dissimilarity --max_frames 900

For the MARS datasaet:

python3 run.py --dataset mars --logs_dir logs/mars_EF_10/ --EF 10 --mode Dissimilarity --max_frames 900

It takes about 10 hours to train EUG (EF=10%) on DukeMTMC-VideoReID with a GTX1080Ti. Please set the max_frames smaller if your GPU memory is less than 11G.

Performances

The performances varies according to random splits for initial labeled data. To reproduce the performances in our paper, please use the one-shot splits at ./examples/

Citation

Please cite the following paper in your publications if it helps your research:

@inproceedings{wu2018cvpr_oneshot,
    title = {Exploit the Unknown Gradually: One-Shot Video-Based Person Re-Identification by Stepwise Learning},
    author = {Wu, Yu and Lin, Yutian and Dong, Xuanyi and Yan, Yan and Ouyang, Wanli and Yang, Yi},
    booktitle = {The IEEE Conference on Computer Vision and Pattern Recognition (CVPR)},
    year = {2018}
}

Contact

To report issues for this code, please open an issue on the issues tracker.

If you have further questions about this paper, please do not hesitate to contact me.

Yu Wu's Homepage