├── .gitignore ├── CFL.png ├── README.md ├── amal.py ├── checkpoints └── README.md ├── data └── README.md ├── dataloader.py ├── datasets ├── __init__.py ├── cub200.py ├── stanford_dogs.py └── utils.py ├── download_data.py ├── kd.py ├── logs ├── acc-densenet121.png ├── acc-resnet34.png ├── acc-resnet50.png ├── amal_densenet121.txt ├── amal_resnet34.txt ├── amal_resnet50.txt ├── draw_acc_curve.py ├── kd_densenet121.txt ├── kd_resnet34.txt └── kd_resnet50.txt ├── loss ├── __init__.py ├── loss.py └── mmd.py ├── models ├── __init__.py ├── cfl.py ├── densenet.py └── resnet.py ├── requirements.txt ├── run_all.sh ├── train_teachers.py ├── tsne_common_space.py ├── tsne_results ├── densenet121 │ ├── common_space_tsne_0.png │ └── common_space_tsne_1.png ├── resnet34 │ ├── common_space_tsne_0.png │ ├── common_space_tsne_1.png │ ├── feature_space_tsne_0.png │ └── feature_space_tsne_1.png └── resnet50 │ ├── common_space_tsne_0.png │ └── common_space_tsne_1.png └── utils ├── __init__.py ├── logger.py ├── stream_metrics.py └── visualizer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/.gitignore -------------------------------------------------------------------------------- /CFL.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/CFL.png -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/README.md -------------------------------------------------------------------------------- /amal.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/amal.py -------------------------------------------------------------------------------- /checkpoints/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/checkpoints/README.md -------------------------------------------------------------------------------- /data/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/data/README.md -------------------------------------------------------------------------------- /dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/dataloader.py -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/cub200.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/datasets/cub200.py -------------------------------------------------------------------------------- /datasets/stanford_dogs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/datasets/stanford_dogs.py -------------------------------------------------------------------------------- /datasets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/datasets/utils.py -------------------------------------------------------------------------------- /download_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/download_data.py -------------------------------------------------------------------------------- /kd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/kd.py -------------------------------------------------------------------------------- /logs/acc-densenet121.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/logs/acc-densenet121.png -------------------------------------------------------------------------------- /logs/acc-resnet34.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/logs/acc-resnet34.png -------------------------------------------------------------------------------- /logs/acc-resnet50.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/logs/acc-resnet50.png -------------------------------------------------------------------------------- /logs/amal_densenet121.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/logs/amal_densenet121.txt -------------------------------------------------------------------------------- /logs/amal_resnet34.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/logs/amal_resnet34.txt -------------------------------------------------------------------------------- /logs/amal_resnet50.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/logs/amal_resnet50.txt -------------------------------------------------------------------------------- /logs/draw_acc_curve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/logs/draw_acc_curve.py -------------------------------------------------------------------------------- /logs/kd_densenet121.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/logs/kd_densenet121.txt -------------------------------------------------------------------------------- /logs/kd_resnet34.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/logs/kd_resnet34.txt -------------------------------------------------------------------------------- /logs/kd_resnet50.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/logs/kd_resnet50.txt -------------------------------------------------------------------------------- /loss/__init__.py: -------------------------------------------------------------------------------- 1 | from .loss import * -------------------------------------------------------------------------------- /loss/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/loss/loss.py -------------------------------------------------------------------------------- /loss/mmd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/loss/mmd.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/cfl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/models/cfl.py -------------------------------------------------------------------------------- /models/densenet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/models/densenet.py -------------------------------------------------------------------------------- /models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/models/resnet.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/requirements.txt -------------------------------------------------------------------------------- /run_all.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/run_all.sh -------------------------------------------------------------------------------- /train_teachers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/train_teachers.py -------------------------------------------------------------------------------- /tsne_common_space.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/tsne_common_space.py -------------------------------------------------------------------------------- /tsne_results/densenet121/common_space_tsne_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/tsne_results/densenet121/common_space_tsne_0.png -------------------------------------------------------------------------------- /tsne_results/densenet121/common_space_tsne_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/tsne_results/densenet121/common_space_tsne_1.png -------------------------------------------------------------------------------- /tsne_results/resnet34/common_space_tsne_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/tsne_results/resnet34/common_space_tsne_0.png -------------------------------------------------------------------------------- /tsne_results/resnet34/common_space_tsne_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/tsne_results/resnet34/common_space_tsne_1.png -------------------------------------------------------------------------------- /tsne_results/resnet34/feature_space_tsne_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/tsne_results/resnet34/feature_space_tsne_0.png -------------------------------------------------------------------------------- /tsne_results/resnet34/feature_space_tsne_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/tsne_results/resnet34/feature_space_tsne_1.png -------------------------------------------------------------------------------- /tsne_results/resnet50/common_space_tsne_0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/tsne_results/resnet50/common_space_tsne_0.png -------------------------------------------------------------------------------- /tsne_results/resnet50/common_space_tsne_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/tsne_results/resnet50/common_space_tsne_1.png -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/stream_metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/utils/stream_metrics.py -------------------------------------------------------------------------------- /utils/visualizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/VainF/CommonFeatureLearning/HEAD/utils/visualizer.py --------------------------------------------------------------------------------