将tflite模型转换为ONNX模型

时间:2019-09-25 09:53:26

标签: tensorflow tensorflow-lite onnx

我有一个tflite模型,我想将其转换为tensorflow或keras或ONNX格式。有办法吗?

我可以使用tf解释器将其导入并在python上运行。但是我想将其转换为上述格式之一。

import tensorflow as tf
interpreter = tf.lite.Interpreter(model_path="conv_actions_frozen.tflite")
tensors = interpreter.get_tensor_details()

1 个答案:

答案 0 :(得分:0)

现在有转换器可以将 tflite 模型转换为 onnx 格式。安装 tflite2onnx 库。 你所要做的就是:

import tflite2onnx

tflite_path = "path/to/the/tflitemodel"
onnx_path = "path/where/you/want/to/save/your/model" #modelname.onnx


tflite2onnx.convert(tflite_path,onnx_path)