View source on GitHub
|
Visualizer for supervised image datasets.
Inherits From: Visualizer
Methods
match
match(
ds_info: dataset_info.DatasetInfo
) -> bool
See base class.
show
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. |