如何在python中解析toco生成的文件（.tflite）？

Question

我正在使用toco来优化冻结模型（.pb）。 我如何阅读python中的.tflite文件 - 类似于tf.gfile.GFile（'frozen.pb'，'rb'）？

Answer 1

关键是不能在Python中阅读 - 它适用于Android和iOS，其中有C ++库可供阅读（使用适用于Android的Java Wrapper）

Answer 2

.tflite文件是平面缓冲区格式，据我所知，有两种方法可以解析.tflite文件中的信息：
1.通过flatc和json解析Tensorflow已经在tensorflow / contrib / lite / tools中的visualize.py中实现了解析功能，您可以引用它进行解析。
2.通过纯python解析Flatbuffer格式的文件具有一个架构，可以为不同的编程语言（链接：https://google.github.io/flatbuffers/flatbuffers_guide_tutorial.html）生成代码，您将获得一系列python文件，并且可以使用以下代码来解析.tflite文件：

from Model import Model
buf = open('you-tflite-file', 'rb').read()
buf = bytearray(buf)
model = Model.getRootAsModel(buf, 0)

现在您可以从模型对象获取信息。