如何测试.tflite模型以证明其行为与使用相同测试数据的原始模型相同?

Jor*_*nez 18 python tensorflow tensorflow-lite

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

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

Jin*_*hao 14

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

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

在这里回答了这个问题.

您可以阅读此TensorFlow官方github文档 以获取详细信息.

我还发现了一个非常好的可视化工具,它可以直接加载.tflite文件,让你检查你的模型架构和模型权重.