sgonzalez2000

dermai-vit-base-patch16-224

sgonzalez2000/dermai-vit-base-patch16-224

dermai-vit-base-patch16-224 is a image classification model from sgonzalez2000. Use it when you need a label for an image. It is set up for transformers. The card lists the license as apache-2.0.

Fine-tuned google/vit-base-patch16-224 for 7-class skin lesion classification on HAM10000. Part of the DermAI explainability project comparing CNN and Vision Transformer explanations.