Skip to content

An Image Classifier web app built with Fastai and Pytorch using Transfer Learning. These notebooks contain the code from the medium blog articles from the series: A Fast Introduction to Fastai-My Experience.

Notifications You must be signed in to change notification settings

yashprakash13/RockPaperScissorsFastAI

Repository files navigation

Rock Paper Scissors Image Classification with FastAI

An end-to-end deep learning image classifier app built with Fastai and Pytorch.

The notebooks contain the code from the articles from the series: A Fast Introduction to Fastai-My Experience.


Table of Contents


Medium Article Series

This project began with a viral article about my first experience with Fastai. Then it evolved. Troughout these tutorials, I write about:


Notebooks and code

The ipynb files are as follows:

  1. RPC.ipynb: Simple image classification with fastai library, an introduction
  2. RPC-Pixel-sim.ipynb: Image classification with Pixel Similarity Approach
  3. RPC SGD.ipynb: Image classification with Stochastic Gradient Descent and Cross Entropy Loss function
  4. RPC-Learning Rate, Progressive Resizing Approach.ipynb : Image Classification, state of the art training with learning rate finder, progressing resizing approach.
  5. Text Sentiment Classification(Covid-19 Tweets).ipynb : Text Classification in just 20 lines of code

The Folder RockPaperScissorsClassification contains:

  1. An interactive web app made with streamlit with the ability to run image classification on any uploaded image.
  2. Contanerising the streamlit app with Docker.

The streamlit web app

The web app built through streamlit is now live. The code is located in RockPaperScissorsClassification folder.


Screenshot of the app in action

Steps to build and run the Streamlit web app

  1. Install Docker on your machine.
  2. In the root directory of the terminal, run to build the image:
docker build --tag rps:1.0 .
  1. Lastly, run the app with:
docker run --publish 8501:8501 -it rps:1.0

Open your browser and go to http://localhost:8501/ and play with the app!

About

An Image Classifier web app built with Fastai and Pytorch using Transfer Learning. These notebooks contain the code from the medium blog articles from the series: A Fast Introduction to Fastai-My Experience.

Topics

Resources

Stars

Watchers

Forks

Languages