+-- NAIC_reid_2020
| +-- reid_baseline(source code)
| +-- model(dir to save the output)
| +-- NAIC_Person_Reid(put dataset here)
| +--image_B
-
cd
to folder where you want to download this repo -
Run
git clone https://github.com/bochuanwu/Naic_reid_2020.git
-
Install dependencies:
- pytorch>=1.1.0
- python>=3.5
- torchvision
- yacs
- cv2
I use cuda 10.1/python 3.6.10/torch 1.5.0/torchvision 0.6.0 for training and testing.
-
ResNet-ibn is applied as the backbone. Download ImageNet pretrained model here or using the link in the source code 'pth.txt'
-
If you want to test the Person ReID Compitition of NAIC . Put the test dataset to correct position and Use the following commands:
bash run.sh
-
If you want to use our baseline for training.
python train.py --config_file [CHOOSE WHICH config TO RUN] # E.g #python train.py --config_file configs/naic_round2_model_a.yml