Skip to main content
TReed
Visitor II
March 11, 2019
Question

Tensorflow to STM32 AI

  • March 11, 2019
  • 1 reply
  • 1050 views

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.​

    This topic has been closed for replies.

    1 reply

    ACapo
    Associate II
    April 2, 2019

    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