/SkinDistilVit

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

Primary LanguagePython

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.

Train

Follow the train files for each individual training.

Evaluation

Follow the eval files for each type of model.