SkinDistilVit

The repo with the code from our paper, SkinDistilViT: Lightweight Vision Transformer for Skin Lesion Classification

Download the dataset

Download the official ISIC2019 datasert from the original website and place it in "../ISIC2019/" for the code to work.

Follow the train files for each individual training.

Follow the eval files for each type of model.

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
data		data
distillation		distillation
model		model
pseudolabeling		pseudolabeling
README.md		README.md
__init__.py		__init__.py
attack_distil.py		attack_distil.py
cancer_Vs_non_cancer.py		cancer_Vs_non_cancer.py
distil.py		distil.py
distil_matryoshka.py		distil_matryoshka.py
eval.py		eval.py
eval_effnet.py		eval_effnet.py
eval_fcvity.py		eval_fcvity.py
plot.py		plot.py
plot_tsne.py		plot_tsne.py
train.py		train.py
train_distilvit.py		train_distilvit.py
train_effnet.py		train_effnet.py
train_fcvit_probs.py		train_fcvit_probs.py
train_pseudolabels.py		train_pseudolabels.py