├── LICENSE ├── README.md ├── TUTORIAL.md ├── configs ├── env.yml ├── pretext │ ├── moco_imagenet100.yml │ ├── moco_imagenet200.yml │ ├── moco_imagenet50.yml │ ├── simclr_cifar10.yml │ ├── simclr_cifar20.yml │ └── simclr_stl10.yml ├── scan │ ├── imagenet_eval.yml │ ├── scan_cifar10.yml │ ├── scan_cifar20.yml │ ├── scan_imagenet_100.yml │ ├── scan_imagenet_200.yml │ ├── scan_imagenet_50.yml │ └── scan_stl10.yml └── selflabel │ ├── selflabel_cifar10.yml │ ├── selflabel_cifar20.yml │ ├── selflabel_imagenet_100.yml │ ├── selflabel_imagenet_200.yml │ ├── selflabel_imagenet_50.yml │ └── selflabel_stl10.yml ├── data ├── augment.py ├── cifar.py ├── custom_dataset.py ├── imagenet.py ├── imagenet_subsets │ ├── imagenet_100.txt │ ├── imagenet_200.txt │ └── imagenet_50.txt └── stl.py ├── eval.py ├── images ├── pipeline.png ├── prototypes_cifar10.jpg ├── teaser.jpg └── tutorial │ ├── confusion_matrix_stl10.png │ └── prototypes_stl10.jpg ├── logs └── scan_stl10.txt ├── losses └── losses.py ├── moco.py ├── models ├── models.py ├── resnet.py ├── resnet_cifar.py └── resnet_stl.py ├── requirements.txt ├── scan.py ├── selflabel.py ├── simclr.py ├── tutorial_nn.py └── utils ├── collate.py ├── common_config.py ├── config.py ├── ema.py ├── evaluate_utils.py ├── memory.py ├── mypath.py ├── train_utils.py └── utils.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/README.md -------------------------------------------------------------------------------- /TUTORIAL.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/TUTORIAL.md -------------------------------------------------------------------------------- /configs/env.yml: -------------------------------------------------------------------------------- 1 | root_dir: /path/where/to/store/results/ 2 | -------------------------------------------------------------------------------- /configs/pretext/moco_imagenet100.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/configs/pretext/moco_imagenet100.yml -------------------------------------------------------------------------------- /configs/pretext/moco_imagenet200.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/configs/pretext/moco_imagenet200.yml -------------------------------------------------------------------------------- /configs/pretext/moco_imagenet50.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/configs/pretext/moco_imagenet50.yml -------------------------------------------------------------------------------- /configs/pretext/simclr_cifar10.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/configs/pretext/simclr_cifar10.yml -------------------------------------------------------------------------------- /configs/pretext/simclr_cifar20.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/configs/pretext/simclr_cifar20.yml -------------------------------------------------------------------------------- /configs/pretext/simclr_stl10.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/configs/pretext/simclr_stl10.yml -------------------------------------------------------------------------------- /configs/scan/imagenet_eval.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/configs/scan/imagenet_eval.yml -------------------------------------------------------------------------------- /configs/scan/scan_cifar10.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/configs/scan/scan_cifar10.yml -------------------------------------------------------------------------------- /configs/scan/scan_cifar20.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/configs/scan/scan_cifar20.yml -------------------------------------------------------------------------------- /configs/scan/scan_imagenet_100.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/configs/scan/scan_imagenet_100.yml -------------------------------------------------------------------------------- /configs/scan/scan_imagenet_200.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/configs/scan/scan_imagenet_200.yml -------------------------------------------------------------------------------- /configs/scan/scan_imagenet_50.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/configs/scan/scan_imagenet_50.yml -------------------------------------------------------------------------------- /configs/scan/scan_stl10.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/configs/scan/scan_stl10.yml -------------------------------------------------------------------------------- /configs/selflabel/selflabel_cifar10.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/configs/selflabel/selflabel_cifar10.yml -------------------------------------------------------------------------------- /configs/selflabel/selflabel_cifar20.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/configs/selflabel/selflabel_cifar20.yml -------------------------------------------------------------------------------- /configs/selflabel/selflabel_imagenet_100.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/configs/selflabel/selflabel_imagenet_100.yml -------------------------------------------------------------------------------- /configs/selflabel/selflabel_imagenet_200.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/configs/selflabel/selflabel_imagenet_200.yml -------------------------------------------------------------------------------- /configs/selflabel/selflabel_imagenet_50.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/configs/selflabel/selflabel_imagenet_50.yml -------------------------------------------------------------------------------- /configs/selflabel/selflabel_stl10.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/configs/selflabel/selflabel_stl10.yml -------------------------------------------------------------------------------- /data/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/data/augment.py -------------------------------------------------------------------------------- /data/cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/data/cifar.py -------------------------------------------------------------------------------- /data/custom_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/data/custom_dataset.py -------------------------------------------------------------------------------- /data/imagenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/data/imagenet.py -------------------------------------------------------------------------------- /data/imagenet_subsets/imagenet_100.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/data/imagenet_subsets/imagenet_100.txt -------------------------------------------------------------------------------- /data/imagenet_subsets/imagenet_200.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/data/imagenet_subsets/imagenet_200.txt -------------------------------------------------------------------------------- /data/imagenet_subsets/imagenet_50.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/data/imagenet_subsets/imagenet_50.txt -------------------------------------------------------------------------------- /data/stl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/data/stl.py -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/eval.py -------------------------------------------------------------------------------- /images/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/images/pipeline.png -------------------------------------------------------------------------------- /images/prototypes_cifar10.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/images/prototypes_cifar10.jpg -------------------------------------------------------------------------------- /images/teaser.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/images/teaser.jpg -------------------------------------------------------------------------------- /images/tutorial/confusion_matrix_stl10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/images/tutorial/confusion_matrix_stl10.png -------------------------------------------------------------------------------- /images/tutorial/prototypes_stl10.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/images/tutorial/prototypes_stl10.jpg -------------------------------------------------------------------------------- /logs/scan_stl10.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/logs/scan_stl10.txt -------------------------------------------------------------------------------- /losses/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/losses/losses.py -------------------------------------------------------------------------------- /moco.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/moco.py -------------------------------------------------------------------------------- /models/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/models/models.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/models/resnet.py -------------------------------------------------------------------------------- /models/resnet_cifar.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/models/resnet_cifar.py -------------------------------------------------------------------------------- /models/resnet_stl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/models/resnet_stl.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/requirements.txt -------------------------------------------------------------------------------- /scan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/scan.py -------------------------------------------------------------------------------- /selflabel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/selflabel.py -------------------------------------------------------------------------------- /simclr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/simclr.py -------------------------------------------------------------------------------- /tutorial_nn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/tutorial_nn.py -------------------------------------------------------------------------------- /utils/collate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/utils/collate.py -------------------------------------------------------------------------------- /utils/common_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/utils/common_config.py -------------------------------------------------------------------------------- /utils/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/utils/config.py -------------------------------------------------------------------------------- /utils/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/utils/ema.py -------------------------------------------------------------------------------- /utils/evaluate_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/utils/evaluate_utils.py -------------------------------------------------------------------------------- /utils/memory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/utils/memory.py -------------------------------------------------------------------------------- /utils/mypath.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/utils/mypath.py -------------------------------------------------------------------------------- /utils/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/utils/train_utils.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/wvangansbeke/Unsupervised-Classification/HEAD/utils/utils.py --------------------------------------------------------------------------------