zmgul commited on
Commit
69866f6
·
verified ·
1 Parent(s): a19bc5c

update model card

Browse files
Files changed (1) hide show
  1. README.md +6 -5
README.md CHANGED
@@ -33,8 +33,6 @@ Prototypical network (ProtoNet) based **1-way 5-shot binary segmentation** on [L
33
 
34
  **Key finding:** Pre-training source dominates other factors — self-supervised **SeCo** outperforms all supervised sources, including domain-aligned Million-AID. Δ_pretrain = 0.103 (≈ 2× backbone effect, ≈ 8× fine-tuning effect).
35
 
36
- 5-fold CV mean ± std across all 90 runs is documented in the [thesis (PDF)](https://github.com/zmgul/few-shot-woodland-segmentation/blob/main/docs/TEZ.pdf).
37
-
38
  ## Dataset
39
 
40
  [LandCover.ai v1](https://landcover.ai/) — 41 high-resolution aerial images (25–50 cm/pixel) over Poland. 4 labeled land-cover classes + background. Woodland (33.3%) is held out as the **novel class**; building, water, road are **base classes** used for episodic training.
@@ -45,10 +43,13 @@ Prototypical network (ProtoNet) based **1-way 5-shot binary segmentation** on [L
45
  import torch
46
  from huggingface_hub import hf_hub_download
47
  from src.model import ProtoNet # from GitHub repo
48
- from src.config import Config
49
 
50
- cfg = Config(BACKBONE="resnet50", PRETRAINED="seco", UNFREEZE_FROM="none")
51
- model = ProtoNet(cfg)
 
 
 
 
52
 
53
  ckpt_path = hf_hub_download(
54
  repo_id="zmgul/few-shot-woodland-segmentation",
 
33
 
34
  **Key finding:** Pre-training source dominates other factors — self-supervised **SeCo** outperforms all supervised sources, including domain-aligned Million-AID. Δ_pretrain = 0.103 (≈ 2× backbone effect, ≈ 8× fine-tuning effect).
35
 
 
 
36
  ## Dataset
37
 
38
  [LandCover.ai v1](https://landcover.ai/) — 41 high-resolution aerial images (25–50 cm/pixel) over Poland. 4 labeled land-cover classes + background. Woodland (33.3%) is held out as the **novel class**; building, water, road are **base classes** used for episodic training.
 
43
  import torch
44
  from huggingface_hub import hf_hub_download
45
  from src.model import ProtoNet # from GitHub repo
 
46
 
47
+ # Backbone must match the checkpoint; pretrained init is overwritten by load_state_dict
48
+ model = ProtoNet(
49
+ backbone_name="resnet50",
50
+ pretrained="imagenet_v1",
51
+ unfreeze_from="none",
52
+ )
53
 
54
  ckpt_path = hf_hub_download(
55
  repo_id="zmgul/few-shot-woodland-segmentation",