• Stars
    star
    169
  • Rank 224,453 (Top 5 %)
  • Language
    Python
  • Created about 7 years ago
  • Updated over 5 years ago

Reviews

There are no reviews yet. Be the first to send feedback to the community and the maintainers!

Repository Details

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.