File size: 1,008 Bytes
b62d916 8bbbbb6 b62d916 9335db2 b62d916 9335db2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
---
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). |