SLynMi's picture
Update model card with checkpoint summary table
c3002d0 verified
|
Raw
History Blame Contribute Delete
1.91 kB
metadata
license: apache-2.0
tags:
  - image-segmentation
  - building-footprint
  - remote-sensing
datasets:
  - BONAI
model-index:
  - name: v6_8ch
    results:
      - task:
          type: image-segmentation
        metrics:
          - name: mIoU
            type: miou
            value: 0.5249

Building Footprint Extraction — FootprintNet checkpoints

Checkpoints for the BONAI building-footprint extraction project. Each folder holds the best checkpoint for one model run. The model family lineage is V0–V3 (ResNet-50 base), V4-Attention (RegNetY-6.4GF), V5-Attention (18-channel pruned), and V6 (SegFormer mit_b2) which is the current family.

Current best model: v6_8ch

File Description
v6_8ch/best_v6_8ch.pt Best checkpoint, 8 input channels, mIoU 0.5249 (TTA 0.5317)
v6_8ch/best_v6_8ch_stripped.pt Deployable stripped weights (inference only)
v6_8ch/eval_summary.txt Held-out eval metrics
v6_8ch/tta_summary.txt HFlip+VFlip TTA comparison

Best config: in_channels=8, SegFormer mit_b2, batch_size=24, accum_steps=2, lr=3e-5, warmup_epochs=8, early_stop_patience=30, grad_checkpoint=true, amp=true.

Model summary

Folder Input channels Epoch mIoU Status
v6_8ch 8 129 0.5249 Current best
v6_19ch 19 83 0.4977 Deprecated
v6_14ch 14 63 0.5065 Deprecated (Phase 69)
v6_11ch 11 73 0.4958 Deprecated (Phase 70)
v5_attn_1 18 24 0.3867 Deprecated
v5_attn_1_continued 18 35 0.4109 Deprecated
v4_attn_retry_p41 19 23 0.4046 Deprecated
v4_attn_synthetic_roof_p44 19 34 0.4025 Deprecated
channel_attention_phase41_earlystop 2 0.3864 Probe (early stop)
channel_attention_phase44_earlystop 3 0.3713 Probe (early stop)