Get coverage for Chesapeake CVPR prior

microsoft · Dec 28, 2021 · e3811cf · e3811cf
1 parent 66c015c
commit e3811cf
Show file tree

Hide file tree

Showing 5 changed files with 33 additions and 1 deletion.
diff --git a/conf/task_defaults/chesapeake_cvpr_5.yaml b/conf/task_defaults/chesapeake_cvpr_5.yaml
@@ -26,3 +26,4 @@ experiment:
     batch_size: 2
     num_workers: 0
     class_set: ${experiment.module.num_classes}
+    use_prior_labels: False
diff --git a/conf/task_defaults/chesapeake_cvpr_7.yaml b/conf/task_defaults/chesapeake_cvpr_7.yaml
@@ -26,3 +26,4 @@ experiment:
     batch_size: 2
     num_workers: 0
     class_set: ${experiment.module.num_classes}
+    use_prior_labels: False
diff --git a/conf/task_defaults/chesapeake_cvpr_prior.yaml b/conf/task_defaults/chesapeake_cvpr_prior.yaml
@@ -0,0 +1,29 @@
+experiment:
+  task: "chesapeake_cvpr"
+  module:
+    loss: "ce"
+    segmentation_model: "unet"
+    encoder_name: "resnet50"
+    encoder_weights: null
+    encoder_output_stride: 16
+    learning_rate: 1e-3
+    learning_rate_schedule_patience: 6
+    in_channels: 4
+    num_classes: 5
+    num_filters: 1
+    ignore_zeros: False
+    imagenet_pretraining: False
+  datamodule:
+    root_dir: "tests/data/chesapeake/cvpr"
+    train_splits:
+    - "de-test"
+    val_splits:
+    - "de-test"
+    test_splits:
+    - "de-test"
+    patches_per_tile: 2
+    patch_size: 64
+    batch_size: 2
+    num_workers: 0
+    class_set: ${experiment.module.num_classes}
+    use_prior_labels: True
diff --git a/tests/trainers/test_segmentation.py b/tests/trainers/test_segmentation.py
@@ -23,7 +23,7 @@ class TestSemanticSegmentationTask:
     @pytest.mark.parametrize(
         "name,classname",
         [
-            ("chesapeake_cvpr_7", ChesapeakeCVPRDataModule),
+            ("chesapeake_cvpr_prior", ChesapeakeCVPRDataModule),
             ("etci2021", ETCI2021DataModule),
             ("landcoverai", LandCoverAIDataModule),
             ("naipchesapeake", NAIPChesapeakeDataModule),

diff --git a/train.py b/train.py
@@ -49,6 +49,7 @@
     "byol": (BYOLTask, ChesapeakeCVPRDataModule),
     "chesapeake_cvpr_5": (ChesapeakeCVPRSegmentationTask, ChesapeakeCVPRDataModule),
     "chesapeake_cvpr_7": (ChesapeakeCVPRSegmentationTask, ChesapeakeCVPRDataModule),
+    "chesapeake_cvpr_prior": (ChesapeakeCVPRSegmentationTask, ChesapeakeCVPRDataModule),
     "cowc_counting": (RegressionTask, COWCCountingDataModule),
     "cyclone": (RegressionTask, CycloneDataModule),
     "eurosat": (ClassificationTask, EuroSATDataModule),