Update README.md

README.md

@@ -1,3 +1,22 @@
----
-license: mit
----
+---
+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))
+```