Code for paper [ACE: Ally Complementary Experts for Solving Long-Tailed Recognition in One-Shot] (ICCV 2021, oral))

Last update: May 09, 2022

Related tags

Overview

ACE: Ally Complementary Experts for Solving Long-Tailed Recognition in One-Shot

This repository is the official PyTorch implementation of ICCV-21 paper ACE: Ally Complementary Experts for Solving Long-Tailed Recognition in One-Shot.

Prerequirements

To install the environment.

conda env create -f environment.yml
conda activate ACE
git clone https://github.com/NVIDIA/apex
cd apex
pip install -v --disable-pip-version-check --no-cache-dir --global-option="--cpp_ext" --global-option="--cuda_ext" ./

Data format

The annotation of a dataset is a dict consisting of two field: annotations and num_classes. The field annotations is a list of dict with image_id, fpath, im_height, im_width and category_id.

Here is an example.

{
    'annotations': [
                    {
                        'image_id': 1,
                        'fpath': '/data/iNat18/images/train_val2018/Plantae/7477/3b60c9486db1d2ee875f11a669fbde4a.jpg',
                        'im_height': 600,
                        'im_width': 800,
                        'category_id': 7477
                    },
                    ...
                   ]
    'num_classes': 8142
}

Usage

Training

#bash data_parallel_train.sh configuration_file_path GPU_indexes
bash data_parallel_train.sh configs/cifar100_im100.yaml 0,1

Testing

#python valid.py configuration_file_path
python valid.py configs/cifar100_im100.yaml

Trained models

Acknowledgement

This project is developed based on Bag of tricks @AAAI-21, thanks for their works!

Citation

@inproceedings{cai2021ace,
  title={ACE: Ally Complementary Experts for Solving Long-Tailed Recognition in One-Shot},
  author={Cai, Jiarui and Wang, Yizhou and Hwang, Jenq-Neng},
  booktitle={Proceedings of the IEEE/CVF International Conference on Computer Vision},
  pages={112--121},
  year={2021}
}

Code for paper [ACE: Ally Complementary Experts for Solving Long-Tailed Recognition in One-Shot] (ICCV 2021, oral))

Related tags

Overview

ACE: Ally Complementary Experts for Solving Long-Tailed Recognition in One-Shot

Prerequirements

Data format

Usage

Training

Testing

Acknowledgement

Citation

Owner

Jiarui

The codes reproduce the figures and statistics in the paper, "Controlling for multiple covariates," by Mark Tygert.

Marine debris detection with commercial satellite imagery and deep learning.

This repository contains an implementation of ConvMixer for the ICLR 2022 submission "Patches Are All You Need?".

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

Simulated garment dataset for virtual try-on

Variational Attention: Propagating Domain-Specific Knowledge for Multi-Domain Learning in Crowd Counting (ICCV, 2021)

Simple data balancing baselines for worst-group-accuracy benchmarks.

Codebase for the Summary Loop paper at ACL2020

Code for the paper "TadGAN: Time Series Anomaly Detection Using Generative Adversarial Networks"

Code for CVPR2019 paper《Unequal Training for Deep Face Recognition with Long Tailed Noisy Data》

ATAC: Adversarially Trained Actor Critic

LowRankModels.jl is a julia package for modeling and fitting generalized low rank models.

Object tracking implemented with YOLOv4, DeepSort, and TensorFlow.

Exporter for Storage Area Network (SAN)

How Do Adam and Training Strategies Help BNNs Optimization? In ICML 2021.

A simple rest api serving a deep learning model that classifies human gender based on their faces. (vgg16 transfare learning)

PyGCL: Graph Contrastive Learning Library for PyTorch

ECAENet (TensorFlow and Keras)

Official PyTorch code of DeepPanoContext: Panoramic 3D Scene Understanding with Holistic Scene Context Graph and Relation-based Optimization (ICCV 2021 Oral).

Contrastive Learning Inverts the Data Generating Process