Downloads · 30 days
0
skin-lesion-segmentation-unet
Farhan2000/skin-lesion-segmentation-unet
skin-lesion-segmentation-unet is a machine learning model from Farhan2000. Use it for the machine learning task on the model card, and read the license before you ship it in a product. The card lists the license as mit.
Trained on ISIC 2017 Task 1 lesion-boundary masks. Educational project — not diagnostic-grade, not a medical device.
Access
Public
Updated Sep 18, 2026
Repo size
97.9 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pth97.9 MB · 100%
At a glance
- License
- mit
- Access
- Public
- Created
- Sep 18, 2026
- Updated
- Sep 18, 2026
- SHA
- fabc3db4
Model card
From the Hugging Face model README
Skin Lesion Segmentation (U-Net, ResNet34 encoder)
Trained on ISIC 2017 Task 1 lesion-boundary masks. Educational project — not diagnostic-grade, not a medical device.
Load
import segmentation_models_pytorch as smp, torch
from huggingface_hub import hf_hub_download
model = smp.Unet(encoder_name="resnet34", encoder_weights=None, in_channels=3, classes=1)
weights_path = hf_hub_download(repo_id="Farhan2000/skin-lesion-segmentation-unet", filename="model.pth")
model.load_state_dict(torch.load(weights_path, map_location="cpu"))
model.eval()
Mean validation Dice: 0.8520
- License
- mit
- Created
- Sep 18, 2026
- Updated
- Sep 18, 2026