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

Question

我已经克隆了这个 Image Completion repository and it works on my windows machine in python. The models for face-completion and general-completion 被托管。但是，这些文件没有 .pb 扩展名。如何获得输入和输出张量的名称？我尝试使用 .pb 扩展名重命名文件，并尝试使用 this article 在 tensorboard 中加载，但它给出了屏幕截图中显示的错误。

我也在 python 中使用 答案尝试了 运行 这个，但我在输出中没有得到任何节点。

如何获取输入和输出张量等详细信息，以将此模型转换为 android 的 .tflite 模型？截图：
import model error
tensorboard error

Answer 1

存储库中提供的文件是 checkpoints, and not .pb files that can be converted to .tflite. This can be seen from the saver.restore() 在 demo.py 中进行的调用。

要获取 pb 文件，您必须构建一个 SavedModel. The resulting folder will contain a .py file，您可以将其用于转换（假设它包含 TFLite 支持的操作）。