There are some examples of using PyTorch for image classification
Each file of this project is an example of image classification, you can learn from level1 to levelN. For more explaination of these codes, please visit CSDN Blog
level1: Using default Dataset class(
torchvision.datasets.ImageFolder) to read Image. This code is modified from PyTorch
level2: Using custom Dataset class(a custom class inherit from base class
torch.utils.data.Dataset) to read Image.