Added tests/conf/south_america_soybean_s2.yaml

microsoft · Feb 16, 2024 · 7252d5a · 7252d5a
1 parent e8c94b9
commit 7252d5a
Showing 1 changed file with 18 additions and 0 deletions.
diff --git a/tests/conf/south_america_soybean_s2.yaml b/tests/conf/south_america_soybean_s2.yaml
@@ -0,0 +1,18 @@
+model:
+  class_path: SemanticSegmentationTask
+  init_args:
+    loss: "ce"
+    model: "deeplabv3+"
+    backbone: "resnet34"
+    in_channels: 13
+    num_classes: 2
+    num_filters: 1
+    ignore_index: 4
+data:
+  class_path: SouthAmericaSoybeanSentinel2DataModule
+  init_args:
+    batch_size: 2
+    patch_size: 16
+  dict_kwargs:
+    south_america_soybean_paths: "tests/data/south_america_soybean"
+    sentinel2_paths: "tests/data/sentinel2"