There are some examples of using PyTorch for image classification
Usage
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.