cancel
Showing results for 
Search instead for 
Did you mean: 

Tensorflow to STM32 AI

TReed
Associate

I am looking for a way to convert from Tensorflow trained models to STM32 AI, which the release blog entry makes me believe is possible. However, when looking at the tool, this doesn't appear to be an option. Is Keras the only Tensorflow related tool it is possible to convert from? I am working with a system trained directly in TF and the conversion of all of the models would be a significant amount of work.​

1 REPLY 1
ACapo
Associate II

The tool supports only: Keras, Caffe, Lasagne.

As you know there is not a straight path for the conversion of tf models to keras. But there are some solutions:

https://stackoverflow.com/questions/44466066/how-can-i-convert-a-trained-tensorflow-model-to-keras

But you have to be aware that not all the layers are supported by STMCube.Ai for Keras. Take a look to the User Manual to get the full list of supported layers (Page 44. Section 12.1)

Best,

Alessandro