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

This is Official implementation for "Pose-guided Feature Disentangling for Occluded Person Re-Identification Based on Transformer" in AAAI2022

VOLO: Vision Outlooker for Visual Recognition

This repository contains the needed resources to build the HIRID-ICU-Benchmark dataset

This game was designed to encourage young people not to gamble on lotteries, as the probablity of correctly guessing the number is infinitesimal!

transfer attack; adversarial examples; black-box attack; unrestricted Adversarial Attacks on ImageNet; CVPR2021 天池黑盒竞赛

Control-Robot-Arm-using-PS4-Controller - A Robotic Arm based on Raspberry Pi and Arduino that controlled by PS4 Controller

3DIAS: 3D Shape Reconstruction with Implicit Algebraic Surfaces (ICCV 2021)

audioLIME: Listenable Explanations Using Source Separation

A python3 tool to take a 360 degree survey of the RF spectrum (hamlib + rotctld + RTL-SDR/HackRF)

3D HourGlass Networks for Human Pose Estimation Through Videos

Open Source Light Field Toolbox for Super-Resolution

Source code for models described in the paper "AudioCLIP: Extending CLIP to Image, Text and Audio" (https://arxiv.org/abs/2106.13043)

Cache Requests in Deta Bases and Echo them with Deta Micros

Iowa Project - My second project done at General Assembly, focused on feature engineering and understanding Linear Regression as a concept

SPT_LSA_ViT - Implementation for Visual Transformer for Small-size Datasets

Source code for our paper "Do Not Trust Prediction Scores for Membership Inference Attacks"

Joint-task Self-supervised Learning for Temporal Correspondence (NeurIPS 2019)

Retina blood vessel segmentation with a convolutional neural network

Official Code for "Constrained Mean Shift Using Distant Yet Related Neighbors for Representation Learning"

NeoDTI: Neural integration of neighbor information from a heterogeneous network for discovering new drug-target interactions