Skip to content

pengzhiliang/LDA

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

2 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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
  1. 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

About

Code for "Long-tailed Distribution Adaptation"

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages