It goes like this:
/root/train/[class names]/[images]
/root/test/[only one class]/[testing images]
Since my test data is not labeled, I’m only reading the train folder into ImageFolder and trying to split that into a training and validation set