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

State of the Art Neural Networks for Generative Deep Learning

NeurIPS 2021 Datasets and Benchmarks Track

Semantic Bottleneck Scene Generation

A collection of differentiable SVD methods and also the official implementation of the ICCV21 paper "Why Approximate Matrix Square Root Outperforms Accurate SVD in Global Covariance Pooling?"

Official code of ICCV2021 paper "Residual Attention: A Simple but Effective Method for Multi-Label Recognition"

Numerical differential equation solvers in JAX. Autodifferentiable and GPU-capable.

deep learning for image processing including classification and object-detection etc.

An API-first distributed deployment system of deep learning models using timeseries data to analyze and predict systems behaviour

Example repository for custom C++/CUDA operators for TorchScript

Single Image Deraining Using Bilateral Recurrent Network (TIP 2020)

Indonesian Car License Plate Character Recognition using Tensorflow, Keras and OpenCV.

Check out the StyleGAN repo and place it in the same directory hierarchy as the present repo

Semi-supervised Representation Learning for Remote Sensing Image Classification Based on Generative Adversarial Networks

A framework to train language models to learn invariant representations.

This repository contains the source code for the paper "DONeRF: Towards Real-Time Rendering of Compact Neural Radiance Fields using Depth Oracle Networks",

Depth image based mouse cursor visual haptic

An SMPC companion library for Syft

PyTorch implementation of "Efficient Neural Architecture Search via Parameters Sharing"

Details about the wide minima density hypothesis and metrics to compute width of a minima

Paddle-Skeleton-Based-Action-Recognition - DecoupleGCN-DropGraph, ASGCN, AGCN, STGCN