Skip to content

tlrx19/HowToLoadPretrainedModels

Repository files navigation

Let's Keep IT Simple And Short

Using pre-trained model for Deep Learning and Machine Learning projects and application is known as Transfer Learning.

The models that have been introduced below has been developed by Machine Learning And Deep Learning Experts for many years and have been performed as the state-of-the-art

HowToLoadPretrainedModels

All the files that have been used here have been taken (and modified) from a free course on Deep Learning offered by Udacity (You can check it here - https://www.udacity.com/course/intro-to-tensorflow-for-deep-learning--ud187)

In this, You are going to learn how to load and use the state-of-the-art pre-trained models for your deep learning and machine learning projects or applications. The models that you can use for your DL/ML projects or applications have been loaded from Tensorflow Pre-Trained Model hub. You can view them here - https://tfhub.dev/ The models that are included on Tensorflow Hub are :

Refer to the given link for more details - https://tfhub.dev/s?module-type=image-augmentation,image-classification,image-feature-vector,image-generator,image-object-detection,image-others,image-style-transfer,image-rnn-agent

I. For Image Classification -

  1. EfficientNet
  2. MobileNet
  3. ImageNet
  4. ResNet
  5. Faster_RCNn You can view all the models here - https://tfhub.dev/s?module-type=image-augmentation,image-classification,image-feature-vector,image-generator,image-object-detection,image-others,image-style-transfer,image-rnn-agent

II. Text Classification -

  1. NNLM
  2. BERT
  3. ELMO
  4. LABSE You can view all the models here : https://tfhub.dev/s?module-type=text-embedding

Ist Colab - Transfer Learning And Tensorflow Hub

In this colab,

  1. You will be introduced to Tensorflow Hub.
  2. You are going to use a state-of-the-art pre-trained model "MobileNet" for Flower-Set Dataset Classification task.
  3. This will a simple introduction to the Transfer Learning.

IInd Colab - Loading Pre-Trained Model for Cats/Dogs Classification -

In this Colab,

  1. You will find yourself diving in the power of MobileNet
  2. You are going to use the model for the infamous Cats/Dogs Dataset Classification task.

IIIrd Colab - BeansDatasetClassification

In this Colab,

  1. You will work with Beans dataset taken from the tensorflow hub

Feel Free To Modify And Change Pre-Trained Model

Feel Free to Modify the repository.

Releases

No releases published

Packages

No packages published