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

Differentiable Optimizers with Perturbations in Pytorch

Code for Multiple Instance Active Learning for Object Detection, CVPR 2021

A BaSiC Tool for Background and Shading Correction of Optical Microscopy Images

S2-BNN: Bridging the Gap Between Self-Supervised Real and 1-bit Neural Networks via Guided Distribution Calibration (CVPR 2021)

Fastquant - Backtest and optimize your trading strategies with only 3 lines of code!

Code for "The Box Size Confidence Bias Harms Your Object Detector"

Source code and dataset for ACL2021 paper: "ERICA: Improving Entity and Relation Understanding for Pre-trained Language Models via Contrastive Learning".

This repository focus on Image Captioning & Video Captioning & Seq-to-Seq Learning & NLP

An Inverse Kinematics library aiming performance and modularity

Tensorflow/Keras Plug-N-Play Deep Learning Models Compilation

Official implementation of "Implicit Neural Representations with Periodic Activation Functions"

Diverse Image Generation via Self-Conditioned GANs

BC3407-Group-5-Project - BC3407 Group Project With Python

Visual Tracking by TridenAlign and Context Embedding

PyTorch Implementations for DeeplabV3 and PSPNet

Introduction to AI assignment 1 HCM University of Technology, term 211

Benchmark library for high-dimensional HPO of black-box models based on Weighted Lasso regression

DockStream: A Docking Wrapper to Enhance De Novo Molecular Design

R-package accompanying the paper "Dynamic Factor Model for Functional Time Series: Identification, Estimation, and Prediction"

STBP is a way to train SNN with datasets by Backward propagation.