Skip to content

A Streamlit-based dog breed prediction app powered by a deep learning model. Accurately identifies dog breeds from uploaded images using CNN architecture, trained on multiple dog breed classes with real-time predictions and intuitive interface.

Notifications You must be signed in to change notification settings

rivu-intel45/dogbreed-deepvision

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

4 Commits
Β 
Β 
Β 
Β 
Β 
Β 

Repository files navigation

🐢 Dog Breed Detection

Dog Breed Detection is a Streamlit web application that predicts the breed of a dog from an uploaded image using a deep learning model. It provides fast, accurate breed identification through a simple, user-friendly interface.

πŸ“– Overview

This Streamlit-based application uses computer vision techniques to:

  • Classify uploaded dog images into predefined breeds
  • Preprocess and normalize images for consistent model input
  • Display the predicted breed with confidence scores
  • Enable quick experimentation with a trained CNN model in the browser

πŸš€ Features

  • Upload dog images (JPG/PNG) directly in the browser

  • Automatic resizing and preprocessing

  • Prediction powered by a trained Keras .h5 model

  • Clean, interactive Streamlit UI

  • Ready for local use or cloud deployment (e.g., Streamlit Community Cloud)

  • 🧠 Model

  • Built with TensorFlow / Keras

  • Trained on a multi-class dog breed dataset

  • Saved as dog_breed.h5 and loaded at runtime

  • Uses a CNN (optionally with transfer learning) for robust feature extraction


πŸ”§ Installation

Prerequisites

  • Python 3.8 or higher
  • pip (Python package manager)

Step 1: Clone the Repository

-git clone https://github.com/rivu-intel45/dogbreed-deepvision.git

Step 2: Install Dependencies

-pip install -r requirements.txt


▢️ Running the App

-streamlit run main_app.py

Then open the URL shown in the terminal (usually http://localhost:8501).


πŸ–ΌοΈ How to Use

  1. Start the app with streamlit run main_app.py.
  2. Upload a clear image of a dog.
  3. Wait for the model to process the image.
  4. View the predicted breed and confidence.

πŸ“ˆ Future Improvements

  • Support more breeds and larger datasets
  • Show top-k predictions with probabilities

πŸ’‘ Contributing

Contributions are welcome! Feel free to fork, submit issues, or make pull requests.

About

A Streamlit-based dog breed prediction app powered by a deep learning model. Accurately identifies dog breeds from uploaded images using CNN architecture, trained on multiple dog breed classes with real-time predictions and intuitive interface.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published