/image-classification-website-with-flask

This is an image classification website using flask and pytorch pre-trained model.

Primary LanguagePython

A simple image classification website demo based on flask

This is a cnn based classification website with pytorch and flask, click the following image to watch the demo video.

Watch the video

run the following code to set up the image classification website.

  $ python image_classification.py