如何知道Tensorflow Lite模型的输入/输出功能信息？

Question

我是移动开发人员。我想用各种Tensorflow精简版机型（.tflite含）MLKit。

但是有一些问题，我不知道如何知道.tflite模型的输入/输出功能信息（这些将是设置参数）。

有什么办法知道吗？

对不起，英语不好，谢谢。

---

更新（18.06.13。）：

我找到了这个网站https://lutzroeder.github.io/Netron/。这可以根据您上载的模型（如.mlmode或类似模型.tflite）可视化图形并查找输入/输出形式。

这是示例屏幕截图！ https://lutzroeder.github.io/Netron示例

Answer 1

如果您已经拥有自己无法创建的tflite模型，并且 想要查看tflite文件内部并了解输入和输出，则可以使用flatc工具将模型转换为.json文件并进行读取。

首先克隆flatbuffers回购并构建flatc。

git clone https://github.com/google/flatbuffers.git

然后您必须将tensorflow schema.fbs存储在本地。检出tensorflow github或下载该文件。然后，您可以运行flatc，json然后从输入tflite模型生成文件。

flatc -t schema.fbs -- input_model.tflite

这将创建一个input_model.json易于阅读的文件。