def build_dataset_from_catalog(catalog: Catalog, class_config: ClassConfig,
use_hand: bool,
three_class: bool) -> DatasetConfig:
# Read taining scenes from STAC
train_collection = catalog.get_child(id="train")
train_scenes = make_scenes(train_collection, use_hand, three_class)
# Read validation scenes from STAC
validation_collection = catalog.get_child(id="validation")
validation_scenes = make_scenes(validation_collection, use_hand,
three_class)
# Read test scenes from STAC
test_collection = catalog.get_child(id="test")
test_scenes = make_scenes(test_collection, use_hand, three_class)
return DatasetConfig(
class_config=class_config,
train_scenes=train_scenes,
validation_scenes=(validation_scenes + test_scenes),
)
def build_dataset_from_catalog(catalog: Catalog, channel_order: [int],
class_config: ClassConfig) -> DatasetConfig:
# Read taining scenes from STAC
train_collection = catalog.get_child(id="training_imagery")
train_scenes = make_scenes(train_collection, channel_order)
# Read validation scenes from STAC
validation_collection = catalog.get_child(id="validation_imagery")
validation_scenes = make_scenes(validation_collection, channel_order)
# Read testing scenes from STAC
test_collection_0 = catalog.get_child(id="test_imagery_0")
test_scenes_0 = make_scenes(test_collection_0, channel_order)
test_collection_1 = catalog.get_child(id="test_imagery_1")
test_scenes_1 = make_scenes(test_collection_1, channel_order)
return DatasetConfig(
class_config=class_config,
train_scenes=train_scenes,
validation_scenes=(test_scenes_0 + test_scenes_1),
)
def build_dataset_from_catalog(catalog: Catalog, channel_order: [int],
class_config: ClassConfig) -> DatasetConfig:
# Read taining scenes from STAC
train_collection = catalog.get_child(id="train")
train_scenes = make_scenes(train_collection, channel_order)
# Read validation scenes from STAC
validation_collection = catalog.get_child(id="validation")
validation_scenes = make_scenes(validation_collection, channel_order)
# Read testing scenes from STAC
test_collection = catalog.get_child(id="test")
test_scenes = make_scenes(test_collection, channel_order)
return DatasetConfig(
class_config=class_config,
train_scenes=train_scenes,
test_scenes=test_scenes,
validation_scenes=validation_scenes,
img_channels=3,
)
