Datasets

RELAI Segmentation Dataset

class relai.datasets.segmentation_dataset.RELAISegmentationDataset(*args, **kwargs)

Bases: RELAIDataset

abstract get_dataset(batch_size: int)

Method to generate batches of dataset

Example

segmentation_dataset = RELAISegmentationDataset(…) for batch in segmentation_dataset.get_dataset(batch_size):

process_batch(batch)

abstract load_dataset()

Method to load the dataset

class relai.datasets.segmentation_dataset.RELAISegmentationDatasetCSV(dataset_path: str)

Bases: RELAISegmentationDataset

get_dataset(batch_size: int)

Method to generate batches of dataset

Example

segmentation_dataset = RELAISegmentationDataset(…) for batch in segmentation_dataset.get_dataset(batch_size):

process_batch(batch)

load_dataset()

Method to load the dataset

class relai.datasets.segmentation_dataset.SegmentationDataPair(input_image_path, input_image, ground_truth_path, ground_truth)

Bases: tuple

ground_truth

Alias for field number 3

ground_truth_path

Alias for field number 2

input_image

Alias for field number 1

input_image_path

Alias for field number 0