Loading Script for Vision Dataset

[mariosasko]

You can use the imagefolder loader to load such a dataset:

from datasets import load_dataset
ds = load_dataset("imagefolder", data_files="path/to/train_dir/**", split="train")