Skip to content

Commit

Permalink
added change detection
Browse files Browse the repository at this point in the history
  • Loading branch information
SebastianHafner committed Oct 16, 2024
1 parent 9ee5bc3 commit 85d9b70
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion pangaea/utils/subset_sampler.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
)
Expand Down

0 comments on commit 85d9b70

Please sign in to comment.