Skip to content

bithapa/neural-network-image-classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

28 Commits
 
 
 
 
 
 
 
 

Repository files navigation

Image Classification using Neural Networks (from scratch)

This project was a part of AI Scholarship Course that I won from Udacity & Facebook. It implements neural network from scratch using Python and PyTorch Library to create deep learning models that can predict hand-written digits and classify MNIST fashion images.

Technologies Used: Python, PyTorch, NumPy

Pytorch's torch documentation: https://pytorch.org/docs/stable/torch.html

Torchvision datasets: https://pytorch.org/docs/stable/torchvision/index.html


How to Run The Code:

  1. Open neural_network.ipynb for all the code and results. OR
  2. Clone the repository with git clone https://github.com/tito-coffee/aiproject and open the nn.ipynb file in Jupyter Notebook Editor and run the cells. OR
  3. Run the notebook in Google Colab (Google Sign in required): Open In Google Colab

helper.py is a python file that helps to view images and graphs at the same time


About

Neural networks with Python for image classificatio

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published