Awesome Lists with GitHub stars

awesome tflite

Awesome TensorFlow Lite Awesome PRs Welcome Twitter

TensorFlow Lite is a set of tools that help convert and optimize TensorFlow models to run on mobile and edge devices. It’s currently running on more than 4 billion devices! With TensorFlow 2.x, you can train a model with tf.Keras, easily convert a model to .tflite and deploy it; or you can download a pretrained TensorFlow Lite model from the model zoo.

This is an awesome list of TensorFlow Lite models with sample apps, helpful tools and learning resources -

Please submit a PR if you would like to contribute and follow the guidelines here.

## Contents

Past announcements:

Here are some past feature annoucements of TensorFlow Lite:

Models with samples

Here are the TensorFlow Lite models with app / device implementations, and references. Note: pretrained TensorFlow Lite models from MediaPipe are included, which you can implement with or without MediaPipe.

Computer vision

Classification

Task Model App | Reference Source
Classification MobileNetV1 (download) Android GitHub Repo Stars GitHub last commit | iOS GitHub Repo Stars GitHub last commit | Raspberry Pi GitHub Repo Stars GitHub last commit | Overview tensorflow.org
Classification MobileNetV2 Recognize Flowers on Android Codelab | Android GitHub Repo Stars GitHub last commit TensorFlow team
Classification MobileNetV2 Skin Lesion Detection Android GitHub Repo Stars GitHub last commit Community
Classification MobileNetV2 American Sign Language Detection | Colab Notebook | Android GitHub Repo Stars GitHub last commit Community
Classification CNN + Quantisation Aware Training Stone Paper Scissor Detection Colab Notebook | Flutter GitHub Repo Stars GitHub last commit Community
Classification EfficientNet-Lite0 (download GitHub Repo Stars GitHub last commit) Icon Classifier Colab & Android GitHub Repo Stars GitHub last commit | tutorial 1 | tutorial 2 Community

Detection

| Task | Model | App | Reference | Source | | -|-|-|-| | Object detection | Quantized COCO SSD MobileNet v1 (download) | Android GitHub Repo Stars GitHub last commit | iOS GitHub Repo Stars GitHub last commit | Overview | tensorflow.org | | Object detection | YOLO | Flutter | Paper | Community | | Object detection | YOLOv5 | Yolov5 Inference GitHub Repo Stars GitHub last commit | Community | | Object detection | MobileNetV2 SSD (download GitHub Repo Stars GitHub last commit) | Reference GitHub Repo Stars GitHub last commit | MediaPipe | | Object detection | MobileDet (Paper) | Blog post (includes the TFLite conversion process) | MobileDet is from University of Wisconsin-Madison and Google and the blog post is from the Community | | License Plate detection | SSD MobileNet (download) GitHub Repo Stars GitHub last commit | Flutter GitHub Repo Stars GitHub last commit | Community | | Face detection | BlazeFace (download GitHub Repo Stars GitHub last commit) | Paper | MediaPipe | | Face Authentication | FaceNet | Flutter GitHub Repo Stars GitHub last commit | Community | | Hand detection & tracking | Palm detection & hand landmarks (download GitHub Repo Stars GitHub last commit) | Blog post | Model card | Android GitHub Repo Stars GitHub last commit | MediaPipe & Community |

Segmentation

| Task | Model | App | Reference | Source | | -|-|-|-| | Segmentation | DeepLab V3 (download) | Android & iOS GitHub Repo Stars GitHub last commit | Overview | Flutter Image GitHub Repo Stars GitHub last commit | Realtime GitHub Repo Stars GitHub last commit | Paper | tf.org & Community | | Segmentation | Different variants of DeepLab V3 models GitHub Repo Stars GitHub last commit | Models on TF Hub with Colab Notebooks | Community | | Segmentation | DeepLab V3 model | Android GitHub Repo Stars GitHub last commit | Tutorial | Community | | Hair Segmentation | Download GitHub Repo Stars GitHub last commit | Paper | MediaPipe |

Style Transfer

| Task | Model | App | Reference | Source | | -|-|-|-| | Style transfer | Arbitrary image stylization GitHub Repo Stars GitHub last commit | Overview | Android GitHub Repo Stars GitHub last commit | Flutter GitHub Repo Stars GitHub last commit | tf.org & Community | | Style transfer | Better-quality style transfer models in .tflite | Models on TF Hub with Colab Notebooks | Community | | Video Style Transfer | Download:
Dynamic range models) | Android GitHub Repo Stars GitHub last commit | Tutorial | Community | | Segmentation & Style transfer | DeepLabV3 & Style Transfer models GitHub Repo Stars GitHub last commit | Project repo GitHub Repo Stars GitHub last commit | Android GitHub Repo Stars GitHub last commit | Tutorial | Community |

Generative

| Task | Model | App | Reference | Source | | -|-|-|-| | GANs | U-GAT-IT GitHub Repo Stars GitHub last commit (Selfie2Anime) | Project repo GitHub Repo Stars GitHub last commit | Android GitHub Repo Stars GitHub last commit | Tutorial | Community | | GANs | White-box CartoonGAN GitHub Repo Stars GitHub last commit (download) | Project repo GitHub Repo Stars GitHub last commit | Android GitHub Repo Stars GitHub last commit | Tutorial | Community | | GANs - Image Extrapolation | Boundless on TF Hub | Colab Notebook | Original Paper | Community |

Post estimation

| Task | Model | App | Reference | Source | | -|-|-|-| | Pose estimation | Posenet (download) | Android GitHub Repo Stars GitHub last commit | iOS GitHub Repo Stars GitHub last commit | Overview | tensorflow.org | | Pose Classification based Video Game Control | MoveNet Lightning (download GitHub Repo Stars GitHub last commit) | Project Repository GitHub Repo Stars GitHub last commit | Community |

Other

| Task | Model | App | Reference | Source | | -|-|-|-| | Low-light image enhancement | Models on TF Hub | Project repo GitHub Repo Stars GitHub last commit | Original Paper | Flutter GitHub Repo Stars GitHub last commit| | Community | | OCR |Models on TF Hub | Project Repository GitHub Repo Stars GitHub last commit | Community

Text

| Task | Model | Sample apps | Source | | ——————- |———————————————————————————————————————————| ————————————————————————————————————————————————————————————————————————————————- | —————— | | Question & Answer | DistilBERT | Android GitHub Repo Stars GitHub last commit | Hugging Face | | Text Generation | GPT-2 / DistilGPT2 | Android GitHub Repo Stars GitHub last commit | Hugging Face | | Text Classification | Download | Android GitHub Repo Stars GitHub last commit |iOS GitHub Repo Stars GitHub last commit | Flutter GitHub Repo Stars GitHub last commit | tf.org & Community | | Text Detection | CRAFT Text Detector (Paper) |Download GitHub Repo Stars GitHub last commit | Project Repository GitHub Repo Stars GitHub last commit | Blog1-Conversion to TFLite | Blog2-EAST vs CRAFT | Models on TF Hub | Android (Coming Soon) | Community | | Text Detection | EAST Text Detector (Paper) |Models on TF Hub | Conversion and Inference Notebook | Community |

Speech

| Task | Model | App | Reference | Source | | —————— |————————————| ————————————————————————————- | ———— | | Speech Recognition | DeepSpeech | Reference GitHub Repo Stars GitHub last commit | Mozilla | | Speech Recognition | CONFORMER | Inference GitHub Repo Stars GitHub last commit Android GitHub Repo Stars GitHub last commit | Community | | Speech Synthesis | Tacotron-2, FastSpeech2, MB-Melgan | Android GitHub Repo Stars GitHub last commit | TensorSpeech | | Speech Synthesis(TTS) | Tacotron2, FastSpeech2, MelGAN, MB-MelGAN, HiFi-GAN, Parallel WaveGAN | Inference Notebook GitHub Repo Stars GitHub last commit | Project Repository GitHub Repo Stars GitHub last commit | Community |

Recommendation

| Task | Model | App | Reference | Source | | —————— |————————————| ————————————————————————————- | ———— | | On-device Recommendation | Dual-Encoder GitHub Repo Stars GitHub last commit | Android GitHub Repo Stars GitHub last commit | iOS GitHub Repo Stars GitHub last commit | Reference | tf.org & Community |

Game

| Task | Model | App | Reference | Source | | —————— |————————————| ————————————————————————————- | ———— | | Game agent | Reinforcement learning | Flutter GitHub Repo Stars GitHub last commit | Tutorial | Community |

Model zoo

TensorFlow Lite models

These are the TensorFlow Lite models that could be implemented in apps and things:

TensorFlow models

These are TensorFlow models that could be converted to .tflite and then implemented in apps and things:

Ideas and Inspiration

ML Kit examples

ML Kit is a mobile SDK that brings Google’s ML expertise to mobile developers.

Plugins and SDKs

Learning resources

Interested but not sure how to get started? Here are some learning resources that will help you whether you are a beginner or a practitioner in the field for a while.

Blog posts

Books

Videos

Podcasts

MOOCs