Skip to content

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

0 stars

Watchers

1 watching

Forks

Releases

Packages

Contributors

Languages