Tensoflow Lite：如何将模型输出更改为图像坐标列表？

Question

我可以在移动设备上运行tensorflow lite图像classification example。但是我想将图像分类模型交换为姿势识别模型。就我而言，输出应包含（x，y）坐标列表。

代码中的respective line如下所示：

  @Override
  protected void runInference() {
    tflite.run(imgData, labelProbArray);
  }

但是tflite.run函数没有源代码（仅以二进制形式提供）。所以我不知道它是如何工作的或如何操纵它的返回值。

我之前使用过tensorflow，但是我不知道如何创建与tflite期望的输入和输出兼容的TF模型。

除了official documentation之外，还有谁能帮助我或为我提供更详细的教程？

Answer 1

必须先将其转换为TF模型，然后再将其转换为tflite。可以使用工具“ netron”检查预先存在的tflite模型

使用自训练模型（.ckpt文件）时，必须执行

的步骤

• 创建图形定义文件进行评估
• 使用freeze_graph使用您训练中的最新.ckpt文件冻结先前创建的图形定义文件，以为其分配一些权重
• 使用tflite_convert（例如从命令行）将冻结的图形转换为tflite文件，您可以将该文件推送到android应用程序中