Tensorflow 如何在python中解析toco生成的文件(.tflite)?

Tensorflow 如何在python中解析toco生成的文件(.tflite)?,tensorflow,Tensorflow,我正在使用toco优化冻结模型(.pb)。 我如何在Python中读取.tFLITE文件——类似于TF.gfile?GFile(“冻结.PB”,“RB”)?< /P> < P>点不是在Python中读取它——它是针对Android和iOS的,那里有C++库来读取它(用Android的java包装器)< P>。有两种方法可以解析.tflite文件中的信息: 1.使用flatc和json进行解析。Tensorflow在visualize.py中实现了解析函数,该函数位于Tensorflow/cont

我正在使用toco优化冻结模型(.pb)。
我如何在Python中读取.tFLITE文件——类似于TF.gfile?GFile(“冻结.PB”,“RB”)?< /P> < P>点不是在Python中读取它——它是针对Android和iOS的,那里有C++库来读取它(用Android的java包装器)

< P>。有两种方法可以解析.tflite文件中的信息:
1.使用flatc和json进行解析。Tensorflow在visualize.py中实现了解析函数,该函数位于Tensorflow/contrib/lite/tools中,您可以参考它进行解析。
2.使用纯python进行解析。Flatbuffer格式文件有一个模式,可以为不同的编程语言生成代码(链接:),您将得到一系列python文件,您可以使用以下代码来解析.tflite文件:

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

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

那么我该如何调试?我同意!这将是很好的,如果有一个Python包装器上的C++代码在桌面上进行测试/调试。谢谢。我以为我把它投了更高的票,但它没有注册