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]] | dict[str, Sequence[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