I have been doing an image classification problem where in the objective is to train a predefined neural network model with set of tfrecords and do inference. This all is happening with reasonable accuracy In Colab.
Subsequent to this I converted the saved_model.pb into model.tflite file. I have checked it against the netron app it is seemingly taking correct inputs (which is an image tensor).
After this I called interpreter.invoke().
Following this when I try to decipher the output tensor I should be able to at least render the output the image, but i am having difficulty in doing this.
This is the link of colab notebook Google Colaboratory 2  where I have maintained the code.
I have other colab notebooks where similar code was done with training for upto 7500 iterations, but i am stuck in all the case at the interpreter level, since i have to port this app on to Android platform