logo标签列表

将tflite模型转换为ONNX模型

tensorflowtensorflow-liteonnx
4

我有一个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()
2
你解决了这个问题吗？我也有同样的问题，因为我想在tfjs中运行一个tflite模型。 - NateW
你有得到任何解决方案吗？ - Meysam Sadeghi
1个回答
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)
网页内容由stack overflow 提供, 点击上面的
可以查看英文原文，
原文链接