cat-or-dog

🐈 a simple classifier app 🐕

Classifier

Makes use of a Resnet 50 network pretrained on ImageNet and then tuned on this Cat and Dog dataset. The model was trained for a challenge based on transfer learning. Framework used is Pytorch

Frontend

The frontend was created using React using create-react-app, had to use eject to change the build path.

Backend

The Backend was created using Flask.

---

To run the app, clone the repo; make sure npm and other Javascript requirements along with Python requirements are installed.

$ npm install from the cat-or-dog sub directory, $ npm run build then from the flask_server sub directory set $ export FLASK_APP=flask_server and $ flask run.

---

The app maybe hosted somewhere in the near future because the above procedure is almost cumbersome.