无法将模型文件转换为.tflite

时间:2019-02-27 13:30:46

标签: python tensorflow deep-learning tensorboard tensorflow-lite

我已经克隆了这个Image Completion存储库,它可以在python的Windows计算机上运行。 face-completiongeneral-completion的模型 托管。但是,文件没有扩展名.pb。如何获得输入和输出张量的名称?我尝试使用.pb扩展名重命名文件,并尝试使用this article将其加载到tensorboard中,但是它给出了屏幕截图中显示的错误。

我也尝试使用this答案在python中运行此命令,但输出中没有节点。

如何获取输入和输出张量之类的详细信息,以将该模型转换为适用于Android的.tflite模型?截图:
import model error
tensorboard error

1 个答案:

答案 0 :(得分:0)

存储库中提供的文件是checkpoints,而不是可以转换为.pb的{​​{1}}文件。从.tflite进行的saver.restore()调用中可以看出。

要获取pb文件,您将必须构建SavedModel。生成的文件夹将是contain a .py file,可用于转换(假设它包含TFLite支持的操作)。