PyTorch implementation of Learning by Aligning (ICCV 2021)
This is an official PyTorch implementation of the paper "Learning by Aligning: Visible-Infrared Person Re-identification using Cross-Modal Correspondences", ICCV 2021.
For more details, visit our project site or see our paper.
Requirements
- Python 3.8
- PyTorch 1.7.1
- GPU memory >= 11GB
Getting started
First, clone our git repository.
git clone https://github.com/cvlab-yonsei/LbA.git
cd LbA
Docker
You can use docker pull sanghslee/ps:1.7.1-cuda11.0-cudnn8-runtime
Prepare datasets
- SYSU-MM01: download from this link.
- For SYSU-MM01, you need to preprocess the .jpg files into .npy files by running:
python utils/pre_preprocess_sysu.py --data_dir /path/to/SYSU-MM01
- Modify the dataset directory below accordingly.
- L63 of
train.py - L54 of
test.py
- L63 of
- For SYSU-MM01, you need to preprocess the .jpg files into .npy files by running:
Train
-
run
python train.py --method full -
Important:
- Performances reported during training does not reflect exact performances of your model. This is due to 1) evaluation protocols of the datasets and 2) random seed configurations.
- Make sure you seperately run
test.pyto obtain correct results to be reported in your paper.
- Performances reported during training does not reflect exact performances of your model. This is due to 1) evaluation protocols of the datasets and 2) random seed configurations.
Test
- run
python test.py --method full - The results should be around:
| dataset | method | mAP | rank-1 |
|---|---|---|---|
| SYSU-MM01 | baseline | 49.54 | 50.43 |
| SYSU-MM01 | full | 54.14 | 55.41 |
Pretrained weights
- Download [SYSU-MM01]
- The results should be:
| dataset | method | mAP | rank-1 |
|---|---|---|---|
| SYSU-MM01 | full | 55.22 | 56.31 |
Bibtex
@article{park2021learning,
title={Learning by Aligning: Visible-Infrared Person Re-identification using Cross-Modal Correspondences},
author={Park, Hyunjong and Lee, Sanghoon and Lee, Junghyup and Ham, Bumsub},
journal={arXiv preprint arXiv:2108.07422},
year={2021}
}
1 Nov 11, 2021
2 Oct 13, 2021
42 Dec 22, 2022
67 Nov 30, 2022
20 Oct 25, 2022
4 Oct 29, 2021
6 Nov 17, 2021
1.3k Jan 05, 2023
2.1k Dec 23, 2022
1 Dec 22, 2021
157 Aug 23, 2022
1 Feb 13, 2022
4 Sep 04, 2021
5 Feb 20, 2022
356 Dec 08, 2022
17 Dec 30, 2022
165 Dec 31, 2022
23 Dec 30, 2022
85 Dec 31, 2022
2 Nov 06, 2021