Flask app wrapper for convolutional neural network (CNN) model covered in lesson 2 of the fast.ai deep learning MOOC
License: MIT License
Python 72.42%JavaScript 7.16%CSS 7.74%HTML 12.68%
flask_fastai_cnn's Introduction
### 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!