Image Segmentation
PyTorch
timm
segformer
mit-b4
transformer
segmentation-models-pytorch
remote-sensing
sentinel-2
rgb
cloud-detection
Instructions to use Burdenthrive/cloud-detection-segformer-mit_b4-RGB with libraries, inference providers, notebooks, and local apps. Follow these links to get started.
- Libraries
- timm
How to use Burdenthrive/cloud-detection-segformer-mit_b4-RGB with timm:
import timm model = timm.create_model("hf_hub:Burdenthrive/cloud-detection-segformer-mit_b4-RGB", pretrained=True) - Notebooks
- Google Colab
- Kaggle
Update README.md
Browse files
README.md
CHANGED
|
@@ -30,7 +30,7 @@ This model predicts per‑pixel labels among: **clear**, **thick cloud**, **thin
|
|
| 30 |
|
| 31 |
- **Input:** Sentinel‑2 L1C RGB tiles/patches (float32, shape `B×3×512×512`, bands **B04‑B03‑B02**).
|
| 32 |
- **Backbone:** `mit_b4` (MiT encoder via `segmentation_models_pytorch`).
|
| 33 |
-
- **Output:** Logits `B×4×
|
| 34 |
- **Files:** `model.py`, `config.json`, and weights.
|
| 35 |
|
| 36 |
---
|
|
|
|
| 30 |
|
| 31 |
- **Input:** Sentinel‑2 L1C RGB tiles/patches (float32, shape `B×3×512×512`, bands **B04‑B03‑B02**).
|
| 32 |
- **Backbone:** `mit_b4` (MiT encoder via `segmentation_models_pytorch`).
|
| 33 |
+
- **Output:** Logits `B×4×512×512` (apply softmax + argmax).
|
| 34 |
- **Files:** `model.py`, `config.json`, and weights.
|
| 35 |
|
| 36 |
---
|