sugam24's picture
card epoch 0
4731937 verified
|
Raw
History Blame
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 0
val mIoU 0.0629
val mF1 0.1128
val OA 0.1808
val kappa -0.0059

Per-class (validation)

class IoU F1
Residential 0.1800 0.3051
Road 0.0238 0.0465
River 0.0080 0.0158
Forest 0.0220 0.0430
UnusedLand 0.0541 0.1027
Agricultural 0.0893 0.1639

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