Image-Classifier
Image classifier built with Pytorch. Runs on either GPU or CPU.
Table of Contents
Installation
This project uses the following software and python libraries
You will also need to have software installed to run and execute a Jupyter Notebook
Project Motivation
Build a command line application that classifies images.
File Descriptions
- 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
Licensing, Authors, Acknowledgements
- Udacity for creating a beautiful learning experince.
- MIT license