isaaccorley commited on
Commit
334c2f1
·
verified ·
1 Parent(s): 0b7ff1e

Upload README.md with huggingface_hub

Browse files
Files changed (1) hide show
  1. README.md +7 -7
README.md CHANGED
@@ -22,7 +22,7 @@ pipeline_tag: image-segmentation
22
  3m PlanetScope field-boundary segmentation model from **Fields of the Planet
23
  (FTP)**, a PlanetScope companion to [Fields of the World
24
  (FTW)](https://github.com/fieldsoftheworld/ftw-baselines). Larger-backbone
25
- variant of the paper's headline model ([`ftp-b3`](https://huggingface.co/taylor-geospatial/ftp-b3)).
26
 
27
  U-Net decoder over an EfficientNet-B7 encoder, trained on paired planting-
28
  and harvest-window PlanetScope surface-reflectance imagery (8 input
@@ -59,8 +59,8 @@ logits = model(image) # image: (B, 8, H, W) float32, 2 seasonal windows x 4 ban
59
 
60
  ### Lightning checkpoint (raw `smp.Unet`)
61
 
62
- The checkpoint's `state_dict` is just an `smp.Unet` under a `model.` prefix
63
- unless you need the Lightning training wrapper, `smp.from_pretrained` above
64
  is simpler.
65
 
66
  ```python
@@ -95,7 +95,7 @@ model = exported.module()
95
  logits = model(image) # image: (B, 8, 512, 512) float32, any batch size
96
  ```
97
 
98
- Output is 3-class logits (background / field / boundary); argmax + the
99
  repo's watershed post-processing (`scripts/eval/postprocess_eval.py`)
100
  recovers instance polygons.
101
 
@@ -121,9 +121,9 @@ confidence setting. † Pixel IoU is not comparable across sensors due to
121
  differences in resolution. ‡ PQ for small (<0.5 ha), medium (0.5-2 ha), and
122
  large (>2 ha) ground-truth fields.
123
 
124
- The B3 variant scores marginally higher PQ (35.5 vs 35.4) at ~5x fewer
125
- parameters; B7 trades that for better pixel IoU (74.2 vs 68.8) and
126
- medium-field PQ (40.6 vs 39.2).
127
 
128
  ## Citation
129
 
 
22
  3m PlanetScope field-boundary segmentation model from **Fields of the Planet
23
  (FTP)**, a PlanetScope companion to [Fields of the World
24
  (FTW)](https://github.com/fieldsoftheworld/ftw-baselines). Larger-backbone
25
+ variant of the paper's main reported model ([`ftp-b3`](https://huggingface.co/taylor-geospatial/ftp-b3)).
26
 
27
  U-Net decoder over an EfficientNet-B7 encoder, trained on paired planting-
28
  and harvest-window PlanetScope surface-reflectance imagery (8 input
 
59
 
60
  ### Lightning checkpoint (raw `smp.Unet`)
61
 
62
+ The checkpoint's `state_dict` is just an `smp.Unet` under a `model.` prefix.
63
+ Unless you need the Lightning training wrapper, `smp.from_pretrained` above
64
  is simpler.
65
 
66
  ```python
 
95
  logits = model(image) # image: (B, 8, 512, 512) float32, any batch size
96
  ```
97
 
98
+ Output is 3-class logits (background / field / boundary). Argmax plus the
99
  repo's watershed post-processing (`scripts/eval/postprocess_eval.py`)
100
  recovers instance polygons.
101
 
 
121
  differences in resolution. ‡ PQ for small (<0.5 ha), medium (0.5-2 ha), and
122
  large (>2 ha) ground-truth fields.
123
 
124
+ The B3 variant scores higher PQ (35.5 vs 35.4) at about 5x fewer parameters,
125
+ but B7 has better pixel IoU (74.2 vs 68.8) and medium-field PQ (40.6 vs
126
+ 39.2).
127
 
128
  ## Citation
129