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

Question

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

提供相同的测试数据并获得相同的结果。

原文由 Jorge Jiménez 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 1

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

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

我已经 在这里 回答了这个问题。

您可以阅读这份 TensorFlow lite 官方指南 以获取详细信息。

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

原文由 Jing Zhao 发布，翻译遵循 CC BY-SA 4.0 许可协议

Answer 2

除了@miaout17给出的答案，调试/理解你的tflite模型（这是问题的精神），你可以

• 使用 --dump_graphviz 可视化图表，如此处解释 https://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/lite/toco/g3doc/cmdline_examples.md#using—dump_graphviz
• 使用 flatc 生成一个 python api，然后通过该 api 解析模型 https://google.github.io/flatbuffers/flatbuffers_guide_use_python.html
• 使用 json 从 tflite 文件生成--- flatc 并打印出来
• 使用tflite可视化工具： https ://github.com/tensorflow/tensorflow/blob/master/tensorflow/contrib/lite/tools/visualize.py（目前有一个 bug ）

原文由 Pannag Sanketi 发布，翻译遵循 CC BY-SA 4.0 许可协议