(ICML 2020) This repo contains code for our paper "Revisiting Training Strategies and Generalization Performance in Deep Metric Learning" (https://arxiv.org/abs/2002.08473) to facilitate consistent research in the field of Deep Metric Learning.

PyTorch custom dataset APIs -- CUB-200-2011, Stanford Dogs, Stanford Cars, FGVC Aircraft, NABirds, Tiny ImageNet, iNaturalist2017

Official MXNet implementation of "Embedding Expansion: Augmentation in Embedding Space for Deep Metric Learning" (CVPR 2020)

Official Tensorflow implementation of "Symmetrical Synthesis for Deep Metric Learning" (AAAI 2020)

(CVPR 2020) This repo contains code for "PADS: Policy-Adapted Sampling for Visual Similarity Learning", which proposes learnable triplet mining with Reinforcement Learning.

(ICML 2021) Implementation for S2SD - Simultaneous Similarity-based Self-Distillation for Deep Metric Learning. Paper Link: https://arxiv.org/abs/2009.08348

(ECCV 2020) This repo contains code for "DiVA: Diverse Visual Feature Aggregation for Deep Metric Learning" (https://arxiv.org/abs/2004.13458), which extends vanilla DML with auxiliary and self-supervised features.

Official PyTorch Implementation for the "Distilling Datasets Into Less Than One Image" paper.

"Exploring Vision Transformers for Fine-grained Classification" at CVPRW FGVC8

Implementation and Benchmark Splits to study Out-of-Distribution Generalization in Deep Metric Learning.

Official repository of ICML 2023 paper: Dividing and Conquering a BlackBox to a Mixture of Interpretable Models: Route, Interpret, Repeat

Implementation of Concept-level Debugging of Part-Prototype Networks

[ICML 2022] This work investigates the compatibility between label smoothing (LS) and knowledge distillation (KD). We suggest to use an LS-trained teacher with a low-temperature transfer to render high performance students.

This is a PyTorch-based implementation of the Generative Adversarial Text-to-Image Synthesis paper, utilizing a GAN architecture inspired by DCGAN with text descriptions as inputs to generate images.

Repository of the course project of CMU 16-824 Visual Learning and Recognition

Whole dataset of CUB200-2011