python - 如何使用相同的测试数据测试 .tflite 模型以证明它的行为与原始模型一样？

Question

我已经根据经过训练的模型生成了一个 .tflite 模型，我想测试一下 tfilte 模型是否给出与原始模型相同的结果。

给出相同的测试数据并获得相同的结果。

Answer 1

您可以使用TensorFlow Lite Python 解释器 来测试您的 tflite 模型。

它允许您在 python shell 中提供输入数据并直接读取输出，就像您只是在使用普通的 tensorflow 模型一样。

我已经回答了这个问题here .

你可以阅读这个 TensorFlow lite official guide获取详细信息。

您还可以使用 Netron可视化您的模型。它允许您直接加载 .tflite 文件并检查您的模型架构和模型权重。