CARE

Prepare Datasets

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

data
├── market1501
│   └── images ..
├── MSMT17
│   └── images ..
├── dukemtmcreid
    └── images ..

Prepare ViT Pre-trained Models

You need to download the ImageNet pretrained transformer model : ViT-Base

Training

python train.py --config_file configs/transformer_base.yml MODEL.DEVICE_ID "('your device id')" MODEL.STRIDE_SIZE ${1} MODEL.SIE_CAMERA ${2} MODEL.SIE_VIEW ${3} MODEL.JPM ${4} MODEL.TRANSFORMER_TYPE ${5} OUTPUT_DIR ${OUTPUT_DIR} DATASETS.NAMES "('your dataset name')"

# using following commands:
sh dist_train.sh