Skip to content

Latest commit

 

History

History
9 lines (5 loc) · 624 Bytes

README.md

File metadata and controls

9 lines (5 loc) · 624 Bytes
Raw

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.