alxcnwy / flask_fastai_CNN

Flask app wrapper for convolutional neural network (CNN) model covered in lesson 2 of the fast.ai deep learning MOOC

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

### Flask app wrapper for convolutional neural network (CNN) model covered in lesson 2 of the fast.ai deep learning MOOC 

Allows the user to upload an image then that image is fed through a precomputed CNN (the weights are loaded from static/_model/modelweights.h5 ) and the predictions are returned on the next page. Works for 2 classes but easy to extend to arbitrary number of classes given a pre-trained model. 

To run just do:
python server.py 

Enjoy!

About

Flask app wrapper for convolutional neural network (CNN) model covered in lesson 2 of the fast.ai deep learning MOOC

License:MIT License


Languages

Language:Python 72.4%Language:HTML 12.7%Language:CSS 7.7%Language:JavaScript 7.2%