Due to need of ablation study experiments, you may encounter problems with out-of-bounds labels or other unexpected problems. We have modified our code. You can clone our latest code to reproduce the results in the paper!
We will standardize our code and add a detailed description before the paper is published !
- torch==1.3.1
- torchvision==0.4.2
Market-1501, DukeMTMC-reID, and MSMT17 should be organized as follows:
. +-- data | +-- market | +-- bounding_box_train | +-- query | +-- bounding_box_test | +-- duke | +-- bounding_box_train | +-- query | +-- bounding_box_test | +-- msmt17 | +-- train | +-- test | +-- list_train.txt | +-- list_val.txt | +-- list_query.txt | +-- list_gallery.txt + -- other files in this repo
CUDA_DEVICE_ORDER=PCI_BUS_ID CUDA_VISIBLE_DEVICES='gpu_num' python train.py train --trainset_name market --save_dir='save_dir'
CUDA_DEVICE_ORDER=PCI_BUS_ID CUDA_VISIBLE_DEVICES='gpu_num' python test.py test --testset_name market --save_dir='save_dir'