You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
This is the official implementation for JSIA-ReID. Please refer our paper for more details:
[Paper] [Neural Network 2020] Cross-Modality Paired-Images Generation and Augmentation for RGB-Infrared Person Re-Identification
[Paper, Poster] [AAAI2020] Cross-Modality Paired-Images Generation for RGB-Infrared Person Re-Identification
Bibtex
If you find the code useful, please consider citing our paper:
@InProceedings{wang2020cross,
title={Cross-modality paired-images generation and augmentation for RGB-infrared person re-identification},
author={Guan'an, Wang and Yang, Yang and Zhang, Tianzhu and Cheng, Jian and Hou, Zengguang and Tiwari, Prayag and Pandey, Hari Mohan and others},
journal={Neural Networks},
volume={128},
pages={294--304},
year={2020},
publisher={Elsevier}
}
@InProceedings{wang2020crossmodality,
author = "Guan-An {Wang} and Tianzhu {Zhang} and Yang {Yang} and Jian {Cheng} and Jianlong {Chang} and Xu {Liang} and Zengguang {Hou}",
title = {Cross-Modality Paired-Images Generation for RGB-Infrared Person Re-Identification},
booktitle = {AAAI-20 AAAI Conference on Artificial Intelligence},
year = {2020}
}
# train, please replace sysu-mm01-path with your own path
python main.py --dataset_path sysu-mm01-path --mode train
Test with Pre-trained Model
pretrained model (Google Drive, Baidu Disk(pwd:656y)), please download all the 4 files into a folder.
test with the pre-trained model
# test with pretrained model, please replace sysu-mm01-path and pretrained-model-path with your own paths
python main.py --dataset_path sysu-mm01-path --mode test --pretrained_model_path pretrained-model-path --pretrained_model_epoch 649
Experimental Results
Settings: We trained our model with 2 GTX1080ti GPUs.
Comparison with SOTA
License
This repo is released under the MIT License.
Contacts
If you have any question about the project, please feel free to contact with me.