tfds.visualization.ImageGridVisualizer  |  TensorFlow Datasets (original) (raw)

tfds.visualization.ImageGridVisualizer

Visualizer for supervised image datasets.

Inherits From: Visualizer

Methods

match

View source

match(
    ds_info: dataset_info.DatasetInfo
) -> bool

See base class.

show

View source

show(
    ds: tf.data.Dataset,
    ds_info: dataset_info.DatasetInfo,
    rows: int = 3,
    cols: int = 3,
    plot_scale: float = 3.0,
    image_key: Optional[str] = None,
    label_key: Optional[str] = None
)

Display the dataset.

Args
ds tf.data.Dataset. The tf.data.Dataset object to visualize. Examples should not be batched. Examples will be consumed in order until (rows * cols) are read or the dataset is consumed.
ds_info tfds.core.DatasetInfo object of the dataset to visualize.
rows int, number of rows of the display grid.
cols int, number of columns of the display grid.
plot_scale float, controls the plot size of the images. Keep this value around 3 to get a good plot. High and low values may cause the labels to get overlapped.
image_key string, name of the feature that contains the image. If not set, the system will try to auto-detect it.
label_key string, name of the feature that contains the label. If not set, the system will try to auto-detect it.
Returns
fig The pyplot figure.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2024-04-26 UTC.