Downloads · 30 days
19
9% of all-time downloads
ai4g_flood
torchgeo/ai4g_flood
ai4g_flood is a machine learning model from torchgeo. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for torchgeo. The card lists the license as mit.
All-time downloads
214
Public
Repo size
26.8 MB
Likes
0
Public
Repo makeup
Click a slice to open those files.
.pth26.8 MB · 100%
At a glance
- Library
- torchgeo
- License
- mit
- Access
- Public
- Created
- May 23, 2025
- Updated
- Sep 29, 2026
- SHA
- cdc117c1
Model card
From the Hugging Face model README
Model Weights extracted below:
import os
import hashlib
import torch
import segmentation_models_pytorch as smp
url = "https://github.com/microsoft/ai4g-flood/raw/refs/heads/main/models/ai4g_sar_model.ckpt"
state_dict = torch.hub.load_state_dict_from_url(url, weights_only=False, map_location="cpu")["state_dict"]
state_dict = {k.replace("model.model.", ""): v for k, v in state_dict.items() if "model.model." in k}
model = smp.Unet(
encoder_name="mobilenet_v2",
encoder_weights=None,
in_channels=2,
classes=2,
)
model.load_state_dict(state_dict, strict=True)
filename = "unet_mobilenetv2_sentinel1_ai4g_flood.pth"
torch.save(model.state_dict(), filename)
md5 = hashlib.md5(open(filename, "rb").read()).hexdigest()[:8]
os.rename(filename, filename.replace(".pth", f"-{md5}.pth"))
- Library
- torchgeo
- License
- mit
- Created
- May 23, 2025
- Updated
- Sep 29, 2026