From 85d9b703c65f98b1923bf00e8dc39f0dc547e970 Mon Sep 17 00:00:00 2001 From: SebastianHafner Date: Wed, 16 Oct 2024 18:43:33 +0200 Subject: [PATCH] added change detection --- pangaea/utils/subset_sampler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pangaea/utils/subset_sampler.py b/pangaea/utils/subset_sampler.py index a8f3796..e833736 100644 --- 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 )