8.1.1.9.1.2.1.3. blueoil.utils.tfds_builders.segmentation

8.1.1.9.1.2.1.3.1. Module Contents

8.1.1.9.1.2.1.3.1.1. Classes

SegmentationBuilder

A custom TFDS builder for segmentation dataset.

class blueoil.utils.tfds_builders.segmentation.SegmentationBuilder(dataset_name, dataset_class=None, dataset_kwargs=None, **kwargs)

Bases: tensorflow_datasets.core.GeneratorBasedBuilder

A custom TFDS builder for segmentation dataset. This class loads data from existing dataset classes and generate TFDS formatted dataset which is equivalent to the original one. See also: https://www.tensorflow.org/datasets/add_dataset

VERSION

_info(self)

_split_generators(self, dl_manager)

_generate_examples(self, dataset)

_num_shards(self, dataset)

Decide a number of shards so as not the size of each shard exceeds 256MiB