Views
No views yet
model_f{N}.safetensors per cross-validation fold plus a shared config.json carrying model configuration, class names, per-fold and ensemble decision thresholds, normalization mode, recommended sliding-window patch size (512), and number of folds.| Subfolder | Architecture | Encoder | Folds |
|---|---|---|---|
unet_scse | U-Net + SCSE attention | EfficientNet-B3 | 2 |
1pip install torch segmentation-models-pytorch safetensors huggingface_hub pillow numpy
2
3python inference.py \
4 --hf-repo davidclara/building-block-vectorization \
5 --model-name unet_scse \
6 --image map.jpg \
7 --out mask.pnginference.py is a minimal example: Gaussian-weighted sliding-window prediction with stride = patch_size // 2, normalization driven by config.json (simple = /255, imagenet = mean/std), sigmoid averaged across folds per patch, thresholding with the ensemble threshold, and writing a single binary PNG mask. For the full training and inference pipeline see the GitHub repository.