dehaoterryzhang / Image_Classification_App

Build a image classification python web app with Streamlit and PyTorch :cat: :dog:

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
• Motivation
• Procedure
• Model
• Installation

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