├── README.md ├── classes.py ├── config └── ImageNet_LT │ ├── clip_A_rn50.yaml │ ├── clip_B_rn50.yaml │ └── test.yaml ├── data ├── ClassAwareSampler.py ├── ClassPrioritySampler.py ├── ImageNet │ ├── ImageNet_val.txt │ └── gen_txt.py ├── ImageNet_LT │ ├── ImageNet_LT_test.txt │ ├── ImageNet_LT_train.txt │ └── ImageNet_LT_val.txt ├── MixedPrioritizedSampler.py ├── Places_LT │ ├── Places_LT_test.txt │ ├── Places_LT_train.txt │ └── Places_LT_val.txt └── dataloader.py ├── figure.PNG ├── logger.py ├── loss └── SoftmaxLoss.py ├── main.py ├── train.py └── utils.py /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaopengcuhk/BALLAD/HEAD/README.md -------------------------------------------------------------------------------- /classes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaopengcuhk/BALLAD/HEAD/classes.py -------------------------------------------------------------------------------- /config/ImageNet_LT/clip_A_rn50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaopengcuhk/BALLAD/HEAD/config/ImageNet_LT/clip_A_rn50.yaml -------------------------------------------------------------------------------- /config/ImageNet_LT/clip_B_rn50.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaopengcuhk/BALLAD/HEAD/config/ImageNet_LT/clip_B_rn50.yaml -------------------------------------------------------------------------------- /config/ImageNet_LT/test.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaopengcuhk/BALLAD/HEAD/config/ImageNet_LT/test.yaml -------------------------------------------------------------------------------- /data/ClassAwareSampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaopengcuhk/BALLAD/HEAD/data/ClassAwareSampler.py -------------------------------------------------------------------------------- /data/ClassPrioritySampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaopengcuhk/BALLAD/HEAD/data/ClassPrioritySampler.py -------------------------------------------------------------------------------- /data/ImageNet/ImageNet_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaopengcuhk/BALLAD/HEAD/data/ImageNet/ImageNet_val.txt -------------------------------------------------------------------------------- /data/ImageNet/gen_txt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaopengcuhk/BALLAD/HEAD/data/ImageNet/gen_txt.py -------------------------------------------------------------------------------- /data/ImageNet_LT/ImageNet_LT_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaopengcuhk/BALLAD/HEAD/data/ImageNet_LT/ImageNet_LT_test.txt -------------------------------------------------------------------------------- /data/ImageNet_LT/ImageNet_LT_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaopengcuhk/BALLAD/HEAD/data/ImageNet_LT/ImageNet_LT_train.txt -------------------------------------------------------------------------------- /data/ImageNet_LT/ImageNet_LT_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaopengcuhk/BALLAD/HEAD/data/ImageNet_LT/ImageNet_LT_val.txt -------------------------------------------------------------------------------- /data/MixedPrioritizedSampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaopengcuhk/BALLAD/HEAD/data/MixedPrioritizedSampler.py -------------------------------------------------------------------------------- /data/Places_LT/Places_LT_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaopengcuhk/BALLAD/HEAD/data/Places_LT/Places_LT_test.txt -------------------------------------------------------------------------------- /data/Places_LT/Places_LT_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaopengcuhk/BALLAD/HEAD/data/Places_LT/Places_LT_train.txt -------------------------------------------------------------------------------- /data/Places_LT/Places_LT_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaopengcuhk/BALLAD/HEAD/data/Places_LT/Places_LT_val.txt -------------------------------------------------------------------------------- /data/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaopengcuhk/BALLAD/HEAD/data/dataloader.py -------------------------------------------------------------------------------- /figure.PNG: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaopengcuhk/BALLAD/HEAD/figure.PNG -------------------------------------------------------------------------------- /logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaopengcuhk/BALLAD/HEAD/logger.py -------------------------------------------------------------------------------- /loss/SoftmaxLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaopengcuhk/BALLAD/HEAD/loss/SoftmaxLoss.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaopengcuhk/BALLAD/HEAD/main.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaopengcuhk/BALLAD/HEAD/train.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/gaopengcuhk/BALLAD/HEAD/utils.py --------------------------------------------------------------------------------