A newer version of the Gradio SDK is available:
5.33.0
metadata
title: Skin Lesion Segmentation Demo
emoji: 💻
colorFrom: pink
colorTo: purple
sdk: gradio
sdk_version: 5.31.0
app_file: app.py
pinned: false
license: mit
🧴 Skin Lesion Segmentation Demo
This interactive web application allows you to upload a skin image, select a segmentation model, and visualize the predicted lesion mask with optional post-processing.
🧠 Available Models
- UNet
- SegFormer
- Inception-based CNN
- KMeans (Unsupervised)
- Gaussian Mixture Model (Unsupervised)
🔧 Postprocessing Options
- None
- Morphological Open
- Morphological Close
- Erosion
- Dilation
💡 How it Works
The model takes in a skin lesion image (resized to 128×128), performs segmentation, and visualizes both:
- A binary predicted mask
- A color overlay on the original image
🛠️ Built With
- PyTorch & torchvision
- scikit-learn
- Hugging Face Transformers
- Gradio
- OpenCV
Demo created by Theodore Ioannidis (https://github.com/TheodoreIoannidis).