[ICCV 2021] Official PyTorch implementation for Deep Relational Metric Learning.

Last update: Dec 10, 2022

Related tags

Deep Learning DRML

Overview

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.

[ICCV 2021] Official PyTorch implementation for Deep Relational Metric Learning.

Related tags

Overview

Deep Relational Metric Learning

Framework

Datasets

CUB-200-2011

Cars196

Requirements

Training

Baseline models

DRML models

Device

Results

COMING SOON

Owner

Borui Zhang

An image processing project uses Viola-jones technique to detect faces and then use SIFT algorithm for recognition.

A Python package for performing pore network modeling of porous media

Implementation of "Meta-rPPG: Remote Heart Rate Estimation Using a Transductive Meta-Learner"

🔪 Elimination based Lightweight Neural Net with Pretrained Weights

Stochastic Downsampling for Cost-Adjustable Inference and Improved Regularization in Convolutional Networks

LSTM-VAE Implementation and Relevant Evaluations

Readings for "A Unified View of Relational Deep Learning for Polypharmacy Side Effect, Combination Therapy, and Drug-Drug Interaction Prediction."

Research - dataset and code for 2016 paper Learning a Driving Simulator

The official MegEngine implementation of the ICCV 2021 paper: GyroFlow: Gyroscope-Guided Unsupervised Optical Flow Learning

Pose estimation with MoveNet Lightning

Job-Recommend-Competition - Vectorwise Interpretable Attentions for Multimodal Tabular Data

UA-GEC: Grammatical Error Correction and Fluency Corpus for the Ukrainian Language

PyTorch code accompanying our paper on Maximum Entropy Generators for Energy-Based Models

A simple python library for fast image generation of people who do not exist.

Encode and decode text application

Task-related Saliency Network For Few-shot learning

QHack—the quantum machine learning hackathon

This repository contains the code, data, and models of the paper titled "XL-Sum: Large-Scale Multilingual Abstractive Summarization for 44 Languages" published in Findings of the Association for Computational Linguistics: ACL 2021.

This repository contains part of the code used to make the images visible in the article "How does an AI Imagine the Universe?" published on Towards Data Science.

Multi-angle c(q)uestion answering