Skip to content

lukashalim/Udacity-ImageClassifier

Repository files navigation

Udacity ImageClassfier PyTorch Project

This work is related to the Intro to Machine Learning with Pytorch Nanodegree Program that I'm working on in September 2020. The image classfier project asks us to use PyTorch to predict the species based on flower images.

Part I

Image Classifier Project.ipynb uses a pretrained network and trains the classifier using the training data. Evaluates accuracy on test, and then on validation. Saves the resulting model. Finally, applies the train network to make prediction for individual jpg images. Program is set to work on either the GPU or CPU.

Part 2

This part turns the results from Part I into a command line train.py is the command line application used to train the model. predict.py is the command line application to predict the most likely species based on a jpg image of a flower.

About

No description, website, or topics provided.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published