How can I define Rescaling layer on the .ino file?

TensorFlow Special Interest Groups Micro
, ,
1

Hi,

I use a Rescaling layer as the first layer of my model. I’m new to TensorFlow lite micro. Do I need to define this Rescaling layer on the .ino file of the project? Does this layer require memory? On the other hand, this file ( https://github.com/tensorflow/tflite-micro/blob/main/tensorflow/lite/micro/all_ops_resolver.cc) does not contain something like AddRescaling().

Thank you so much!

2

Per my understanding since rescaling is not part of the supported layer, you will have to build it as a custom layer. This can be done during the model conversion process to tensorflow lite format. Follow this resource on the tensorflow site on how to do that.

1 Like
4

Thank you so much! I’m going to try it.