Giter VIP home page Giter VIP logo

dog-breed-identification's Introduction

Dog Breed Identifier using tensorflow image classification.

Dataset used: Stanford Dogs Dataset

I've also used other images for extra dog breeds. I will create a link to a google drive featuring the latest built models.

Technologies

FARM Stack

  • FastAPI
  • React
  • MongoDB

+ Other

  • Tensorflow 2.15
  • TypeScript
  • Vite.js

Installation

Docker Compose

docker compose up

This is the only command needed to run the application locally, apart from the model itself.

Local

If you don't have docker installed or want to use docker, you can run these commands instead. Before running the backend though, you will need to change each of the imports from from backend import... to import ....

cd backend

pip install -r ../requirements.txt

uvicorn main:app --reload

If you wish to use your GPU to build the model yourself, please follow this guide.

cd frontend

npm install

npm run dev

Other

I haven't uploaded the model itself since there seems to be a bug with git lfs, when downloaded the model is corrupted. In backend, create a folder named model and insert the model from this google drive download link.

dog-breed-identification's People

Contributors

dominikkawka avatar dependabot[bot] avatar 20093701wit avatar

Watchers

 avatar

dog-breed-identification's Issues

Update README

Update README to include information about docker... update it when docker compose gets implemented? #34

Adding users

The values in predBreed are

  • predictedBreed Which is what the model thinks the breed is based off the image
  • Confidence How confident the model is in its guess.
  • actualBreed The user may submit the actual breed, or if the guess was correct.
  • Image The image path.

NEW PROPOSED DATABASE ARCHITECTURE
user - username, password,
prediction- predictedBreed, confidence, actualBreed, image, username

What functionality can this introduce?
If a user wants to see a history of their previous images submitted, then can.

Migrating from tf 2.8 -> 2.14

Starting in tf 2.11, GPU support for Windows stopped.
tf 2.9-2.10 had issues with data augmentation, resulting in much longer model building times. LINK

Reason I am using WindowsOS right now due to my hard-drive with Ubuntu breaking.

Log out button not working

(#29) Username displays on navbar, however there is a bug when logging out, as the log out button doesn’t work. I decided to push this up anyway since I plan on doing a front-end design overhaul.

Webcam functionality

The webcam creates a new image each frame and the latest image gets analysed. Add a "take picture" feature so that the image only gets analysed when a button is pressed

Recommend Projects

  • React photo React

    A declarative, efficient, and flexible JavaScript library for building user interfaces.

  • Vue.js photo Vue.js

    🖖 Vue.js is a progressive, incrementally-adoptable JavaScript framework for building UI on the web.

  • Typescript photo Typescript

    TypeScript is a superset of JavaScript that compiles to clean JavaScript output.

  • TensorFlow photo TensorFlow

    An Open Source Machine Learning Framework for Everyone

  • Django photo Django

    The Web framework for perfectionists with deadlines.

  • D3 photo D3

    Bring data to life with SVG, Canvas and HTML. 📊📈🎉

Recommend Topics

  • javascript

    JavaScript (JS) is a lightweight interpreted programming language with first-class functions.

  • web

    Some thing interesting about web. New door for the world.

  • server

    A server is a program made to process requests and deliver data to clients.

  • Machine learning

    Machine learning is a way of modeling and interpreting data that allows a piece of software to respond intelligently.

  • Game

    Some thing interesting about game, make everyone happy.

Recommend Org

  • Facebook photo Facebook

    We are working to build community through open source technology. NB: members must have two-factor auth.

  • Microsoft photo Microsoft

    Open source projects and samples from Microsoft.

  • Google photo Google

    Google ❤️ Open Source for everyone.

  • D3 photo D3

    Data-Driven Documents codes.