update model card
Browse files
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 |
-
|
| 51 |
-
model = ProtoNet(
|
|
|
|
|
|
|
|
|
|
|
|
|
| 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",
|