Module: tfds.features

API defining dataset features (image, text, scalar,...).

See the guide.


class Audio: tfds.features.FeatureConnector for audio.

class BBox: BBox(ymin, xmin, ymax, xmax)

class BBoxFeature: FeatureConnector for a normalized bounding box.

class ClassLabel: FeatureConnector for integer class labels.

class Dataset: A Dataset feature encodes a nested dataset.

class Documentation: Feature documentation such as a textual description of what this feature means.

class Encoding: Encoding type of tfds.features.Tensor.

class FeatureConnector: Abstract base class for feature types.

class FeaturesDict: Composite FeatureConnector; each feature in dict has its own connector.

class Image: FeatureConnector for images.

class LabeledImage: Image with additional label metadata.

class Scalar: A scalar feature of a particular dtype (e.g., np.int64 or np.float).

class Sequence: Composite FeatureConnector for a dict where each value is a list.

class Tensor: FeatureConnector for generic data of arbitrary shape and type.

class TensorInfo: Structure containing info on the tf.Tensor shape/dtype.

class Text: FeatureConnector for text, encoding to integers with a TextEncoder.

class Video: FeatureConnector for videos, encoding frames individually on disk.

Type Aliases