Upload README.md with huggingface_hub
Browse files
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
|
| 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 |
-
|
| 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)
|
| 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
|
| 125 |
-
|
| 126 |
-
|
| 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 |
|