Modify configs for multiclass and binary multisource

NRCan · May 16, 2024 · 563eef5 · 563eef5
1 parent d9b2caa
commit 563eef5
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 2 deletions.
diff --git a/config/dataset/test_ci_segmentation_binary_RGBL.yaml b/config/dataset/test_ci_segmentation_binary_RGBL.yaml
@@ -14,6 +14,6 @@ dataset:
   attribute_field: properties/class
   attribute_values: [1]
   class_name: # will follow in the next version
-  classes_dict: {'WAER':1, 'FORE':2, 'ROAI':3, 'BUIL':4}
+  classes_dict: {'BUIL':1}
   class_weights:
   ignore_index: 255
diff --git a/config/dataset/test_ci_segmentation_multiclass_RGBL.yaml b/config/dataset/test_ci_segmentation_multiclass_RGBL.yaml
@@ -0,0 +1,19 @@
+# @package _global_
+dataset:
+  # dataset-wide
+  name:
+  raw_data_csv: ${general.raw_data_csv}
+  raw_data_dir: ${general.raw_data_dir}
+  download_data: False
+
+  # imagery
+  bands: [R,G,B]
+  dsm_dir: null
+
+  # ground truth
+  attribute_field: properties/class
+  attribute_values: [1]
+  class_name: # will follow in the next version
+  classes_dict: {'WAER':1, 'FORE':2, 'ROAI':3, 'BUIL':4}
+  class_weights:
+  ignore_index: 255
diff --git a/config/gdl_config_template.yaml b/config/gdl_config_template.yaml
@@ -7,7 +7,7 @@ defaults:
   - optimizer: adamw
   - callbacks: default_callbacks
   - scheduler: plateau
-  - dataset: test_ci_segmentation_binary_RGBL
+  - dataset: test_ci_segmentation_binary
   - augmentation: basic_augmentation_segmentation
   - tracker: # set logger here or use command line (e.g. `python GDL.py tracker=mlflow`)
   - visualization: default_visualization