Robotics Stack Exchange is a question and answer site for professional robotic engineers, hobbyists, researchers and students. It only takes a minute to sign up.
Sign up to join this community
I am working on a project that requires training a neural network with humidity and temperature values. After spending a while googling on how to run a neural network on ESP32 (Wemos LOLIN D32 Pro), I came to the conclusion that off-board training and then uploading would be the best option. But I didn't get any resource on doing that.
How can I upload a trained neural network model(TensorFlow, scikit-learn ...) to an ESP32 board?
You can printout your weights and bias value from your model in TensorFlow, PyTorch or other deep learning library. If you want to convert your model to ESP32, you must create your own code to define your feed forward network. It's very simple, like create a matrix multiplication program and also create your own activation function. Then, you take weights and bias value from your trained model and define manually in your code as variable or constant. It is should works.
If you develop and train you model with tensorflow you can probably (only experimental releases, no guarantees) use Tensorflow lite to deploy your neural network models. The idea is to train usig Tensorflow, export the serialized model and import that in TF Lite. You can see the intended workflow for deployin models here.
TF Lite is meant for Android and iOS devices, but it has experimental support for other devices (incl. running bare metal on microcontroller). There is also a tutorial for Raspberry Pis.
An alternative route would be to use Matlab. It has a Statistics and Machine Learning toolbox and a seperate Deep Learning Toolbox. However, this it not enough, you will also need the Matlab Coder toolbox. The Matlab Coder toolbox can generate c/c++ source code for (trained) neural network models that can run without external library dependencies and therefore can be easily ported to any platform. Costs invoving the needed Matlab toolboxes are signifcant comparted to the free open source solutions.
