Skip to content

chenmozxh/pytorch_studying

Repository files navigation

Image Classification example1

  1. download images from https://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html

    wget https://download.pytorch.org/tutorial/hymenoptera_data.zip

    unzip hymenoptera_data.zip

    The diretory tree is:

    hymenoptera_data

    --train

     ---ants
    
     ---bees
    

    --val

     ---ants
    
     ---bees
    
  2. python3 example1.py

Image Classification example2

  1. copy all images to data

  2. python3 example2.py

Dependencies:

python3.6 torch0.4.1

Reference:

  1. pytorch tutorial: https://pytorch.org/tutorials/beginner/transfer_learning_tutorial.html
  2. https://github.com/miraclewkf/ImageClassification-PyTorch

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages