craigsidcarlson / PytorchFlaskApp

Flask application to support pytorch model prediction

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Welcome to Cat or Cheese

This is the full source code for Cat or Cheese. Cat or Cheese is a flask application capable of running on an AWS elastic beanstalk linux server. The application allows to to drop and image file (png, jpg, or jpeg) and the pre built model will predict if the image is a Cat or Cheese.

Also included is the pytorch convolutional neural net classifier. This code is label agnostic and can be rebuilt to any N number of catagories given train and test data is supplied. Once the model is complete the flask application can use it for prediction.

app - contains the flask application

ml - contains the machine learning code

Train and test images are not included. Recommend using http://www.image-net.org/ for image data.

About

Flask application to support pytorch model prediction


Languages

Language:Python 78.4%Language:JavaScript 11.3%Language:CSS 6.7%Language:HTML 3.7%