Code for PSMA

This is the code for paper: Progressive Sample Mining and Representation Learning for One-Shot Person Re-identification with Adversarial Samples

https://arxiv.org/abs/1911.00666

envs

This code is mmodified from https://github.com/manutdzou/Person_ReID_Baseline with a same environment.

dataset

We tested our framework on two main datasets: Market1501 and Duke

To achieve the same performance as I did, I recommand to download the dataset I use and put it under datasets:

(BAIDU) link: https://pan.baidu.com/s/1Mr71cnCCB0SY7jHuk2JOdQ keys: 6z9d

The PATH for dataset is in ./config/default.py (for example: _C.DATASETS.STORE_DIR = ('/home/lihui/datasets/PSM_DATA'))

train

python train.py ./config/market_softmax_Htriplet_GAN.yaml

The performance log for checking

The log I print is under the /log_answer directory. This can be used as a reference for your training process.

Cite

@misc{li2019progressive,
    title={Progressive Sample Mining and Representation Learning for One-Shot Person Re-identification with Adversarial Samples},
    author={Hui Li and Jimin Xiao and Mingjie Sun and Eng Gee Lim and Yao Zhao},
    year={2019},
    eprint={1911.00666},
    archivePrefix={arXiv},
    primaryClass={cs.CV}
}