Image classifier built with Pytorch. Runs on either GPU or CPU.
This project uses the following software and python libraries
You will also need to have software installed to run and execute a Jupyter Notebook
Build a command line application that classifies images.
- Please refer to the notebook for the data source, detailed description of the project.
- train.py successfully trains a new network on a dataset of images
- The predict.py script successfully reads in an image and a checkpoint then prints the most likely image class and it's associated probability
- Udacity for creating a beautiful learning experince.
- MIT license