fiftyone.utils.tf.TFExampleGenerator

class documentation

class TFExampleGenerator(object): (source)

Known subclasses: fiftyone.utils.tf.TFImageClassificationExampleGenerator, fiftyone.utils.tf.TFObjectDetectionExampleGenerator

Constructor: TFExampleGenerator(force_rgb)

View In Hierarchy

Base class for sample writers that emit tf.train.Example protos.

Parameters
force_rgbwhether to force convert all images to RGB
Method __init__ Undocumented
Method make_tf_example Makes a tf.train.Example for the given data.
Instance Variable force_rgb Undocumented
Method _parse_image_or_path Undocumented
def __init__(self, force_rgb=False): (source)
overridden in fiftyone.utils.tf.TFObjectDetectionExampleGenerator

Undocumented

def make_tf_example(self, image_or_path, label, *args, **kwargs): (source)
overridden in fiftyone.utils.tf.TFImageClassificationExampleGenerator, fiftyone.utils.tf.TFObjectDetectionExampleGenerator

Makes a tf.train.Example for the given data.

Parameters
image_or_pathan image or the path to the image on disk
labela fiftyone.core.labels.Label
*argssubclass-specific positional arguments
**kwargssubclass-specific keyword arguments
Returns
a tf.train.Example
force_rgb: False = (source)
overridden in fiftyone.utils.tf.TFImageClassificationExampleGenerator, fiftyone.utils.tf.TFObjectDetectionExampleGenerator

Undocumented

def _parse_image_or_path(self, image_or_path, filename=None): (source)

Undocumented