This project is built upon BBN.
pip install -r requirements.txt
- Training
python main/train.py --cfg configs/ImageNet.yaml DATASET.ROOT /path/ImageNet_ILSVRC2012
- 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