Skip to content

Build a image classification python web app with Streamlit and PyTorch 🐱 🐶

Notifications You must be signed in to change notification settings

dehaoterryzhang/Image_Classification_App

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

48 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Image Classification Web Application 🐱 🐶

Medium Post

Check out my Medium post "Create an Image Classification Web App using PyTorch and Streamlit" here.

Table of Content

Overview

This is a simple image classification web application, using both Streamlit and PyTorch.

Motivation

For people who are not experts at Django or Flask, Streamlit can be a good alternative to build custom Python web apps for data science. I chose image classification as the task here because computer vision is one of the most popular areas of AI currently, powered by the advancement of deep learning algorithms.

Procedure

  • Install Streamlit
  • Build UI
  • Build Model
  • Test Results
  • Next Steps

Model

I have chosen the pretrained ResNet101 model to perform classification.

Installation

pip install -r requirements.txt

About

Build a image classification python web app with Streamlit and PyTorch 🐱 🐶

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages