Awesome
ICE
This is the official PyTorch implementation of the ICCV 2021 paper ICE: Inter-instance Contrastive Encoding for Unsupervised Person Re-identification.
Installation
git clone https://github.com/chenhao2345/ICE
cd ICE
python setup.py develop
Prepare Datasets
Download the raw datasets DukeMTMC-reID, Market-1501, MSMT17, and then unzip them under the directory like
ICE/examples/data
├── dukemtmc-reid
│ └── DukeMTMC-reID
├── market1501
└── msmt17
└── MSMT17_V1(or MSMT17_V2)
Training
We used 4 GPUs to train our model.
Train Market-1501:
python examples/unsupervised_train.py --dataset-target market1501
Train DukeMTMC-reID:
python examples/unsupervised_train.py --dataset-target dukemtmc-reid
Train MSMT17:
python examples/unsupervised_train.py --dataset-target msmt17
Citation
If you find this project useful, please kindly star our project and cite our paper.
@InProceedings{Chen_2021_ICCV,
author = {Chen, Hao and Lagadec, Benoit and Bremond, Fran\c{c}ois},
title = {ICE: Inter-Instance Contrastive Encoding for Unsupervised Person Re-Identification},
booktitle = {Proceedings of the IEEE/CVF International Conference on Computer Vision (ICCV)},
month = {October},
year = {2021},
pages = {14960-14969}
}