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

This is a collection of our NAS and Vision Transformer work.

Abstractive opinion summarization system (SelSum) and the largest dataset of Amazon product summaries (AmaSum). EMNLP 2021 conference paper.

Code for NeurIPS 2021 paper: Invariant Causal Imitation Learning for Generalizable Policies

Res2Net for Instance segmentation and Object detection using MaskRCNN

A graph adversarial learning toolbox based on PyTorch and DGL.

IRON Kaggle project done while doing IRONHACK Bootcamp where we had to analyze and use a Machine Learning Project to predict future sales

Experiments with the Robust Binary Interval Search (RBIS) algorithm, a Query-Based prediction algorithm for the Online Search problem.

A code repository associated with the paper A Benchmark for Rough Sketch Cleanup by Chuan Yan, David Vanderhaeghe, and Yotam Gingold from SIGGRAPH Asia 2020.

Data Augmentation with Variational Autoencoders

This is the dataset and code release of the OpenRooms Dataset.

This is an early in-development version of training CLIP models with hivemind.

Symbolic Parallel Adaptive Importance Sampling for Probabilistic Program Analysis in JAX

TCPNet - Temporal-attentive-Covariance-Pooling-Networks-for-Video-Recognition

use tensorflow 2.0 to tell a dog and cat from a specified picture

Source code for CAST - Crisis Domain Adaptation Using Sequence-to-sequence Transformers (Accepted to ISCRAM 2021, CorePaper).

Pytorch implementation of our paper under review — Lottery Jackpots Exist in Pre-trained Models

EXplainable Artificial Intelligence (XAI)

Api's bulid in Flask perfom to manage Todo Task.

Rainbow DQN implementation that outperforms the paper's results on 40% of games using 20x less data 🌈

The devkit of the nuScenes dataset.