dataeval.utils.data.to_image_classification_dataset

dataeval.utils.data.to_image_classification_dataset(images, labels, metadata, classes, name=None)

Helper function to create custom ImageClassificationDataset classes.

Parameters:
images : Array | Sequence[Array]

The images to use in the dataset.

labels : Array | Sequence[int]

The labels to use in the dataset.

metadata : Sequence[dict[str, Any]] | None

The metadata to use in the dataset.

classes : Sequence[str] | None

The classes to use in the dataset.

name : str | None

Return type:

ImageClassificationDataset