Hi @Nehhar,
you could choose what is more convenient for your
a) Create a folder “train” another folder “valid”, then add one folder with the class name for each class under each of them. Then you could use ImageDataLoaders.from_folder
e.g.
[Path(‘/home/yizhang/.fastai/data/mnist_tiny/valid/7/9413.png’),
Path(‘/home/yizhang/.fastai/data/mnist_tiny/train/7/9263.png’),
Path(‘/home/yizhang/.fastai/data/mnist_tiny/valid/7/7565.png’)]
b) Create a csv with the columns “filename”, “class”, and “is_valid” and use ImageDataLoaders.from_csv
c) … Have a look at the other ImageDataLoaders.