Search code examples
pythontensorflowdeep-learningtensorboardtensorflow-lite

Cannot convert model file to .tflite


I have cloned this Image Completion repository and it works on my windows machine in python. The models for face-completion and general-completion are hosted. However, the files don't have a .pb extension. How can I get the name of input and output tensors? I have tried renaming the files with .pb extension and tried loading in tensorboard using this article but it gives the error shown in the screenshot.

I have also tried running this in python using this answer but I get no nodes in output.

How can I get the details like input and output tensors to convert this model into a .tflite model for android? Screenshots:
import model error
tensorboard error


Solution

  • The files provided in the repository are checkpoints, and not .pb files that can be converted to .tflite. This can be seen from the saver.restore() call made in demo.py.

    To get a pb file, you will have to build a SavedModel. The resulting folder will contain a .py file, which you can use for conversion (assuming it contains TFLite-supported ops).