Robb-S / capy2

Capybara Detector - deployment app in Flask

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

capy2 - Capybara Detector (deployment in Flask)

The Capybara Detector is a deployment of an AI-driven image-classification model that uses a Convolutional Neural Net for transfer learning. The model is trained with the help of Python's PyTorch and FastAI libraries, and it recognizes images of capybaras and 11 other animals, with over 99% accuracy on training data.

The deployment uses Flask plus a bit of Javascript to load user-supplied images and sample images, then access the model for predictions. It's deployed at DreamHost, running under Passenger.

Details of the model training process can be found at https://github.com/Robb-S/capybara-train

And here's the deployment if you need to use it yourself: https://capybara.animalcafes.com/

About

Capybara Detector - deployment app in Flask


Languages

Language:HTML 32.7%Language:CSS 30.4%Language:Python 25.6%Language:JavaScript 11.2%