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

PyTorch implementation for our NeurIPS 2021 Spotlight paper "Long Short-Term Transformer for Online Action Detection".

Time Series Cross-Validation -- an extension for scikit-learn

Official repository for the paper "Instance-Conditioned GAN"

This is the research repository for Vid2Doppler: Synthesizing Doppler Radar Data from Videos for Training Privacy-Preserving Activity Recognition.

Cancer-and-Tumor-Detection-Using-Inception-model - In this repo i am gonna show you how i did cancer/tumor detection in lungs using deep neural networks, specifically here the Inception model by google.

Simple torch.nn.module implementation of Alias-Free-GAN style filter and resample

Python SDK for building, training, and deploying ML models

BEGAN in PyTorch

Official implementation of "OpenPifPaf: Composite Fields for Semantic Keypoint Detection and Spatio-Temporal Association" in PyTorch.

This repo is duplication of jwyang/faster-rcnn.pytorch

Implementation of the SUMO (Slim U-Net trained on MODA) model

The implementation of the lifelong infinite mixture model

Official implementation of "Not only Look, but also Listen: Learning Multimodal Violence Detection under Weak Supervision" ECCV2020

Bayesian Deep Learning and Deep Reinforcement Learning for Object Shape Error Response and Correction of Manufacturing Systems

Automated detection of anomalous exoplanet transits in light curve data.

Source Code for our paper: Understand me, if you refer to Aspect Knowledge: Knowledge-aware Gated Recurrent Memory Network

Normalizing Flows with a resampled base distribution

Deep Occlusion-Aware Instance Segmentation with Overlapping BiLayers [CVPR 2021]

This is the pytorch code for the paper Curious Representation Learning for Embodied Intelligence.

A medical imaging framework for Pytorch