This project is a easy classfication task. Maybe a good example for pytorch learning. Dataset is downloaded from http://vision.stanford.edu/aditya86/ImageNetDogs/. If you are interested in it. Take a try:
- download the Images and Annotations from the link below. unzip them in ./data
- python prepare.py to deal with the dataset
- python train.py for training
- python test.py for test