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

3D-Transformer: Molecular Representation with Transformer in 3D Space

Highway networks implemented in PyTorch.

Barbershop: GAN-based Image Compositing using Segmentation Masks (SIGGRAPH Asia 2021)

A PaddlePaddle implementation of STGCN with a few modifications in the model architecture in order to forecast traffic jam.

Spatial-Temporal Transformer for Dynamic Scene Graph Generation, ICCV2021

The 3rd place solution for competition

Code for the paper "Next Generation Reservoir Computing"

Code for CVPR2021 paper 'Where and What? Examining Interpretable Disentangled Representations'.

Subgraph Based Learning of Contextual Embedding

Entity-Based Knowledge Conflicts in Question Answering.

Lung Pattern Classification for Interstitial Lung Diseases Using a Deep Convolutional Neural Network

This is a Deep Leaning API for classifying emotions from human face and human audios.

Next-Best-View Estimation based on Deep Reinforcement Learning for Active Object Classification

Deepface is a lightweight face recognition and facial attribute analysis (age, gender, emotion and race) framework for python

Examples of using f2py to get high-speed Fortran integrated with Python easily

95.47% on CIFAR10 with PyTorch

Pytorch implementation of winner from VQA Chllange Workshop in CVPR'17

Open source Python module for computer vision

Elastic weight consolidation technique for incremental learning.

A complete, self-contained example for training ImageNet at state-of-the-art speed with FFCV