Skip to content

akessela/Image-Classifier

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Image-Classifier

Image classifier built with Pytorch. Runs on either GPU or CPU.

Table of Contents

  1. Installation
  2. Project Motivation
  3. File Descriptions
  4. Licensing, Authors, and Acknowledgements

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