Views
No views yet
paper/paper_draft.md in the code repo, §4.2| Folder | Training data | Steps | Dice (native) | IoU | HD95@512 | NSD@2 |
|---|---|---|---|---|---|---|
real_only | 2594 real ISIC 2018 training images | 32,400 (100 epochs) | 90.39 | 83.79 | 32.27 | 22.0 |
hard_synthetic_augmented | real + quality-filtered hard-mask-conditioned synthetic pairs | 32,400 (step-matched) | 89.01 | 81.54 | 34.09 | 16.7 |
best.pt (checkpoint at the epoch with the best validation Dice,
used for the numbers above) and last.pt (final-epoch checkpoint).1import torch
2import segmentation_models_pytorch as smp
3
4model = smp.Segformer(encoder_name="mit_b2", classes=1)
5state = torch.load("real_only/best.pt", map_location="cpu")
6model.load_state_dict(state["model"] if "model" in state else state)
7model.eval()config.json / summary.json in each folder for the exact training configuration
and final metrics; loading code matches src/models/registry.py and
src/engine/trainer.py in the code repo.segmentation_models_pytorch's mit_b2 (SegFormer/MiT) ImageNet
weights — check that library's upstream licensing before commercial use. Trained and
fine-tuned on ISIC 2018 Task 1 (CC0 masks, publicly released dermoscopy images).