RainMap Nowcasting
This repository hosts model weights for
TechieMoon/rainmap-nowcasting.
Intended use
The model accepts 6 grayscale rain map frames and predicts the next 6 grayscale
frames. Pixel intensity is interpreted as 0..255 -> 0..1 rain intensity.
Important warning
Small SEVIR VIL subset fine-tuned benchmark model. Not for operational weather forecasting.
Model
- Dataset:
SEVIR VIL mini local fine-tune - Weights:
rainmap-nowcasting-sevir-finetuned.safetensors - Input frames:
6 - Target frames:
6 - Image size:
[64, 64]
Files
rainmap-nowcasting-sevir-finetuned.safetensors: PyTorch state dict.model_config.json: architecture, frame count, image size, and encoding.training_metrics.json: training metrics.
Benchmark
- Dataset:
data\sevir_mini\local\val - Sequences: 4
| Model | MAE | RMSE | CSI@0.3 | HSS@0.3 | ETS@0.3 | FSS@0.3/w15 |
|---|---|---|---|---|---|---|
| persistence | 0.0406 | 0.0930 | 0.5944 | 0.7117 | 0.5524 | 0.9675 |
| base | 0.0764 | 0.0983 | 0.6432 | 0.7520 | 0.6026 | 0.9605 |
| fine_tuned | 0.0694 | 0.0888 | 0.6699 | 0.7747 | 0.6323 | 0.9727 |
Small-subset benchmark results are not operational forecasting claims.
Example
python -m rainmap_nowcasting.predict --input-dir samples/input --output-dir outputs