/Dog-Breed-Prediction-Streamlit-App

The App will predict the breed of the Dog. Using the input image you provide.

Primary LanguageJupyter Notebook

Dog-Breed-Prediction-Streamlit-App

This Model can be used to predict different breeds of dogs which can be further used by different NGO's working on saving animals for educational purposes also. The App will predict the breed of the Dog. Using the input image you provide.

IN THIS PROJECT, WE WILL SEE HOW TO USE KERAS AND TENSORFLOW TO BUILD. TRAIN. AND TEST A CONVOLUTIONAL NEURAL NETWORK CAPABLE OF IDENTIFYING THE BREED OF A DOG IN A SUPPLIED IMAGE. THIS IS A SUPERVISED LEARNING PROBLEM, SPECIFICALLY A MULTICLASS CLASSIFICATION PROBLEM.


The Steps to solve the problem

1. LOADING THE DATA FROM KAGGLE.
2. LOAD LABELS CSV FOR LABELS THAT CONTAIN IMAGE ID AND BREED.
3. CHECKING THE BREED COUNT.
4. ONE-HOT ENCODING ON LABELS DATA PREDICTION COLUMN.
5. LOAD THE IMAGES, CONVERT THEM TO AN ARRAY, AND NORMALIZE THEM.
6. CHECKED THE SHAPE AND SIZE OF THE X AND Y DATA.
7. BUILDING THE MODEL NETWORK ARCHITECTURE.
8. SPLIT THE DATA AND FIT IT INTO THE MODEL AND CREATE AN ACCURACY PLOT.
9. EVALUATE THE MODEL FOR ACCURACY SCORE.
10. USING THE MODEL FOR PREDICTION.



We will start by downloading the dataset creating the model and finding out the predictions using this model. We can optimize different hyperparameters in order to tune this model for higher accuracy. This model can be used to predict different breeds of dogs which can be further used by different NGOs working on saving animals and for educational purposes also.



Screenshots

Input Screen



Output Screen



Give It a Star if you liked the project

Thanks