File size: 711 Bytes
479aaf2 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 |
---
license: mit
---
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](https://github.com/fieldsoftheworld/ftw-baselines?tab=readme-ov-file#cc-byor-equivalent-trained-models).
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))
``` |