Stay organized with collections Save and categorize content based on your preferences.

Visualizer for supervised image datasets.

Inherits From: Visualizer



View source

See base class.


View source

Display the dataset.

ds The 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.

fig The pyplot figure.