├── .gitignore ├── Dockerfile ├── README.md ├── imgs └── method.png ├── losses.py ├── main.py ├── metric ├── __init__.py ├── nearest_neighbor.py ├── plugin_ce_loss_metric.py ├── plugin_centerDist_loss_metric.py ├── plugin_cl_accuracy.py ├── plugin_contrastive_loss_metric.py ├── plugin_global_loss_metric.py ├── plugin_kd_loss_metric.py ├── plugin_recall_at_k.py ├── plugin_triplet_loss_metric.py └── recall_at_k.py ├── requirements.txt ├── requirements_cont.txt ├── resnet.py ├── train_container.sh └── trainer ├── plugins.py └── strategies.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NiccoBiondi/ContrastiveSupervisedDistillation/HEAD/.gitignore -------------------------------------------------------------------------------- /Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NiccoBiondi/ContrastiveSupervisedDistillation/HEAD/Dockerfile -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NiccoBiondi/ContrastiveSupervisedDistillation/HEAD/README.md -------------------------------------------------------------------------------- /imgs/method.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NiccoBiondi/ContrastiveSupervisedDistillation/HEAD/imgs/method.png -------------------------------------------------------------------------------- /losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NiccoBiondi/ContrastiveSupervisedDistillation/HEAD/losses.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NiccoBiondi/ContrastiveSupervisedDistillation/HEAD/main.py -------------------------------------------------------------------------------- /metric/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NiccoBiondi/ContrastiveSupervisedDistillation/HEAD/metric/__init__.py -------------------------------------------------------------------------------- /metric/nearest_neighbor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NiccoBiondi/ContrastiveSupervisedDistillation/HEAD/metric/nearest_neighbor.py -------------------------------------------------------------------------------- /metric/plugin_ce_loss_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NiccoBiondi/ContrastiveSupervisedDistillation/HEAD/metric/plugin_ce_loss_metric.py -------------------------------------------------------------------------------- /metric/plugin_centerDist_loss_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NiccoBiondi/ContrastiveSupervisedDistillation/HEAD/metric/plugin_centerDist_loss_metric.py -------------------------------------------------------------------------------- /metric/plugin_cl_accuracy.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NiccoBiondi/ContrastiveSupervisedDistillation/HEAD/metric/plugin_cl_accuracy.py -------------------------------------------------------------------------------- /metric/plugin_contrastive_loss_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NiccoBiondi/ContrastiveSupervisedDistillation/HEAD/metric/plugin_contrastive_loss_metric.py -------------------------------------------------------------------------------- /metric/plugin_global_loss_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NiccoBiondi/ContrastiveSupervisedDistillation/HEAD/metric/plugin_global_loss_metric.py -------------------------------------------------------------------------------- /metric/plugin_kd_loss_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NiccoBiondi/ContrastiveSupervisedDistillation/HEAD/metric/plugin_kd_loss_metric.py -------------------------------------------------------------------------------- /metric/plugin_recall_at_k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NiccoBiondi/ContrastiveSupervisedDistillation/HEAD/metric/plugin_recall_at_k.py -------------------------------------------------------------------------------- /metric/plugin_triplet_loss_metric.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NiccoBiondi/ContrastiveSupervisedDistillation/HEAD/metric/plugin_triplet_loss_metric.py -------------------------------------------------------------------------------- /metric/recall_at_k.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NiccoBiondi/ContrastiveSupervisedDistillation/HEAD/metric/recall_at_k.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NiccoBiondi/ContrastiveSupervisedDistillation/HEAD/requirements.txt -------------------------------------------------------------------------------- /requirements_cont.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NiccoBiondi/ContrastiveSupervisedDistillation/HEAD/requirements_cont.txt -------------------------------------------------------------------------------- /resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NiccoBiondi/ContrastiveSupervisedDistillation/HEAD/resnet.py -------------------------------------------------------------------------------- /train_container.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NiccoBiondi/ContrastiveSupervisedDistillation/HEAD/train_container.sh -------------------------------------------------------------------------------- /trainer/plugins.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NiccoBiondi/ContrastiveSupervisedDistillation/HEAD/trainer/plugins.py -------------------------------------------------------------------------------- /trainer/strategies.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/NiccoBiondi/ContrastiveSupervisedDistillation/HEAD/trainer/strategies.py --------------------------------------------------------------------------------