[ACM MM 2020 Oral] Black Re-ID: A Head-shoulder Descriptor for the Challenging Problem of Person Re-Identification paper
2020-08-12: Update Code.
2020-08-20: Update paper link.
2021-01-14: Update White group data and corresponding stn model.
If you find the code useful, please consider citing our paper:
@InProceedings{xu2020ACM,
author = {Boqiang, Xu and Lingxiao, He and Xingyu, Liao and Wu,Liu and Zhenan, Sun and Tao, Mei},
title = {Black Re-ID: A Head-shoulder Descriptor for the Challenging Problem of Person Re-Identification},
booktitle = {Proceedings of the 28th ACM International Conference on Multimedia (MM '20)},
month = {October},
year = {2020}
}
please add the path of the Black re-ID dataset to DATASETS.DATASETS_ROOT in ./projects/Black_reid/configs/Base-HAA.yml
please add the path of the STN model to DATASETS.STN_ROOT in ./projects/Black_reid/configs/Base-HAA.yml
cd
to folder:
cd projects/Black_reid
- If you want to train with 1-GPU, run:
CUDA_VISIBLE_DEVICES=0 python train_net.py --config-file="configs/HAA_baseline_blackreid.yml"
if you want to train with 4-GPU, run:
CUDA_VISIBLE_DEVICES=0,1,2,3 python train_net.py --config-file="configs/HAA_baseline_blackreid.yml"
To evaluate a model's performance, use:
CUDA_VISIBLE_DEVICES=0 python train_net.py --config-file="configs/HAA_baseline_blackreid.yml" --eval-only MODEL.WEIGHTS /path/to/checkpoint_file
If you have any question about the project, please feel free to contact me.
E-mail: boqiang.xu@cripac.ia.ac.cn
The code was developed based on the ’fast-reid’ toolbox https://github.com/JDAI-CV/fast-reid.