This file collect a set of pre-train models.
# train
python train.py --config_file ./base_config.yaml
# infer
python infer.py --config_file ./base_config.yaml --model_path /path/to/model.pth
Training a model with multiple datasets, makes it stronger and more generalizable
| Training Dataset | Test Dataset | map | rank-1 | rank-5 | rank-10 |
|---|---|---|---|---|---|
| DukeMTMC-reID Market-1501 MSMT17 CUHK03 WildTrack NJUST RAP(private) | DukeMTMCreID | 0.6824988661464477 | 0.835278276481149 | 0.914721723518851 | 0.9349192100538599 |
| Market1501 | 0.7929966369451702 | 0.9263657957244655 | 0.9756532066508313 | 0.9842636579572447 | |
| MSMT17 | 0.3968708072353888 | 0.7006604339994854 | 0.8137061497555537 | 0.8516167767389999 | |
| CUHK03 | 0.694728373609779 | 0.7307142857142858 | 0.875 | 0.9235714285714286 |