├── .gitignore ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE ├── README.md ├── assets └── tau_norm.png ├── config ├── CIFAR100_LT │ ├── balanced_softmax_imba200.yaml │ ├── balms_imba200.yaml │ ├── decouple_balanced_softmax_imba200.yaml │ └── softmax_imba200.yaml ├── CIFAR10_LT │ ├── balanced_softmax_imba200.yaml │ ├── balms_imba200.yaml │ ├── decouple_balanced_softmax_imba200.yaml │ └── softmax_imba200.yaml ├── ImageNet_LT │ ├── balanced_softmax_resnet10.yaml │ ├── balms_resnet10.yaml │ ├── cls_crt.yaml │ ├── cls_lws.yaml │ ├── decouple_balanced_softmax_resnet10.yaml │ ├── feat_balance.yaml │ ├── feat_shift.yaml │ ├── feat_squareroot.yaml │ ├── feat_uniform.yaml │ ├── feat_uniform_resnet10.yaml │ └── softmax_resnet10.yaml ├── Places_LT │ ├── balms.yaml │ ├── cls_crt.yaml │ ├── cls_lws.yaml │ └── decouple_balanced_softmax.yaml └── iNaturalist18 │ ├── cls_crt.yaml │ ├── cls_lws.yaml │ └── feat_unifrom.yaml ├── data ├── ClassAwareSampler.py ├── ClassPrioritySampler.py ├── ImageNet │ ├── ImageNet_train.txt │ ├── ImageNet_val.txt │ └── gen_txt.py ├── ImageNet_LT │ ├── ImageNet_LT_test.txt │ ├── ImageNet_LT_train.txt │ └── ImageNet_LT_val.txt ├── ImbalanceCIFAR.py ├── MetaSampler.py ├── MixedPrioritizedSampler.py ├── Places_LT │ ├── Places_LT_test.txt │ ├── Places_LT_train.txt │ └── Places_LT_val.txt ├── autoaugment.py ├── dataloader.py └── iNaturalist18 │ ├── gen_lists.py │ ├── iNaturalist18_train.txt │ └── iNaturalist18_val.txt ├── layers └── ModulatedAttLayer.py ├── logger.py ├── loss ├── BalancedSoftmaxLoss.py ├── DiscCentroidsLoss.py └── SoftmaxLoss.py ├── main.py ├── models ├── CosNormClassifier.py ├── DotProductClassifier.py ├── KNNClassifier.py ├── MetaEmbeddingClassifier.py ├── ResNet101Feature.py ├── ResNet10Feature.py ├── ResNet152Feature.py ├── ResNet152FeatureCaffe.py ├── ResNet32Feature.py ├── ResNet50Feature.py ├── ResNetFeature.py ├── ResNext101Feature.py ├── ResNext152Feature.py ├── ResNext50Feature.py ├── ResNextFeature.py └── TauNormClassifier.py ├── run_networks.py ├── tau_norm.py └── utils.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/.gitignore -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/README.md -------------------------------------------------------------------------------- /assets/tau_norm.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/assets/tau_norm.png -------------------------------------------------------------------------------- /config/CIFAR100_LT/balanced_softmax_imba200.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/CIFAR100_LT/balanced_softmax_imba200.yaml -------------------------------------------------------------------------------- /config/CIFAR100_LT/balms_imba200.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/CIFAR100_LT/balms_imba200.yaml -------------------------------------------------------------------------------- /config/CIFAR100_LT/decouple_balanced_softmax_imba200.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/CIFAR100_LT/decouple_balanced_softmax_imba200.yaml -------------------------------------------------------------------------------- /config/CIFAR100_LT/softmax_imba200.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/CIFAR100_LT/softmax_imba200.yaml -------------------------------------------------------------------------------- /config/CIFAR10_LT/balanced_softmax_imba200.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/CIFAR10_LT/balanced_softmax_imba200.yaml -------------------------------------------------------------------------------- /config/CIFAR10_LT/balms_imba200.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/CIFAR10_LT/balms_imba200.yaml -------------------------------------------------------------------------------- /config/CIFAR10_LT/decouple_balanced_softmax_imba200.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/CIFAR10_LT/decouple_balanced_softmax_imba200.yaml -------------------------------------------------------------------------------- /config/CIFAR10_LT/softmax_imba200.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/CIFAR10_LT/softmax_imba200.yaml -------------------------------------------------------------------------------- /config/ImageNet_LT/balanced_softmax_resnet10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/ImageNet_LT/balanced_softmax_resnet10.yaml -------------------------------------------------------------------------------- /config/ImageNet_LT/balms_resnet10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/ImageNet_LT/balms_resnet10.yaml -------------------------------------------------------------------------------- /config/ImageNet_LT/cls_crt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/ImageNet_LT/cls_crt.yaml -------------------------------------------------------------------------------- /config/ImageNet_LT/cls_lws.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/ImageNet_LT/cls_lws.yaml -------------------------------------------------------------------------------- /config/ImageNet_LT/decouple_balanced_softmax_resnet10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/ImageNet_LT/decouple_balanced_softmax_resnet10.yaml -------------------------------------------------------------------------------- /config/ImageNet_LT/feat_balance.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/ImageNet_LT/feat_balance.yaml -------------------------------------------------------------------------------- /config/ImageNet_LT/feat_shift.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/ImageNet_LT/feat_shift.yaml -------------------------------------------------------------------------------- /config/ImageNet_LT/feat_squareroot.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/ImageNet_LT/feat_squareroot.yaml -------------------------------------------------------------------------------- /config/ImageNet_LT/feat_uniform.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/ImageNet_LT/feat_uniform.yaml -------------------------------------------------------------------------------- /config/ImageNet_LT/feat_uniform_resnet10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/ImageNet_LT/feat_uniform_resnet10.yaml -------------------------------------------------------------------------------- /config/ImageNet_LT/softmax_resnet10.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/ImageNet_LT/softmax_resnet10.yaml -------------------------------------------------------------------------------- /config/Places_LT/balms.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/Places_LT/balms.yaml -------------------------------------------------------------------------------- /config/Places_LT/cls_crt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/Places_LT/cls_crt.yaml -------------------------------------------------------------------------------- /config/Places_LT/cls_lws.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/Places_LT/cls_lws.yaml -------------------------------------------------------------------------------- /config/Places_LT/decouple_balanced_softmax.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/Places_LT/decouple_balanced_softmax.yaml -------------------------------------------------------------------------------- /config/iNaturalist18/cls_crt.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/iNaturalist18/cls_crt.yaml -------------------------------------------------------------------------------- /config/iNaturalist18/cls_lws.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/iNaturalist18/cls_lws.yaml -------------------------------------------------------------------------------- /config/iNaturalist18/feat_unifrom.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/config/iNaturalist18/feat_unifrom.yaml -------------------------------------------------------------------------------- /data/ClassAwareSampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/data/ClassAwareSampler.py -------------------------------------------------------------------------------- /data/ClassPrioritySampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/data/ClassPrioritySampler.py -------------------------------------------------------------------------------- /data/ImageNet/ImageNet_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/data/ImageNet/ImageNet_train.txt -------------------------------------------------------------------------------- /data/ImageNet/ImageNet_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/data/ImageNet/ImageNet_val.txt -------------------------------------------------------------------------------- /data/ImageNet/gen_txt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/data/ImageNet/gen_txt.py -------------------------------------------------------------------------------- /data/ImageNet_LT/ImageNet_LT_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/data/ImageNet_LT/ImageNet_LT_test.txt -------------------------------------------------------------------------------- /data/ImageNet_LT/ImageNet_LT_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/data/ImageNet_LT/ImageNet_LT_train.txt -------------------------------------------------------------------------------- /data/ImageNet_LT/ImageNet_LT_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/data/ImageNet_LT/ImageNet_LT_val.txt -------------------------------------------------------------------------------- /data/ImbalanceCIFAR.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/data/ImbalanceCIFAR.py -------------------------------------------------------------------------------- /data/MetaSampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/data/MetaSampler.py -------------------------------------------------------------------------------- /data/MixedPrioritizedSampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/data/MixedPrioritizedSampler.py -------------------------------------------------------------------------------- /data/Places_LT/Places_LT_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/data/Places_LT/Places_LT_test.txt -------------------------------------------------------------------------------- /data/Places_LT/Places_LT_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/data/Places_LT/Places_LT_train.txt -------------------------------------------------------------------------------- /data/Places_LT/Places_LT_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/data/Places_LT/Places_LT_val.txt -------------------------------------------------------------------------------- /data/autoaugment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/data/autoaugment.py -------------------------------------------------------------------------------- /data/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/data/dataloader.py -------------------------------------------------------------------------------- /data/iNaturalist18/gen_lists.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/data/iNaturalist18/gen_lists.py -------------------------------------------------------------------------------- /data/iNaturalist18/iNaturalist18_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/data/iNaturalist18/iNaturalist18_train.txt -------------------------------------------------------------------------------- /data/iNaturalist18/iNaturalist18_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/data/iNaturalist18/iNaturalist18_val.txt -------------------------------------------------------------------------------- /layers/ModulatedAttLayer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/layers/ModulatedAttLayer.py -------------------------------------------------------------------------------- /logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/logger.py -------------------------------------------------------------------------------- /loss/BalancedSoftmaxLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/loss/BalancedSoftmaxLoss.py -------------------------------------------------------------------------------- /loss/DiscCentroidsLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/loss/DiscCentroidsLoss.py -------------------------------------------------------------------------------- /loss/SoftmaxLoss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/loss/SoftmaxLoss.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/main.py -------------------------------------------------------------------------------- /models/CosNormClassifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/models/CosNormClassifier.py -------------------------------------------------------------------------------- /models/DotProductClassifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/models/DotProductClassifier.py -------------------------------------------------------------------------------- /models/KNNClassifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/models/KNNClassifier.py -------------------------------------------------------------------------------- /models/MetaEmbeddingClassifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/models/MetaEmbeddingClassifier.py -------------------------------------------------------------------------------- /models/ResNet101Feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/models/ResNet101Feature.py -------------------------------------------------------------------------------- /models/ResNet10Feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/models/ResNet10Feature.py -------------------------------------------------------------------------------- /models/ResNet152Feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/models/ResNet152Feature.py -------------------------------------------------------------------------------- /models/ResNet152FeatureCaffe.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/models/ResNet152FeatureCaffe.py -------------------------------------------------------------------------------- /models/ResNet32Feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/models/ResNet32Feature.py -------------------------------------------------------------------------------- /models/ResNet50Feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/models/ResNet50Feature.py -------------------------------------------------------------------------------- /models/ResNetFeature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/models/ResNetFeature.py -------------------------------------------------------------------------------- /models/ResNext101Feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/models/ResNext101Feature.py -------------------------------------------------------------------------------- /models/ResNext152Feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/models/ResNext152Feature.py -------------------------------------------------------------------------------- /models/ResNext50Feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/models/ResNext50Feature.py -------------------------------------------------------------------------------- /models/ResNextFeature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/models/ResNextFeature.py -------------------------------------------------------------------------------- /models/TauNormClassifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/models/TauNormClassifier.py -------------------------------------------------------------------------------- /run_networks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/run_networks.py -------------------------------------------------------------------------------- /tau_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/tau_norm.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/BalancedMetaSoftmax-Classification/HEAD/utils.py --------------------------------------------------------------------------------