Skip to content

This notebook builds an end-to-end multi-class image classifier using TensorFlow 2.0 and TensorFlow Hub.

Notifications You must be signed in to change notification settings

mebestaca/dog-vision-project

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

18 Commits
 
 
 
 
 
 

Repository files navigation

End-to-end Multi-class Dog Breed Classification

This notebook builds an end-to-end multi-class image classifier using TensorFlow 2.0 and TensorFlow Hub.

1. Problem

Identifying the breed of a dog given an image of a dog.

When I'm sitting at the cafe and I take a photo of a dog, I want to know what breed of dog it is.

2. Data

The data the we're using is from Kaggle's dog breed identification competition.

https://www.kaggle.com/competitions/dog-breed-identification/data

3. Evaluation

The evaluation is a file with prediction probablilities for each dog breed of each dog breed of each test image.

https://www.kaggle.com/competitions/dog-breed-identification/overview/evaluation

4. Features

Some information about the data:

  • We're dealing with images (unstructured data) so it's probably best we use deep learning/transfer learning
  • There are 120 breeds of dogs (this means there are 120 different classes)
  • There are around 10,000+ images in the training set (these images have labels)
  • There are around 10,000+ images in the test set (these images have no labels, because we'll want to predict them)

About

This notebook builds an end-to-end multi-class image classifier using TensorFlow 2.0 and TensorFlow Hub.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published