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}
}
19 Dec 07, 2022
81 Jan 01, 2023
575 Dec 31, 2022
111 Nov 13, 2022
4 Jul 18, 2021
39 Sep 26, 2022
3 Oct 05, 2022
141 Dec 20, 2022
13 Mar 31, 2022
815 Dec 29, 2022
141 Dec 31, 2022
10 Dec 19, 2022
37 Dec 01, 2022
65 Dec 01, 2022
21 Mar 11, 2022
5 Aug 01, 2022
9 Nov 11, 2022
154 Dec 07, 2022
1.3k Jan 05, 2023
128 Sep 12, 2022