Machine Learning on the Edge Using TensorFlow Lite


What if you could perform machine learning on the edge, i.e on your mobile device? This would mean that you no longer would need the roundtrip to the server, no data will leave the device and you don't even need an internet connection. In this session you will get an introduction to TensorFlow Lite so that you can use it in your own projects.

8 min
02 Jul, 2021


Sign in or register to post your comment.

AI Generated Video Summary

Håkan Silvernagel introduces TensorFlow Lite, an open-source deep learning framework for deploying machine learning models on mobile and IoT devices. He highlights the benefits of using TensorFlow Lite, such as reduced latency, increased privacy, and improved connectivity. The Talk includes a demonstration of object recognition capabilities and a real-world example of using TensorFlow Lite to detect a disease affecting farmers in Tanzania. References to official TensorFlow documentation, Google IO conference, and TensorFlow courses on Coursera are provided.

1. Introduction to TensorFlow Lite

Short description:

In this part, Håkan Silvernagel introduces TensorFlow Lite, an open-source deep learning framework for deploying machine learning models on mobile and IoT devices. He explains the two main components of TensorFlow Lite, the interpreter, and the converter, and highlights the benefits of using TensorFlow Lite, such as reduced latency, increased privacy, and improved connectivity. Håkan also mentions the availability of multiple APIs and the ability to use pre-trained models for common machine learning tasks. He concludes by mentioning the use of a pre-trained model for object detection on a Raspberry Pi inside a TJ bot.

Good morning, good afternoon, and good evening. My name is Håkan Silvernagel, I'm going to be talking about how you can do machine learning on the edge using TensorFlow Lite. So I'm working as a manager for AI and big data for a Norwegian consultancy company called Myles and I'm also a Microsoft MVP. But the topic here for today is TensorFlow Lite.

So what is TensorFlow Lite? Well, it's an open source deep learning framework so that you can deploy machine learning models on both your mobile and IOT devices. So it can be on, for example, a Raspberry Pi that I will show you a little bit later on, but it can also be on a tiny microcontroller. So the way that this works is TensorFlow Lite has two main components. There's one TensorFlow Lite interpreter. So what it does, it runs optimized TensorFlow models on different types of hardware. And then we have the TensorFlow Lite converter. So the converter converts regular TensorFlow models into much more efficient form, which has been optimized both to improve the binary size and the performance.

So you might think to yourself, why should I use TensorFlow Lite? So there are a number of arguments here why we would prefer to use TensorFlow Lite. So one of them has to do with latency. So since everything that we do will take place on the device, there is no need for a round trip between the client and the server. And also in terms of privacy, the user data will be local. So that means that you have an increased security. And also in terms of connectivity, you don't require an internet connection. And in addition, a lot of these small devices, they don't have enough power in order to power up a radio transmitter. And there are also multiple API's here. So you can use either Java, Swift, Objective-C, C++, or Python. So there are lots of potentials. And the final thing, which is maybe the most important here, is that you can use pre-trained models for common machine learning tasks. The way that it works is that you pick a model, either you pick your own model, or you can pick one of these pre-trained models. And then you can convert your model into the TensorFlow Lite format. And then after that, you deploy it to your device. And then you can also optimize some optimization of the model. On this slide, we can see some of the pre-trained models that you can use. So we will be using a model for object detection in a quick while here. When we run object detection, we will run it on a Raspberry Pi, which is inside a TJ bot. And then it will be able to recognize different types of objects.

2. Using a Preoptimized Mobile Net SSD for Inference

Short description:

We will be using a preoptimized mobile net SSD trained on the CocoaDat dataset. The four main steps for inference are initializing the interpreter, preparing the image input, performing the inference, and obtaining and mapping the results. The process involves loading the interpreter, allocating tensors, preparing the image input from the video camera, performing inference, and obtaining the bounding box coordinates, class index, and confidence level of the detected objects. A quick demo on the Raspberry Pi device shows the recognition of a person but faces some issues with object recognition.

We will be using a preoptimized mobile net SSD trained on the CocoaDat dataset. So the Cocoa is trained on 80 object categories. And we will have a label file that will map the output of the models so that we know what it actually is in text. And this is an example from the GitHub link that you can see on the slide. And you can also download the TF Lite model file and also the label file from the URL below.

So when we do inference, there are four main steps that we need to do. One of them is to initialise the interpreter. The second is to prepare the image input. And the third is to perform the inference. And the last step is to actually obtain and map the results.

So we load the interpreter from the TF Lite runtime. And then we allocate the tensors, the input and the output tensor. And then in our second step, we need to prepare the image input. So now we will take image from the video camera. So we define a class, video stream class. And then what we want to do is we want to read from the stream, from the camera stream. So we start up an infinite loop here, where we just take the latest camera frame from the stream, and then we return it to the calling application. And then in our application, we define a video stream object. And then we just acquire the frame and then we do some resizing. the right size for TensorFlow. And then we perform the inference. So we point to data that will be used for this test, and then we run the interpreter by running invoke. And finally, we just obtain and map the results.

So in this case, we get the bounding box with the coordinates of the objects, and also get the class index of the detected objects and the confidence level of the objects. So let me switch over here for a quick demo. So now we are on the Raspberry Pi device. And then I can hold up an object here for the Raspberry Pi. And then we can see that it recognizes the person, which is me. It has some problems here with recognizing my object. Let me try another object.

3. Object Recognition and Real-World Example

Short description:

In this part, Håkan Silvernagel demonstrates the object recognition capabilities of TensorFlow Lite. He showcases how the model can accurately identify objects, such as a banana and a remote control, displaying the confidence level and bounding box. Håkan also shares a real-world example of using TensorFlow Lite to detect a disease affecting farmers in Tanzania. He concludes by providing references to the official TensorFlow documentation, Google IO conference, and TensorFlow courses on Coursera, and expresses gratitude to the audience.

As you can see, for example, this it recognizes it's a banana. And I can take another object here. This it recognizes that it's a remote control. You can also see the confidence level and the bounding box.

So, I also have an example here from the real world. How you can use TensorFlow in a real-world problem. So, in Tanzania, the farmers have a big problem because there is a disease. And they are using a TensorFlow Lite model in order to do run TensorFlow on the mobile and be able to detect the disease.

So, with that, I think I will just quickly switch over here to some references. So, here you can see the official TensorFlow documentation, and you also have some links to the Google IO conference. And some courses about TensorFlow on the Coursera website. If you have any questions or comments about this, you can always reach me on Twitter or reach me on an email. So, with that said, I want to thank you for tuning in, and I wish you a great conference.

Check out more articles and videos

We constantly think of articles and videos that might spark Git people interest / skill us up or help building a stellar career

ML conf EU 2020ML conf EU 2020
41 min
TensorFlow.js 101: ML in the Browser and Beyond
Discover how to embrace machine learning in JavaScript using TensorFlow.js in the browser and beyond in this speedy talk. Get inspired through a whole bunch of creative prototypes that push the boundaries of what is possible in the modern web browser (things have come a long way) and then take your own first steps with machine learning in minutes. By the end of the talk everyone will understand how to recognize an object of their choice which could then be used in any creative way you can imagine. Familiarity with JavaScript is assumed, but no background in machine learning is required. Come take your first steps with TensorFlow.js!
6 min
Charlie Gerard's Career Advice: Be intentional about how you spend your time and effort
When it comes to career, Charlie has one trick: to focus. But that doesn’t mean that you shouldn’t try different things — currently a senior front-end developer at 
, she is also a sought-after speaker, mentor, and a machine learning trailblazer of the JavaScript universe. "Experiment with things, but build expertise in a specific area," she advises.
What led you to software engineering?
My background is in digital marketing, so I started my career as a project manager in advertising agencies. After a couple of years of doing that, I realized that I wasn't learning and growing as much as I wanted to. I was interested in learning more about building websites, so I quit my job and signed up for an intensive coding boot camp called General Assembly. I absolutely loved it and started my career in tech from there.

What is the most impactful thing you ever did to boost your career?
I think it might be
public speaking
. Going on stage to share knowledge about things I learned while building my side projects gave me the opportunity to meet a lot of people in the industry, learn a ton from watching other people's talks and, for lack of better words, build a personal brand.

What would be your three tips for engineers to level up their career?
Practice your communication skills. I can't stress enough how important it is to be able to explain things in a way anyone can understand, but also communicate in a way that's inclusive and creates an environment where team members feel safe and welcome to contribute ideas, ask questions, and give feedback. 
In addition, build some expertise in a specific area. I'm a huge fan of learning and experimenting with lots of technologies but as you grow in your career, there comes a time where you need to pick an area to focus on to build more profound knowledge. This could be in a specific language like JavaScript or Python or in a practice like accessibility or web performance. It doesn't mean you shouldn't keep in touch with anything else that's going on in the industry, but it means that you focus on an area you want to have more expertise in. If you could be the "go-to" person for something, what would you want it to be? 

And lastly, be intentional about how you spend your time and effort. Saying yes to everything isn't always helpful if it doesn't serve your goals. No matter the job, there are always projects and tasks that will help you reach your goals and some that won't. If you can, try to focus on the tasks that will grow the skills you want to grow or help you get the next job you'd like to have.

What are you working on right now?
Recently I've taken a pretty big break from side projects, but the next one I'd like to work on is a prototype of a tool that would allow hands-free coding using gaze detection. 

Do you have some rituals that keep you focused and goal-oriented?
Usually, when I come up with a side project idea I'm really excited about, that excitement is enough to keep me motivated. That's why I tend to avoid spending time on things I'm not genuinely interested in. Otherwise, breaking down projects into smaller chunks allows me to fit them better in my schedule. I make sure to take enough breaks, so I maintain a certain level of energy and motivation to finish what I have in mind.

You wrote a book called
Practical Machine Learning in JavaScript.
What got you so excited about the connection between JavaScript and ML?
The release of TensorFlow.js opened up the world of ML to frontend devs, and this is what really got me excited. I had machine learning on my list of things I wanted to learn for a few years, but I didn't start looking into it before because I knew I'd have to learn another language as well, like Python, for example. As soon as I realized it was now available in JS, that removed a big barrier and made it a lot more approachable. Considering that you can use JavaScript to build lots of different applications, including augmented reality, virtual reality, and IoT, and combine them with machine learning as well as some fun web APIs felt super exciting to me.

Where do you see the fields going together in the future, near or far? 
I'd love to see more AI-powered web applications in the future, especially as machine learning models get smaller and more performant. However, it seems like the adoption of ML in JS is still rather low. Considering the amount of content we post online, there could be great opportunities to build tools that assist you in writing blog posts or that can automatically edit podcasts and videos. There are lots of tasks we do that feel cumbersome that could be made a bit easier with the help of machine learning.

You are a frequent conference speaker. You have your own blog and even a newsletter. What made you start with content creation?
I realized that I love learning new things because I love teaching. I think that if I kept what I know to myself, it would be pretty boring. If I'm excited about something, I want to share the knowledge I gained, and I'd like other people to feel the same excitement I feel. That's definitely what motivated me to start creating content.

How has content affected your career?
I don't track any metrics on my blog or likes and follows on Twitter, so I don't know what created different opportunities. Creating content to share something you built improves the chances of people stumbling upon it and learning more about you and what you like to do, but this is not something that's guaranteed. I think over time, I accumulated enough projects, blog posts, and conference talks that some conferences now invite me, so I don't always apply anymore. I sometimes get invited on podcasts and asked if I want to create video content and things like that. 
Having a backlog of content helps people better understand who you are and quickly decide if you're the right person for an opportunity.
What pieces of your work are you most proud of?
It is probably that I've managed to develop a mindset where I set myself hard challenges on my side project, and I'm not scared to fail and push the boundaries of what I think is possible. I don't prefer a particular project, it's more around the creative thinking I've developed over the years that I believe has become a big strength of mine.
Follow Charlie on Twitter

React Advanced Conference 2021React Advanced Conference 2021
21 min
Using MediaPipe to Create Cross Platform Machine Learning Applications with React
This talk gives an introduction about MediaPipe which is an open source Machine Learning Solutions that allows running machine learning models on low-powered devices and helps integrate the models with mobile applications. It gives these creative professionals a lot of dynamic tools and utilizes Machine learning in a really easy way to create powerful and intuitive applications without having much / no knowledge of machine learning beforehand. So we can see how MediaPipe can be integrated with React. Giving easy access to include machine learning use cases to build web applications with React.
ML conf EU 2020ML conf EU 2020
32 min
An Introduction to Transfer Learning in NLP and HuggingFace
In this talk I'll start introducing the recent breakthroughs in NLP that resulted from the combination of Transfer Learning schemes and Transformer architectures. The second part of the talk will be dedicated to an introduction of the open-source tools released HuggingFace, in particular our Transformers, Tokenizers and Datasets libraries and our models.
TestJS Summit 2021TestJS Summit 2021
18 min
Predictive Testing in JavaScript with Machine Learning
This talk will cover how we can apply machine learning to software testing including in Javascript, to help reduce the number of tests that need to be run.
We can use predictive machine learning model to make an informed decision to rule out tests that are extremely unlikely to uncover an issue. This opens up a new, more efficient way of selecting tests.

Workshops on related topic

ML conf EU 2020ML conf EU 2020
160 min
Hands on with TensorFlow.js
Come check out our workshop which will walk you through 3 common journeys when using TensorFlow.js. We will start with demonstrating how to use one of our pre-made models - super easy to use JS classes to get you working with ML fast. We will then look into how to retrain one of these models in minutes using in browser transfer learning via Teachable Machine and how that can be then used on your own custom website, and finally end with a hello world of writing your own model code from scratch to make a simple linear regression to predict fictional house prices based on their square footage.

ML conf EU 2020ML conf EU 2020
113 min
The Hitchhiker's Guide to the Machine Learning Engineering Galaxy
Are you a Software Engineer who got tasked to deploy a machine learning or deep learning model for the first time in your life? Are you wondering what steps to take and how AI-powered software is different from traditional software? Then it is the right workshop to attend.
The internet offers thousands of articles and free of charge courses, showing how it is easy to train and deploy a simple AI model. At the same time in reality it is difficult to integrate a real model into the current infrastructure, debug, test, deploy, and monitor it properly. In this workshop, I will guide you through this process sharing tips, tricks, and favorite open source tools that will make your life much easier. So, at the end of the workshop, you will know where to start your deployment journey, what tools to use, and what questions to ask.

ML conf EU 2020ML conf EU 2020
146 min
Introduction to Machine Learning on the Cloud
This workshop will be both a gentle introduction to Machine Learning, and a practical exercise of using the cloud to train simple and not-so-simple machine learning models. We will start with using Automatic ML to train the model to predict survival on Titanic, and then move to more complex machine learning tasks such as hyperparameter optimization and scheduling series of experiments on the compute cluster. Finally, I will show how Azure Machine Learning can be used to generate artificial paintings using Generative Adversarial Networks, and how to train language question-answering model on COVID papers to answer COVID-related questions.