Code for the paper "M2m: Imbalanced Classification via Major-to-minor Translation" (CVPR 2020)

Last update: Oct 13, 2022

Related tags

Overview

M2m: Imbalanced Classification via Major-to-minor Translation

This repository contains code for the paper "M2m: Imbalanced Classification via Major-to-minor Translation" by Jaehyung Kim*, Jongheon Jeong* and Jinwoo Shin.

Dependencies

python3
pytorch >= 1.1.0
torchvision
tqdm

Scripts

Please check out run.sh for all the scripts to reproduce the CIFAR-10-LT results reported.

Training procedure of M2m

Train a baseline network g for generating minority samples

python train.py --no_over --ratio 100 --decay 2e-4 --model resnet32 --dataset cifar10 \
--lr 0.1 --batch-size 128 --name 'ERM' --warm 200 --epoch 200

Train another network f using M2m with the pre-trained g

python train.py -gen -r --ratio 100 --decay 2e-4 --model resnet32 --dataset cifar10 \
--lr 0.1 --batch-size 128 --name 'M2m' --beta 0.999 --lam 0.5 --gamma 0.9 \
--step_size 0.1 --attack_iter 10 --warm 160 --epoch 200 --net_g ./checkpoint/pre_trained_g.t7

We also provide a pre-trained ResNet-32 model of g at checkpoint/erm_r100_c10_trial1.t7, so one can directly use M2m without pre-training as follows:

python train.py -gen -r --ratio 100 --decay 2e-4 --model resnet32 --dataset cifar10 \
--lr 0.1 --batch-size 128 --name 'M2m' --beta 0.999 --lam 0.5 --gamma 0.9 \
--step_size 0.1 --attack_iter 10 --warm 160 --epoch 200 --net_g ./checkpoint/erm_r100_c10_trial1.t7

Code for the paper "M2m: Imbalanced Classification via Major-to-minor Translation" (CVPR 2020)

Related tags

Overview

M2m: Imbalanced Classification via Major-to-minor Translation

Dependencies

Scripts

Training procedure of M2m

Owner

Exploring Versatile Prior for Human Motion via Motion Frequency Guidance (3DV2021)

Unofficial implementation of MUSIQ (Multi-Scale Image Quality Transformer)

PyTorch experiments with the Zalando fashion-mnist dataset

Source code for "Roto-translated Local Coordinate Framesfor Interacting Dynamical Systems"

Build and run Docker containers leveraging NVIDIA GPUs

Deepfake Scanner by Deepware.

SemEval2022 Patronizing and Condescending Language (PCL) Detection

Federated Learning Based on Dynamic Regularization

Repository for tackling Kaggle Ultrasound Nerve Segmentation challenge using Torchnet.

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

DeepAL: Deep Active Learning in Python

Code and data accompanying our SVRHM'21 paper.

This repository contains the source code of Auto-Lambda and baselines from the paper, Auto-Lambda: Disentangling Dynamic Task Relationships.

[arXiv'22] Panoptic NeRF: 3D-to-2D Label Transfer for Panoptic Urban Scene Segmentation

An All-MLP solution for Vision, from Google AI

Reducing Information Bottleneck for Weakly Supervised Semantic Segmentation (NeurIPS 2021)

patchmatch和patchmatchstereo算法的python实现

Official PyTorch implementation of "Physics-aware Difference Graph Networks for Sparsely-Observed Dynamics".

The description of FMFCC-A (audio track of FMFCC) dataset and Challenge resluts.

[ICCV '21] In this repository you find the code to our paper Keypoint Communities