Open-source stronger baseline for unsupervised or domain adaptive object re-ID. We will udpate the strong baseline and group-aware label transfer method in domain adaptive person re-identifacation.

*Our method only adopts the clustering algorithm and ReID baseline model with the moving average model.

UDAStrongBaseline is a transitional code based pyTorch framework for both unsupervised learning (USL) and unsupervised domain adaptation (UDA) in the context of object re-ID tasks. It provides stronger baselines on these tasks. It needs the enviorment: Python >=3.6 and PyTorch >=1.1. We will transfer all the codes to the fastreid in the future (ongoing).

• Direct infer models are trained on the source-domain datasets (source_pretrain) and directly tested on the target-domain datasets.
• UDA methods (MMT, SpCL, etc.) starting from ImageNet means that they are trained end-to-end in only one stage without source-domain pre-training. MLT denotes to the implementation of our NeurIPS-2020. Please note that it is a pre-released repository for the anonymous review process, and the official repository will be released upon the paper published.

git https://github.com/zkcys001/UDAStrongBaseline/
cd UDAStrongBaseline
pip install -r requirements
pip install faiss-gpu==1.6.3

Download the person datasets DukeMTMC-reID, Market-1501, MSMT17, Then unzip them under the directory like

./data
├── dukemtmc
│  └── DukeMTMC-reID
├── market1501
│  └── Market-1501-v15.09.15
├── msmt17
   └── MSMT17_V1

You can create the soft link to the dataset:

ln -s /path-to-data ./data

ImageNet-pretrained models for ResNet-50 will be automatically downloaded in the python script.

We utilize 4 GPUs for training. Note that

To train the uncertainty model in the source domain, run this command:

sh pretrain_uncertainty_market1501.sh

Utilizeing the GLT model based on DBSCAN clustering algorithm:

sh scripts/dbscan_uncertainty_duke2market.sh

To train the GLT model in the source domain, run this command:

sh scripts/pretrain_market1501.sh

Utilizeing the GLT model based on K-means clustering algorithm:

sh scripts/GLT_kmeans_duke2market.sh

To train the baseline in the source domain, run this command:

sh scripts/pretrain_market1501.sh

Utilizeing the baseline or uncertainty model(s) based on DBSCAN clustering algorithm:

sh scripts/dbscan_baseline_market2duke.sh

Some parts of UDAstrongbaseline are from MMT and fastreid. We would like to thank for these projects, and we will update our method .

If you find this code useful for your research, please use the following BibTeX entry.

@article{zheng2020exploiting,
  title={Exploiting Sample Uncertainty for Domain Adaptive Person Re-Identification},
  author={Zheng, Kecheng and Lan, Cuiling and Zeng, Wenjun and Zhang, Zhizheng and Zha, Zheng-Jun},
  journal={AAAI},
  year={2021}
}

@article{zheng2021labeltransfer,
  title={Group-aware Label Transfer for Domain Adaptive Person Re-identification},
  author={Zheng, Kecheng and Liu, Wu and Mei, Tao and Luo, Jiebo and Zha, Zheng-Jun},
  journal={CVPR},
  year={2021}
}