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


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 ( tflite-micro/ at main · tensorflow/tflite-micro · GitHub) does not contain something like AddRescaling().

Thank you so much!

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

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