https://github.com/jiko23/dog_breed-classification
Classifies Dog breed using Resnet50 CNN model
https://github.com/jiko23/dog_breed-classification
flask-api image-classification image-processing keras-neural-networks resnet-50 tensorflow
Last synced: 2 months ago
JSON representation
Classifies Dog breed using Resnet50 CNN model
- Host: GitHub
- URL: https://github.com/jiko23/dog_breed-classification
- Owner: jiko23
- Created: 2021-02-27T01:50:25.000Z (over 5 years ago)
- Default Branch: main
- Last Pushed: 2022-09-03T17:08:42.000Z (almost 4 years ago)
- Last Synced: 2025-01-16T03:49:48.215Z (over 1 year ago)
- Topics: flask-api, image-classification, image-processing, keras-neural-networks, resnet-50, tensorflow
- Language: Jupyter Notebook
- Homepage:
- Size: 240 KB
- Stars: 0
- Watchers: 1
- Forks: 0
- Open Issues: 0
-
Metadata Files:
- Readme: README.md
Awesome Lists containing this project
README
# Dog_Breed-Classification
Classifies Dog breed using Resnet50 CNN model
Steps to run files:
1. Run the Jupiter Notebook file to train the model and save the trained model in disc location. Then performs prediction on few test sample. As, labels are not present for the test dataset thus couldnot perform accuracy checking and other matrices checking. ##NOTE: change epochs as per need.
2. Steps to run API :
## NOTE: Kindly change file locations as per your requirement.
1. Run base_64_.py --> Will produce testfile.txt
2. Run classification_API.py --> API will produce {Score(probability), Breed}. [Create a folder in the same directory named templates and store index.html and display.html in it]