My previous work is to implement some neural network with PyTorch. The datasets for this network is mnist so it’s very convenient . But when I want to use my own datasets, I met some problems. The datasets is organized by folders. For instance, All the pictures in the folder named A1  belong to the same class called A and all the pictures in the folder named A2. How can I do it? Can anyone help me? Thanks a lot!
4.jpg737×556 52.9 KB