https://github.com/18alantom/cat-or-dog
🐈 a simple classifier app 🐕
https://github.com/18alantom/cat-or-dog
flask pytorch pytorch-implementation react
Last synced: 3 months ago
JSON representation
🐈 a simple classifier app 🐕
- Host: GitHub
- URL: https://github.com/18alantom/cat-or-dog
- Owner: 18alantom
- License: gpl-3.0
- Created: 2020-04-25T15:46:13.000Z (about 6 years ago)
- Default Branch: master
- Last Pushed: 2021-03-20T03:36:55.000Z (over 5 years ago)
- Last Synced: 2024-12-31T00:29:26.618Z (over 1 year ago)
- Topics: flask, pytorch, pytorch-implementation, react
- Language: JavaScript
- Homepage:
- Size: 84 MB
- Stars: 1
- Watchers: 3
- Forks: 0
- Open Issues: 2
-
Metadata Files:
- Readme: README.md
- License: LICENSE
Awesome Lists containing this project
README
# cat-or-dog
🐈 a simple classifier app 🐕
## Classifier
Makes use of a [Resnet 50](https://pytorch.org/docs/stable/torchvision/models.html#torchvision.models.resnet50) network pretrained on [ImageNet](http://www.image-net.org/) and then tuned on this [Cat and Dog](https://www.kaggle.com/tongpython/cat-and-dog) dataset.
The model was trained for a [challenge](https://github.com/18alantom/sc_challenges/blob/master/week_2/Week%202%20-%20Challenge%202.ipynb) based on transfer learning.
Framework used is [Pytorch](https://pytorch.org/docs/stable/index.html)
## Frontend
The frontend was created using [React](https://reactjs.org/) using [create-react-app](https://create-react-app.dev/), had to use `eject` to change the build path.
## Backend
The Backend was created using [Flask](https://flask.palletsprojects.com/en/1.1.x/).
___
To **run the app**, clone the repo; make sure [npm](https://www.npmjs.com) and other Javascript requirements along with [Python requirements](https://github.com/18alantom/cat-or-dog/blob/master/flask_server/setup.py) 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.