Skip to content

Image Classification using a Pytorch-implemented Neural Network. Tested on CIFAR-10.

Notifications You must be signed in to change notification settings

estods3/ImageClassification-withDeepLearning

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

26 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Image Classification using Deep Learning

Created an image classification algorithm using a deep neural network. NN uses the PyTorch Deep Learning Framework. It was trained on the CIFAR-10 database for the following objects: Airplane, Automobile, Deer, and Dog.

Database

CIFAR-10: http://www.cs.toronto.edu/~kriz/cifar.html

Setup

install python3 on a linux machine or VM. Then do:

user@ubuntu:~/ImageClassification-withDeepLearning$ sudo apt install python3-pip

user@ubuntu:~/ImageClassification-withDeepLearning$ sudo pip3 install http://download.pytorch.org/whl/cpu/torch-0.4.1-cp35-cp35m-linux_x86_64.whl

user@ubuntu:~/ImageClassification-withDeepLearning$ sudo pip3 install -r requirements.txt

How to Run

user@ubuntu:~/ImageClassification-withDeepLearning$ python3 src/main.py

References

Learning Multiple Layers of Features from Tiny Images, Alex Krizhevsky, 2009.

About

Image Classification using a Pytorch-implemented Neural Network. Tested on CIFAR-10.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages