sugam24's picture
card epoch 62
4223513 verified
|
Raw
History Blame Contribute Delete
1.19 kB
metadata
tags:
  - semantic-segmentation
  - remote-sensing
  - unet
  - resnet18
  - geonusaf
library_name: segmentation-models-pytorch

GeoNUSAF - UNet-ResNet18 - block split, fold 1

Kathmandu Valley land-use segmentation, 6 classes, ignore_index=255. Weights are the EMA weights (decay 0.999), not the raw final weights.

field value
architecture smp.Unet, encoder resnet18 (ImageNet), decoder [128, 64, 32, 16, 8]
params 12.46 M
arch version unet-r18-v1
split mode block
fold 1 of 3
seed 42
input 512x512, ImageNet norm, effective GSD 0.586 m/px
regularization wd 0.01 (norm/bias exempt), ls 0.05, drop 0.1, EMA 0.999
classes Residential, Road, River, Forest, UnusedLand, Agricultural
best epoch 62
val mIoU 0.4516
val mF1 0.5807
val OA 0.7898
val kappa 0.6255

Per-class (validation)

class IoU F1
Residential 0.8310 0.9077
Road 0.3965 0.5678
River 0.1329 0.2346
Forest 0.6495 0.7875
UnusedLand 0.1763 0.2998
Agricultural 0.5232 0.6870

Checkpoint best.pt holds model_state (EMA) plus cfg, metrics and arch_sig.