Downloads · 30 days
139
12% of all-time downloads
torchgeo/fields-of-the-world
fields-of-the-world 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.
Model weights from the release at https://github.com/fieldsoftheworld/ftw-baselines/releases/tag/Pretrained-Models
Downloads · 30 days
139
12% of all-time downloads
All-time downloads
1.1K
Public
Repo size
213 MB
Likes
0
Public
Click a slice to open those files.
.pth213 MB · 100%
From the Hugging Face model README
Model weights from the release at https://github.com/fieldsoftheworld/ftw-baselines/releases/tag/Pretrained-Models
The "full" models are trained on all country data in the Fields of The World dataset, while the CCBY models are trained according to the subset described here.
Example usage:
import segmentation_models_pytorch as smp
import torch
model = smp.Unet(
encoder_name="efficientnet-b3",
encoder_weights=None,
in_channels=8,
classes=2
)
model.load_state_dict(torch.load("ftw-2class-full_unet-efficientnetb3_rgbnir_f2444768.pth", weights_only=True))