Skip to content

IvanMatoshchuk/Image-Classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Image-Classification

Udacity Project 2

Table of Contents

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

Installation

It is necesary to install "pytorch" and it is suggested to install "cuda", so that algorithm can be run on the video card. GPU can process linear computation ~100 times faster then CPU, so it will save a lot of time for us. The code should run with no issues using Python versions 3.*.

Project Motivation

It is a basic image classifier for flowers open data set.

File Descriptions

The ~500 Mb of different flowers images were used in training and testing the model. Markdown cells were used to assist in walking through the thought process for individual steps.

Results

The main findings of the code can be found in the jupyter notebook provided in the repository.

Licensing, Authors, Acknowledgements

Must give credit to Kaggle for the data. Feel free to use the code here as you would like!