Long-tailed Distribution Adaptation (Accepted in ACM MM2021)

This project is built upon BBN.

Installation

pip install -r requirements.txt

Usage

1. Training

python main/train.py --cfg configs/ImageNet.yaml DATASET.ROOT /path/ImageNet_ILSVRC2012

2. Inference

python main/valid.py --cfg configs/ImageNet.yaml \
  TEST.MODEL_FILE ./output/LDA/ImageNet/LDA.ImageNet.resnext50.90epoch/models/best_model.pth \
  DATASET.VALID_JSON './datasets/ImageNet_LT/ImageNet_LT_test.txt'\
  DATASET.ROOT /path/ImageNet_ILSVRC2012