My dataset is in the format
root/ 1/ 1.png 2.png 3.png ... 2/ 1.png 2.png 3.png ... ...
I need to be able to load batches containing images only from a specific class, for each class.
Eg. (1st batch are 8 images from root/1, 2nd batch are 8 images from root/2, so on)
Is there any way to do so without having multiple datasets, each for a particular class?
Any help is appreciated.