Upload RainMap Nowcasting demo model

README.md

@@ -0,0 +1,37 @@
+---
+license: mit
+pipeline_tag: image-to-image
+tags:
+- precipitation-nowcasting
+- rainmap
+- pytorch
+- demo
+---
+
+# RainMap Nowcasting
+
+This repository hosts the demo weights for
+[`TechieMoon/rainmap-nowcasting`](https://github.com/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
+
+These first weights are trained on synthetic moving rain blobs only. They are
+for validating installation, model download, inference, and the local client.
+They are not suitable for operational weather forecasting.
+
+## Files
+
+- `rainmap-nowcasting-demo.safetensors`: PyTorch state dict.
+- `model_config.json`: architecture, frame count, image size, and encoding.
+- `training_metrics.json`: synthetic-demo training metrics.
+
+## Example
+
+```bash
+python -m rainmap_nowcasting.predict --input-dir samples/input --output-dir outputs
+```

model_config.json

@@ -0,0 +1,19 @@
+{
+  "model_type": "RainMapUNet",
+  "input_frames": 6,
+  "target_frames": 6,
+  "image_size": [
+    64,
+    64
+  ],
+  "base_channels": 8,
+  "pixel_encoding": "grayscale_0_255_to_0_1",
+  "rain_scale": {
+    "min_mm_per_hour": 0.0,
+    "max_mm_per_hour": 50.0,
+    "description": "Linear placeholder scale for MVP demos; calibrate with real data before use."
+  },
+  "weights": "rainmap-nowcasting-demo.safetensors",
+  "created_at": "2026-05-06T14:03:12.088504+00:00",
+  "warning": "Demo model only. Not for operational weather forecasting."
+}

rainmap-nowcasting-demo.safetensors

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:193379c044d398a5e84de265d263d20ffd466e8d5ba087948273aa690e1dd89f
+size 492304

training_metrics.json

@@ -0,0 +1,35 @@
+{
+  "config": {
+    "train_dir": "data\\demo\\train",
+    "val_dir": "data\\demo\\val",
+    "output_dir": "runs\\demo",
+    "input_frames": 6,
+    "target_frames": 6,
+    "image_size": [
+      64,
+      64
+    ],
+    "batch_size": 4,
+    "epochs": 1,
+    "learning_rate": 0.001,
+    "base_channels": 8,
+    "num_workers": 0,
+    "seed": 42,
+    "device": "auto",
+    "amp": true,
+    "checkpoint_name": "rainmap-nowcasting-demo.safetensors"
+  },
+  "device": "cpu",
+  "train_samples": 84,
+  "val_samples": 28,
+  "history": [
+    {
+      "epoch": 1,
+      "train_mse": 0.09050992024796349,
+      "val_mse": 0.08037208499653,
+      "val_mae": 0.2364002104316439
+    }
+  ],
+  "best_val_mse": 0.08037208499653,
+  "warning": "Synthetic demo data only."
+}