Farhan2000/skin-lesion-segmentation-unet

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