Skip to content

Using Transfer Learning and TensorFlow to Classify Different Dog Breeds (Machine Learning and Data Science course)

Notifications You must be signed in to change notification settings

mpbasto/TL-predicts-doggos

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 

Repository files navigation

Using Transfer Learning and TensorFlow 2.7 to Classify Different Dog Breeds 🐕

In this project, I used machine learning and data from Kaggle's Dog Breed Identification Competition to help identify different breeds of dogs. This kind of problem is called multi-class image classification, as there are mutliple different breeds of dog.

The model used is one pretrained in the domain of image classification from TensorFlow Hub. It makes predictions with MobileNet-V2, a convolutional neural network that is 53 layers deep and a very effective feature extractor for object detection and segmentation.

Even though there are many ways of building a model in TensorFlow, bone of the best ways if one is getting started is to use the Keras API's Sequential model.

Convolutional Neural Networks diagram

Convolutional Neural Networks

There are a few articles worth reading, but this one by Sumit Saha was definitely my go-to when I was researching Convolutional Neural Networks: https://towardsdatascience.com/a-comprehensive-guide-to-convolutional-neural-networks-the-eli5-way-3bd2b1164a53.

Sumit is a data scientist and machine learning engineer who can do some technical blogging too.

About

Using Transfer Learning and TensorFlow to Classify Different Dog Breeds (Machine Learning and Data Science course)

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published