8.1.1.9.1.2.1.1. blueoil.utils.tfds_builders.classification

8.1.1.9.1.2.1.1.1. Module Contents

8.1.1.9.1.2.1.1.1.1. Classes

ClassificationBuilder

A custom TFDS builder for classification dataset.

class blueoil.utils.tfds_builders.classification.ClassificationBuilder(dataset_name, dataset_class=None, dataset_kwargs=None, **kwargs)

Bases: tensorflow_datasets.core.GeneratorBasedBuilder

A custom TFDS builder for classification 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