Views
No views yet
model.safetensors plus a config.json carrying model configuration, class names, per-class decision thresholds, and recommended sliding-window patch size (512).| Subfolder | Architecture | Encoder |
|---|---|---|
unet | U-Net | EfficientNet-B4 |
unet_scse | U-Net + SCSE attention | EfficientNet-B4 |
unetpp | U-Net++ | EfficientNet-B4 |
deeplabv3p | DeepLabV3+ | EfficientNet-B4 |
fpn | FPN | EfficientNet-B4 |
pan | PAN | EfficientNet-B4 |
1pip install torch segmentation-models-pytorch safetensors huggingface_hub pillow numpy
2
3python inference.py \
4 --hf-repo davidclara/siegfried-maps-segmentation \
5 --model-name unetpp \
6 --image map.png \
7 --out-dir predictions/inference.py is a minimal example: sliding-window prediction with stride = patch_size // 2, ImageNet normalization, sigmoid + per-class thresholding, writing one binary PNG per class to --out-dir. For the full training and inference pipeline see the GitHub repository.