Deep Relational Metric Learning
This repository is the official PyTorch implementation of Deep Relational Metric Learning.
Framework
Datasets
CUB-200-2011
Download from here.
Organize the dataset as follows:
- cub200
|- train
| |- class0
| | |- image0_1
| | |- ...
| |- ...
|- test
|- class100
| |- image100_1
| |- ...
|- ...
Cars196
Download from here.
Organize the dataset as follows:
- cars196
|- train
| |- class0
| | |- image0_1
| | |- ...
| |- ...
|- test
|- class98
| |- image98_1
| |- ...
|- ...
Requirements
To install requirements:
pip install -r requirements.txt
Training
Baseline models
To train the baseline model with the ProxyAnchor loss on CUB200, run this command:
CUDA_VISIBLE_DEVICES=0 python examples/train/main.py \
--save_name <experiment-name> \
--data_path <path-of-data> \
--phase train \
--device 0 \
--setting proxy_baseline \
--dataset cub200 \
--num_classes 100 \
--batch_size 120 \
--delete_old
To train the baseline model with the ProxyAnchor loss on Cars196, run this command:
CUDA_VISIBLE_DEVICES=0 python examples/train/main.py \
--save_name <experiment-name> \
--data_path <path-of-data> \
--phase train \
--device 0 \
--setting proxy_baseline \
--dataset cars196 \
--num_classes 98 \
--batch_size 120 \
--delete_old
DRML models
To train the proposed DRML model using the ProxyAnchor loss on CUB200 in the paper, run this command:
CUDA_VISIBLE_DEVICES=0 python examples/train/main.py \
--save_name <experiment-name> \
--data_path <path-of-data> \
--phase train \
--device 0 \
--setting proxy \
--dataset cub200 \
--num_classes 100 \
--batch_size 120 \
--delete_old
To train the proposed DRML model using the ProxyAnchor loss on Cars196 in the paper, run this command:
CUDA_VISIBLE_DEVICES=0 python examples/train/main.py \
--save_name <experiment-name> \
--data_path <path-of-data> \
--phase train \
--device 0 \
--setting proxy \
--dataset cars196 \
--num_classes 98 \
--batch_size 120 \
--delete_old
Device
We tested our code on a linux machine with an Nvidia RTX 3090 GPU card. We recommend using a GPU card with a memory > 8GB (BN-Inception + batch-size of 120 ).
Results
The baseline models achieve the following performances:
| Model name | Recall @ 1 | Recall @ 2 | Recall @ 4 | Recall @ 8 | NMI |
|---|---|---|---|---|---|
| cub200-ProxyAnchor-baseline | 67.3 | 77.7 | 85.7 | 91.4 | 68.7 |
| cars196-ProxyAnchor-baseline | 84.4 | 90.7 | 94.3 | 96.8 | 69.7 |
Our models achieve the following performances:
| Model name | Recall @ 1 | Recall @ 2 | Recall @ 4 | Recall @ 8 | NMI |
|---|---|---|---|---|---|
| cub200-ProxyAnchor-ours | 68.7 | 78.6 | 86.3 | 91.6 | 69.3 |
| cars196-ProxyAnchor-ours | 86.9 | 92.1 | 95.2 | 97.4 | 72.1 |
COMING SOON
- We will upload the code for cross-validation setting soon.
- We will update the optimal hyper-parameters of the experiments soon.
78 Dec 28, 2022
8 Apr 15, 2022
13 Dec 09, 2022
0 Sep 28, 2022
15 Nov 04, 2022
15 May 21, 2022
375 Dec 30, 2022
15 Mar 22, 2022
55 Dec 06, 2022
1k Dec 25, 2022
19 Dec 12, 2022
2 Sep 01, 2022
271 Nov 29, 2022
42 Nov 01, 2022
73 Sep 18, 2022
2.3k Jan 08, 2023
8 Oct 03, 2022
94 Dec 18, 2022
4 Oct 09, 2021
29 Jan 07, 2023