[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

[UNMAINTAINED] Automated machine learning for analytics & production

This is an official implementation for the WTW Dataset in "Parsing Table Structures in the Wild " on table detection and table structure recognition.

PyTorch implementation of PP-LCNet

Run Effective Large Batch Contrastive Learning on Limited Memory GPU

Exporter for Storage Area Network (SAN)

PyTorch Implementations for DeeplabV3 and PSPNet

Materials for my scikit-learn tutorial

Code for ACL'2021 paper WARP 🌀 Word-level Adversarial ReProgramming

FedGS: A Federated Group Synchronization Framework Implemented by LEAF-MX.

Python implementation of a live deep learning based age/gender/expression recognizer

This is a Image aid classification software based on python TK library development

Official re-implementation of the Calibrated Adversarial Refinement model described in the paper Calibrated Adversarial Refinement for Stochastic Semantic Segmentation

A PyTorch Implementation of "Watch Your Step: Learning Node Embeddings via Graph Attention" (NeurIPS 2018).

Survival analysis in Python

Official Pytorch Implementation of 'Learning Action Completeness from Points for Weakly-supervised Temporal Action Localization' (ICCV-21 Oral)

Code and data to accompany the camera-ready version of "Cross-Attention is All You Need: Adapting Pretrained Transformers for Machine Translation" in EMNLP 2021

blind SQLIpy sebuah alat injeksi sql yang menggunakan waktu sql untuk mendapatkan sebuah server database.

PyTorch implementation for "Mining Latent Structures with Contrastive Modality Fusion for Multimedia Recommendation"

Official implementation of "Robust channel-wise illumination estimation"

PyTorch implementation of Value Iteration Networks (VIN): Clean, Simple and Modular. Visualization in Visdom.