OpenUnReID
is an open-source PyTorch-based codebase for both unsupervised learning (USL) and unsupervised domain adaptation (UDA) in the context of object re-ID tasks. It provides strong baselines and multiple state-of-the-art methods with highly refactored codes for both pseudo-label-based and domain-translation-based frameworks. It works with Python >=3.5 and PyTorch >=1.1.
We are actively updating this repo, and more methods will be supported soon. Contributions are welcome.
- Distributed training & testing with multiple GPUs and multiple machines.
- High flexibility on various combinations of datasets, backbones, losses, etc.
- GPU-based pseudo-label generation and k-reciprocal re-ranking with quite high speed.
- Plug-and-play domain-specific BatchNorms for any backbones, sync BN is also supported.
- A strong cluster baseline, providing high extensibility on designing new methods.
- State-of-the-art methods and performances for both USL and UDA problems on object re-ID.
Please refer to MODEL_ZOO.md for trained models and download links.
Method | Reference | USL | UDA |
---|---|---|---|
UDA_TP | PR'20 (arXiv'18) | ✓ | ✓ |
SPGAN | CVPR'18 | n/a | ongoing |
SSG | ICCV'19 | ongoing | ongoing |
strong_baseline | Sec. 3.1 in ICLR'20 | ✓ | ✓ |
MMT | ICLR'20 | ✓ | ✓ |
SpCL | arXiv'20 | ✓ | ✓ |
SDA | arXiv'20 | n/a | ongoing |
- [2020-07-01]
OpenUnReID
v0.1.0 is released.
Please refer to INSTALL.md for installation and dataset preparation.
Please refer to GETTING_STARTED.md for the basic usage of OpenUnReID
.
OpenUnReID
is released under the Apache 2.0 license.
If you use this toolbox or models in your research, please consider cite:
@inproceedings{ge2020mutual,
title={Mutual Mean-Teaching: Pseudo Label Refinery for Unsupervised Domain Adaptation on Person Re-identification},
author={Yixiao Ge and Dapeng Chen and Hongsheng Li},
booktitle={International Conference on Learning Representations},
year={2020},
url={https://openreview.net/forum?id=rJlnOhVYPS}
}
@misc{ge2020selfpaced,
title={Self-paced Contrastive Learning with Hybrid Memory for Domain Adaptive Object Re-ID},
author={Yixiao Ge and Dapeng Chen and Feng Zhu and Rui Zhao and Hongsheng Li},
year={2020},
eprint={2006.02713},
archivePrefix={arXiv},
primaryClass={cs.CV}
}
Some parts of openunreid
are learned from torchreid and fastreid. We would like to thank for their projects, which have boosted the research of supervised re-ID a lot. We hope that OpenUnReID
could well benefit the research community of unsupervised re-ID by providing strong baselines and state-of-the-art methods.
This project is developed by Yixiao Ge (@yxgeee), Tong Xiao (@Cysu), Zhiwei Zhang (@Dreamerzzw).