Update README.md
Browse files
README.md
CHANGED
|
@@ -18,21 +18,6 @@ pipeline_tag: image-classification
|
|
| 18 |
[](https://neurips.cc/virtual/2025/loc/san-diego/135896)
|
| 19 |
[](https://github.com/leharris3/satformer)
|
| 20 |
|
| 21 |
-
SaTformer is a Vision Transformer adapted for spatio-temporal precipitation nowcasting from geostationary satellite (HRIT) imagery. It won **1st place** in the NeurIPS 2025 CUMSUM challenge.
|
| 22 |
-
|
| 23 |
-
## Model Details
|
| 24 |
-
|
| 25 |
-
| Parameter | Value |
|
| 26 |
-
|---|---|
|
| 27 |
-
| Architecture | Vision Transformer (adapted from TimeSformer) |
|
| 28 |
-
| Attention | Joint space-time (ST²) |
|
| 29 |
-
| Embedding dim | 512 |
|
| 30 |
-
| Depth | 12 blocks |
|
| 31 |
-
| Heads | 8 (dim 64) |
|
| 32 |
-
| Input | 4 frames x 11 channels x 32x32 |
|
| 33 |
-
| Output | 64 precipitation bins (classification) |
|
| 34 |
-
| Patch size | 4x4 |
|
| 35 |
-
|
| 36 |
## Usage
|
| 37 |
|
| 38 |
```python
|
|
|
|
| 18 |
[](https://neurips.cc/virtual/2025/loc/san-diego/135896)
|
| 19 |
[](https://github.com/leharris3/satformer)
|
| 20 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 21 |
## Usage
|
| 22 |
|
| 23 |
```python
|