ARM-software / ML-examples

Arm Machine Learning tutorials and examples

Home Page:https://developer.arm.com/technologies/machine-learning-on-arm

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

TensorFlow model conversion to CMSIS-NN calls

marcin-ch opened this issue · comments

Hello!

is there any guide available for converting TensorFlow model to CMSIS-NN calls?
Something similar to this guide converting Caffe model to CMSIS-NN calls but for TensorFlow models.

In this guide there are two Python scripts being used:
nn_quantizer.py for quantization to 8-bit weights and activations
code_gen.py for generating the code consisting of NN function calls

I would like to know if such a tools exist.
Or perhaps you can recommend different approach how to manage TensorFlow models if I want to use them in Cortex-M devices?

Thank you in advance!