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