Struggling with PyTorch My folders are organized as such:
imgs
|
|_classA
|_train
|_img1.png
|_img2.png
|_test
|_img20.png
|_img21.png
|
|
|_classB
|_train
|_test
...
And I would like for each class (!) load data from train, train some network, evaluate on test.
RTFM got me so far:
- ImageLoader for loading of my .png
- Custom Dataset, e.g. to keep track which class my image is from
- DataLoader to get batches of my data and hand it over to my model
Now most of the tutorials have the structure
train
|
|
test
and the classes are within the train / test
folders.
Now I am not sure, how to get started, due to train / test
being sub-folders in my structure.