Add FireWx-FM input channel contract
Browse files- models/wildfire_fm/README.md +16 -0
models/wildfire_fm/README.md
CHANGED
|
@@ -12,3 +12,19 @@ models/wildfire_fm/checkpoints/seed_123/best_firms_prauc.pt
|
|
| 12 |
```
|
| 13 |
|
| 14 |
Use `modeling_unet.py` to instantiate the compact U-Net architecture before loading a checkpoint state dict.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 12 |
```
|
| 13 |
|
| 14 |
Use `modeling_unet.py` to instantiate the compact U-Net architecture before loading a checkpoint state dict.
|
| 15 |
+
|
| 16 |
+
## Input channels
|
| 17 |
+
|
| 18 |
+
The checkpoints expect a fixed 16-channel tensor in `[channel, y, x]` order.
|
| 19 |
+
A machine-readable copy is stored in `input_channels.json`.
|
| 20 |
+
|
| 21 |
+
| Channels | Names | Source |
|
| 22 |
+
|---:|---|---|
|
| 23 |
+
| 0-9 | `t2m`, `d2m`, `u10`, `v10`, `cape`, `sp`, `blh`, `vis`, `prate`, `tp` | NOAA HRRR |
|
| 24 |
+
| 10 | `firewx_valid` | Dynamic/input validity mask for this regional cache |
|
| 25 |
+
| 11 | `static_valid` | Static reprojection validity mask |
|
| 26 |
+
| 12-15 | `fuel_fbfm40`, `canopy_cover`, `housing_density`, `population` | LANDFIRE, WRC housing density, LandScan |
|
| 27 |
+
|
| 28 |
+
FIRMS detections are used to derive the occupancy target, not as an input
|
| 29 |
+
channel. WFIGS and MTBS are supporting event-level resources and are not part of
|
| 30 |
+
the pretrained occupancy input.
|