Skip to content

Transfer Learning using MobileNet_V2 and Inception as pre-trained models to classify a large set of images of flowers from TensorFlow Dataset.

Notifications You must be signed in to change notification settings

harveyvn/Transfer-Learning-Flowers-Classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 

Repository files navigation

Transfer Learning

The Transfer Learning technique's idea is to take advantage of these high-performance pre-trained neural networks, developed by many machine learning experts and trained by large-scale datasets, and then use them for our own purposes. In other words, we can apply pre-trained neural networks' knowledge to a dataset it has never seen before.

Perform Transfer Learning with TensorFlow Hub

In the repo, we will use TensorFlow Hub to do Transfer Learning. In transfer learning, we reuse parts of an already trained model and change the final layer, or several layers, of the model, and then retrain those layers on our own dataset.

We will perform transfer learning with the MobileNet_V2 and Inception Model to create a CNN that uses Inception and MobleNet as the pretrained model to classify the images from the Flowers dataset. Then comparing the accuracy we get with Inception v3 to the accuracy of MobileNet v2.

Plot Model Predictions

About

Transfer Learning using MobileNet_V2 and Inception as pre-trained models to classify a large set of images of flowers from TensorFlow Dataset.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published