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

Self-Supervised Multi-Frame Monocular Scene Flow (CVPR 2021)

Rendering color and depth images for ShapeNet models.

library for nonlinear optimization, wrapping many algorithms for global and local, constrained or unconstrained, optimization

Pytorch implementation for Semantic Segmentation/Scene Parsing on MIT ADE20K dataset

Code for "3D Human Pose and Shape Regression with Pyramidal Mesh Alignment Feedback Loop"

[CVPR-2021] UnrealPerson: An adaptive pipeline for costless person re-identification

Multi-agent reinforcement learning algorithm and environment

Label-Free Model Evaluation with Semi-Structured Dataset Representations

SW components and demos for visual kinship recognition. An emphasis is put on the FIW dataset-- data loaders, benchmarks, results in summary.

利用Tensorflow实现基于CNN的中文短文本分类

New AidForBlind - Various Libraries used like OpenCV and other mentioned in Requirements.txt

An algorithm that handles large-scale aerial photo co-registration, based on SURF, RANSAC and PyTorch autograd.

History Aware Multimodal Transformer for Vision-and-Language Navigation

Cereal box identification in store shelves using computer vision and a single train image per model.

A Deep Learning based project for creating line art portraits.

Code for testing convergence rates of Lipschitz learning on graphs

Contrastive learning of Class-agnostic Activation Map for Weakly Supervised Object Localization and Semantic Segmentation (CVPR 2022)

Automate issue discovery for your projects against Lightning nightly and releases.

[ECCV 2020] Reimplementation of 3DDFAv2, including face mesh, head pose, landmarks, and more.

source code the paper Fast and Robust Iterative Closet Point.