v1.0.0 - Pretrained Models

Pretrained PyTorch model weights for vitiligo classification (trained 2021-04-25).

Assets

• densenet_vitiligo.pth (27 MB) - DenseNet-based classifier
• resnet_vitiligo.pth (43 MB) - ResNet-based classifier
• vgg_vitiligo.pth (492 MB) - VGG-based classifier

Usage

import torch
model = torch.load('densenet_vitiligo.pth', map_location='cpu')