TinyML Book

14 August 2020

I’ve been slowly making my way through the TinyML Book. It’s been a nice introduction to machine learning on resource-constrained devices. Things are constantly evolving though, and I found that using the book and the TensorFlow Lite Micro examples together is the best way to get started.

I also created a GitHub repo which contains my own copies of the examples. Each example is in its own folder. And each folder contains the Jupyter notebooks used for training, as well as the files needed to deploy to an edge device (in my case, the Arduino Nano 33 BLE Sense). The most notable addition is a Dockerfile within each folder to run the notebooks in my own development environment. This has made it easier for me to experiment with and re-train models locally without having to worry about Google Colab sessions timing out.

Here’s a screenshot of the Jupyter notebook used to train the micro_speech model:

Training the micro_speech model
Training the micro_speech model

And here’s the resulting model deployed to a device:

Hope my learnings can help you start on your own journey with TinyML.

Categories: AI, Books, Embedded

comments powered by Disqus