Downloads · 30 days
0
skinvision-ai-model
pthennakoon25/skinvision-ai-model
skinvision-ai-model is a image classification model from pthennakoon25. Use it when you need a label for an image. It is set up for timm. The card lists the license as mit.
EfficientNet-B0 (via timm), fine-tuned for 15 epochs on the HAM10000 dataset to classify skin lesion images into 7 diagnostic categories.
Access
Public
Updated Aug 25, 2026
Repo size
16.4 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pth16.4 MB · 100%
At a glance
- Task
- Image Classification
- Library
- timm
- License
- mit
- Access
- Public
- Created
- Aug 25, 2026
- Updated
- Aug 25, 2026
- SHA
- 9429e5a4
Model card
From the Hugging Face model README
SkinVision AI — EfficientNet-B0 (HAM10000)
EfficientNet-B0 (via timm), fine-tuned for 15 epochs on the HAM10000 dataset to classify skin lesion images into 7 diagnostic categories.
Educational project — not a medical diagnostic tool.
Results (held-out test set, 1,431 images)
| Metric | Score |
|---|---|
| Test accuracy | 82.4% |
| Macro F1 | 0.663 |
| Macro ROC-AUC (OVR) | 0.953 |
Classes
akiec (Actinic Keratosis), bcc (Basal Cell Carcinoma), bkl (Benign Keratosis), df (Dermatofibroma), mel (Melanoma), nv (Melanocytic Nevus), vasc (Vascular Lesion)
Usage
Full inference pipeline, Grad-CAM explainability, and app code: github.com/Pabodha123/Skin-Vision-AI
import timm
import torch
model = timm.create_model("efficientnet_b0", pretrained=False, num_classes=7)
model.load_state_dict(torch.load("best_model.pth", map_location="cpu"))
model.eval()
- Task
- Image Classification
- Library
- timm
- License
- mit
- Created
- Aug 25, 2026
- Updated
- Aug 25, 2026