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.