Skip to content

Dog breed image recognition with Teachable Machine & Tensorflow.js

License

Notifications You must be signed in to change notification settings

jeferson-sb/dogAI

This branch is 1 commit ahead of, 8 commits behind master.

Folders and files

NameName
Last commit message
Last commit date
Jan 15, 2025
Jan 15, 2025
Jan 15, 2025
Sep 10, 2021
Jan 15, 2025
Jan 15, 2025
Sep 10, 2021
Mar 23, 2020
Jan 15, 2025
Jan 15, 2025
Jan 15, 2025
Jan 15, 2025
Jan 15, 2025
Jan 21, 2025
Jan 21, 2025
Jan 15, 2025

Repository files navigation

🐶 Dog AI

Netlify Status

demo.mp4

💡 Project

Recognize over 100 dog breeds by drag and drop an image using Tensorflow.js and Teachable Machine.

🛠 Tools

💻Demo

https://dogai.netlify.com

🚀Quick start

Installation

$ git clone git@github.com:jeferson-sb/dogAI.git && cd dogAI
$ npm install

Usage

$ npm run dev

Tests

$ npm run test

Linting

$ npm run lint
$ npm run format

How to Train your own model

  1. Gather a dataset with a bunch of images
  2. Resize and minify all the images
  3. Separate dogs image by breed and rename all the files
  4. Upload to Teachable Machine
  5. Train your model
  6. Export your trained model

Dataset Reference

Primary: Aditya Khosla, Nityananda Jayadevaprakash, Bangpeng Yao and Li Fei-Fei. Novel dataset for Fine-Grained Image Categorization. First Workshop on Fine-Grained Visual Categorization (FGVC), IEEE Conference on Computer Vision and Pattern Recognition (CVPR), 2011.

Secondary: J. Deng, W. Dong, R. Socher, L.-J. Li, K. Li and L. Fei-Fei, ImageNet: A Large-Scale Hierarchical Image Database. IEEE Computer Vision and Pattern Recognition (CVPR), 2009.

📝License

This project is licensed under the MIT License