make_image_folder_dataset#

make_image_folder_dataset(data_dir: str, classes: Optional[Iterable[str]] = None) torchvision.datasets.folder.DatasetFolder[source]#

Initializes and returns an ImageFolder.

If classes is specified, ImageFolder’s default class-to-index mapping behavior is overridden to use the indices of classes instead.

Parameters
Return type

torchvision.datasets.folder.DatasetFolder