Support stratified sampling for change detection (#96)

* added change detection * turned off dataset multiplier
yurujaja · Oct 17, 2024 · fc3a63f · fc3a63f
1 parent 9ee5bc3
commit fc3a63f
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 3 deletions.
diff --git a/configs/dataset/spacenet7cd.yaml b/configs/dataset/spacenet7cd.yaml
@@ -13,7 +13,7 @@ j_split: 512
 multi_temporal: 2
 multi_modal: False
 
-dataset_multiplier: 5  # multiplies sample in dataset during training
+dataset_multiplier: 1  # multiplies sample in dataset during training
 minimum_temporal_gap: 5
 
 # classes

diff --git a/configs/dataset/spacenet7cd_domainshift.yaml b/configs/dataset/spacenet7cd_domainshift.yaml
@@ -13,7 +13,7 @@ j_split: 512
 multi_temporal: 2
 multi_modal: False
 
-dataset_multiplier: 5  # multiplies sample in dataset during training
+dataset_multiplier: 1  # multiplies sample in dataset during training
 minimum_temporal_gap: 5
 
 # classes

diff --git a/pangaea/utils/subset_sampler.py b/pangaea/utils/subset_sampler.py
@@ -212,7 +212,7 @@ def get_subset_indices(dataset: GeoFMDataset,
         )
         return indices
 
-    elif task == "segmentation":
+    elif task == "segmentation" or task == "change_detection":
         indices, _ = balance_seg_indices(
             dataset, strategy=strategy, label_fraction=label_fraction, num_bins=num_bins, logger=logger
         )