This is example of transfer learning. Deployed using Flask frontend is designed using css, javascript . vgg19(pretrained on imagenet) to find wether image contains dog or not and resnet152 to find the breed of dog. dataset of 133 dog breeds is used to train the model
- models folder contains resent and vgg model weight files instead of read.txt
Download or clone the repository by :
git clone https://github.com/spctr01/dog_breed.git
move into folder:
cd dog_breed
Install the requirements:
pip install -r requirements.txt
Download the model weight files and paste the files (vgg.pth & model.pt) to Models Folder
Vgg19
(change name to vgg.pth)
run the commands(running flask app):
export FLASK_APP=app.py
flask run