- Flask
pip install Flask
- Werkzeug
pip install Werkzeug
- PyTorch
pip install torch torchvision
- OpenCV
pip install opencv-python
- Matplotlib
pip install matplotlib
- Pillow
pip install Pillow
Note: The web app also uses the following preinstalled libraries:
- os
- shutil
- atexit
python3 routes.py
Here is a link to a Demo video of the web app
Folder containing cache.
Folder containing OpenCV's Haar cascades based classifier.
Folder required by the Flask web app to store static content like images.
Folder required by the Flask web app to store the HTML files of the webpages.
5. model_transfer.pt(needs to be downloaded from this link and placed in the same directory as routes.py)
Trained PyTorch model(transfer learning).
Python script with code for the web app. The code is mostly the same as the Jupyter Notebook, except for a few Flask functions at the end.
Just a compiled file of routes.py
These instructions were written for a Linux OS.