我已经克隆了这个Image Completion存储库,它可以在python的Windows计算机上运行。 face-completion和general-completion的模型 托管。但是,文件没有扩展名.pb。如何获得输入和输出张量的名称?我尝试使用.pb扩展名重命名文件,并尝试使用this article将其加载到tensorboard中,但是它给出了屏幕截图中显示的错误。
我也尝试使用this答案在python中运行此命令,但输出中没有节点。
如何获取输入和输出张量之类的详细信息,以将该模型转换为适用于Android的.tflite模型?截图:
import model error
tensorboard error
答案 0 :(得分:0)
存储库中提供的文件是checkpoints,而不是可以转换为.pb
的{{1}}文件。从.tflite
进行的saver.restore()
调用中可以看出。
要获取pb文件,您将必须构建SavedModel。生成的文件夹将是contain a .py
file,可用于转换(假设它包含TFLite支持的操作)。