Machine Learning Nanodegree

Deep Learning

Project: Image Classifier

Install

This project requires Python 3.x and the following Python libraries installed:

• NumPy
• Pandas
• matplotlib
• scikit-learn
• Pytorch

You will also need to have software installed to run and execute an iPython Notebook

We recommend students install Anaconda, a pre-packaged Python distribution that contains all of the necessary libraries and software for this project.

Code

Code is provided in the image_classifier.ipynb notebook file.

Run

In a terminal or command window, navigate to the top-level project directory (that contains this README) and run one of the following commands:

predict.py

Basic usage: python train.py flower_data --gpu --hidden_units 512 256 --epochs 1

train.py

Basic usage: python predict.py ./flower_data/test/1/image_06743.jpg ./models_checkpoints/checkpoint.pth