TensorFlow Lite Model Maker: Create Models for On-Device Machine Learning

#artificialintelligence 

In this blog post, we will learn to create a TensorFlow Lite model using the TF Lite Model Maker Library. We will fine-tune a pre-trained image classification model on the custom dataset and further explore different types of model optimization techniques currently supported by the library and export them to the TF Lite model. Detailed performance comparison of the created TF Lite models and the converted one is done, followed by deploying the model on the web app in the end. The TensorFlow Lite Model Maker Library enables us to train a pre-trained or a custom TensorFlow Lite model on a custom dataset. Similar to the previous blog, we will be using Microsoft's Cats and Dogs Dataset.