This repository contains a machine learning model that predicts the breed of a dog based on input images. The model is trained on a dataset of various dog breeds and employs deep learning techniques using TensorFlow and Keras to achieve accurate predictions.
Clone the repository : https://github.com/SAteroid/Dog-breed_prediction-model.git
Install the required dependencies
-Ensure you have the necessary image(s) of dogs for prediction.
-Run the prediction script.
-The model will process the input image and provide the predicted breed of the dog.
The model is trained using a convolutional neural network (CNN) architecture on a dataset consisting of various dog breeds. TensorFlow and Keras are utilized for model construction, training, and evaluation. The training process involves data preprocessing, model configuration, and optimization to achieve high accuracy.
To train the model:
-Prepare your dataset of dog images, ensuring proper organization by breed.
-Run the training script:
The script will train the model using the provided dataset and save the trained model weights for future use.
This project is licensed under the MIT License - see the LICENSE file for details. MIT