├── .gitignore ├── LICENSE ├── README.md ├── __init__.py ├── cifar100_DTC.py ├── cifar100_classif.py ├── cifar100_est_k.py ├── cifar10_DTC.py ├── cifar10_classif.py ├── data ├── __init__.py ├── cifar.py ├── cifarloader.py ├── datasets ├── experiments ├── imagenetloader.py ├── omniglotloader.py ├── sampler.py ├── svhnloader.py └── utils.py ├── imagenet2cifar_DTC.py ├── imagenet_DTC.py ├── imagenet_classif.py ├── imagenet_est_k.py ├── images └── iccv2019_DTC.png ├── models ├── __init__.py ├── resnet.py ├── resnet_3x3.py └── vgg.py ├── modules ├── __init__.py ├── module.py └── prototypical_loss.py ├── omniglot_DTC.py ├── omniglot_DTC_unknown.py ├── omniglot_est_k.py ├── omniglot_proto.py ├── scripts ├── cifar100_DTC_Baseline_10runs.sh ├── cifar100_DTC_PI_10runs.sh ├── cifar100_DTC_TEP_10runs.sh ├── cifar100_DTC_TE_10runs.sh ├── cifar10_DTC_Baseline_10runs.sh ├── cifar10_DTC_PI_10runs.sh ├── cifar10_DTC_TEP_10runs.sh ├── cifar10_DTC_TE_10runs.sh ├── download_imagenet_splits.sh ├── download_init_pretrained.sh ├── download_pretrained_models.sh ├── imagenet2cifar_DTC_Baseline_10runs.sh ├── imagenet2cifar_DTC_PI_10runs.sh ├── imagenet2cifar_DTC_TEP_10runs.sh ├── imagenet2cifar_DTC_TE_10runs.sh ├── imagenet_DTC_Baseline_ABC.sh ├── imagenet_DTC_PI_ABC.sh ├── imagenet_DTC_TEP_ABC.sh ├── imagenet_DTC_TE_ABC.sh ├── omniglot_DTC_Baseline_10runs.sh ├── omniglot_DTC_PI_10runs.sh ├── omniglot_DTC_TEP_10runs.sh ├── omniglot_DTC_TE_10runs.sh ├── omniglot_DTC_unknown_10runs.sh ├── svhn_DTC_Baseline_10runs.sh ├── svhn_DTC_PI_10runs.sh ├── svhn_DTC_TEP_10runs.sh └── svhn_DTC_TE_10runs.sh ├── svhn_DTC.py ├── svhn_classif.py └── utils ├── __init__.py ├── faster_mix_k_means_pytorch.py ├── pairwise.py ├── ramps.py └── util.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/README.md -------------------------------------------------------------------------------- /__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /cifar100_DTC.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/cifar100_DTC.py -------------------------------------------------------------------------------- /cifar100_classif.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/cifar100_classif.py -------------------------------------------------------------------------------- /cifar100_est_k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/cifar100_est_k.py -------------------------------------------------------------------------------- /cifar10_DTC.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/cifar10_DTC.py -------------------------------------------------------------------------------- /cifar10_classif.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/cifar10_classif.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/data/cifar.py -------------------------------------------------------------------------------- /data/cifarloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/data/cifarloader.py -------------------------------------------------------------------------------- /data/datasets: -------------------------------------------------------------------------------- 1 | /scratch/shared/slow/khan/image_datasets/ -------------------------------------------------------------------------------- /data/experiments: -------------------------------------------------------------------------------- 1 | /scratch/shared/nfs1/khan/DTC -------------------------------------------------------------------------------- /data/imagenetloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/data/imagenetloader.py -------------------------------------------------------------------------------- /data/omniglotloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/data/omniglotloader.py -------------------------------------------------------------------------------- /data/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/data/sampler.py -------------------------------------------------------------------------------- /data/svhnloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/data/svhnloader.py -------------------------------------------------------------------------------- /data/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/data/utils.py -------------------------------------------------------------------------------- /imagenet2cifar_DTC.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/imagenet2cifar_DTC.py -------------------------------------------------------------------------------- /imagenet_DTC.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/imagenet_DTC.py -------------------------------------------------------------------------------- /imagenet_classif.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/imagenet_classif.py -------------------------------------------------------------------------------- /imagenet_est_k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/imagenet_est_k.py -------------------------------------------------------------------------------- /images/iccv2019_DTC.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/images/iccv2019_DTC.png -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/resnet_3x3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/models/resnet_3x3.py -------------------------------------------------------------------------------- /models/vgg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/models/vgg.py -------------------------------------------------------------------------------- /modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /modules/module.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/modules/module.py -------------------------------------------------------------------------------- /modules/prototypical_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/modules/prototypical_loss.py -------------------------------------------------------------------------------- /omniglot_DTC.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/omniglot_DTC.py -------------------------------------------------------------------------------- /omniglot_DTC_unknown.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/omniglot_DTC_unknown.py -------------------------------------------------------------------------------- /omniglot_est_k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/omniglot_est_k.py -------------------------------------------------------------------------------- /omniglot_proto.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/omniglot_proto.py -------------------------------------------------------------------------------- /scripts/cifar100_DTC_Baseline_10runs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/cifar100_DTC_Baseline_10runs.sh -------------------------------------------------------------------------------- /scripts/cifar100_DTC_PI_10runs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/cifar100_DTC_PI_10runs.sh -------------------------------------------------------------------------------- /scripts/cifar100_DTC_TEP_10runs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/cifar100_DTC_TEP_10runs.sh -------------------------------------------------------------------------------- /scripts/cifar100_DTC_TE_10runs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/cifar100_DTC_TE_10runs.sh -------------------------------------------------------------------------------- /scripts/cifar10_DTC_Baseline_10runs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/cifar10_DTC_Baseline_10runs.sh -------------------------------------------------------------------------------- /scripts/cifar10_DTC_PI_10runs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/cifar10_DTC_PI_10runs.sh -------------------------------------------------------------------------------- /scripts/cifar10_DTC_TEP_10runs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/cifar10_DTC_TEP_10runs.sh -------------------------------------------------------------------------------- /scripts/cifar10_DTC_TE_10runs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/cifar10_DTC_TE_10runs.sh -------------------------------------------------------------------------------- /scripts/download_imagenet_splits.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/download_imagenet_splits.sh -------------------------------------------------------------------------------- /scripts/download_init_pretrained.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/download_init_pretrained.sh -------------------------------------------------------------------------------- /scripts/download_pretrained_models.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/download_pretrained_models.sh -------------------------------------------------------------------------------- /scripts/imagenet2cifar_DTC_Baseline_10runs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/imagenet2cifar_DTC_Baseline_10runs.sh -------------------------------------------------------------------------------- /scripts/imagenet2cifar_DTC_PI_10runs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/imagenet2cifar_DTC_PI_10runs.sh -------------------------------------------------------------------------------- /scripts/imagenet2cifar_DTC_TEP_10runs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/imagenet2cifar_DTC_TEP_10runs.sh -------------------------------------------------------------------------------- /scripts/imagenet2cifar_DTC_TE_10runs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/imagenet2cifar_DTC_TE_10runs.sh -------------------------------------------------------------------------------- /scripts/imagenet_DTC_Baseline_ABC.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/imagenet_DTC_Baseline_ABC.sh -------------------------------------------------------------------------------- /scripts/imagenet_DTC_PI_ABC.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/imagenet_DTC_PI_ABC.sh -------------------------------------------------------------------------------- /scripts/imagenet_DTC_TEP_ABC.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/imagenet_DTC_TEP_ABC.sh -------------------------------------------------------------------------------- /scripts/imagenet_DTC_TE_ABC.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/imagenet_DTC_TE_ABC.sh -------------------------------------------------------------------------------- /scripts/omniglot_DTC_Baseline_10runs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/omniglot_DTC_Baseline_10runs.sh -------------------------------------------------------------------------------- /scripts/omniglot_DTC_PI_10runs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/omniglot_DTC_PI_10runs.sh -------------------------------------------------------------------------------- /scripts/omniglot_DTC_TEP_10runs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/omniglot_DTC_TEP_10runs.sh -------------------------------------------------------------------------------- /scripts/omniglot_DTC_TE_10runs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/omniglot_DTC_TE_10runs.sh -------------------------------------------------------------------------------- /scripts/omniglot_DTC_unknown_10runs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/omniglot_DTC_unknown_10runs.sh -------------------------------------------------------------------------------- /scripts/svhn_DTC_Baseline_10runs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/svhn_DTC_Baseline_10runs.sh -------------------------------------------------------------------------------- /scripts/svhn_DTC_PI_10runs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/svhn_DTC_PI_10runs.sh -------------------------------------------------------------------------------- /scripts/svhn_DTC_TEP_10runs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/svhn_DTC_TEP_10runs.sh -------------------------------------------------------------------------------- /scripts/svhn_DTC_TE_10runs.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/scripts/svhn_DTC_TE_10runs.sh -------------------------------------------------------------------------------- /svhn_DTC.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/svhn_DTC.py -------------------------------------------------------------------------------- /svhn_classif.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/svhn_classif.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/faster_mix_k_means_pytorch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/utils/faster_mix_k_means_pytorch.py -------------------------------------------------------------------------------- /utils/pairwise.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/utils/pairwise.py -------------------------------------------------------------------------------- /utils/ramps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/utils/ramps.py -------------------------------------------------------------------------------- /utils/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/k-han/DTC/HEAD/utils/util.py --------------------------------------------------------------------------------