Model Weights
February 7, 2026 ยท View on GitHub
The .pth weights and corresponding model hyperparameters in .yaml files for models described in the publication are provided.
S2 Multispectral Flood Mapping Model
s2_unet_all_best.pth- benchmarked UNet model using all channels (except DEM).- Use threshold
t=0.5for inference. Achieves 89.77% F1, 91.69% Recall, 87.94% Precision on test set.
- Use threshold
s2_unetpp_all_best.pth- benchmarked UNet++ model using all channels (except DEM).- Use calibrated threshold
t=0.75for inference. Achieves 89.98% F1, 91.12% Recall, and 88.86% Precision on test set.
- Use calibrated threshold
S1 SAR Flood Mapping Model
s1_unetpp_all_best.pth- benchmarked UNet++ SAR model using all channels (except DEM).- Use calibrated threshold
t=0.85for inference. Achieves 75.32% F1, 63.61% Recall, and 92.35% Precision on test set.
- Use calibrated threshold
s1_unetpp_all_cvae_best.pth- benchmarked UNet++ SAR model using all channels (except DEM), specifically trained for CVAE despeckled SAR images.- Use calibrated threshold
t=0.87for inference. Achieves 76.06% F1, 64.82% Recall, and 92.02% Precision on test set.
- Use calibrated threshold
S1 Despeckling Model
cvae.pth- benchmarked SAR CVAE despeckler for VV and VH channels with 25.92-27.09 PSNR, 0.63-0.66 SSIM and 188.7-379.5 ENL on test set.