my dataset has forty thousands images, but I do not know how to put my data into the neural network.I have four classes, every class in the same folder, so I have four folders .So how do I label the image?
Write a custom
Dataset (http://pytorch.org/docs/0.2.0/data.html#torch.utils.data.Dataset) and use
DataLoader to load it.
Actually, this may also just work (and is simpler) http://pytorch.org/docs/0.2.0/torchvision/datasets.html#torchvision.datasets.ImageFolder
okay，thanks very much