将TensorFlow Lite模型转换为ONNX模型
tflite2onnx的Python项目详细描述
TFLITE2ONNX-将TensorFlow-Lite模型转换为ONNX
tflite2onnx
将TensorFlow-Lite(TFLite)模型(*.tflite
)转换为ONNX模型(*.onnx
),
正确处理数据布局和量化语义(查看introduction blog了解详细信息)。在
If you'd like to convert a TensorFlow model (frozen graph
*.pb
,SavedModel
or whatever) to ONNX, trytf2onnx
. Or, you can firstly convert it to a TFLite (*.tflite
) model, and then convert the TFLite model to ONNX.
使用
通过pippip install tflite2onnx
安装。
安装后,您可以尝试其中之一。在
Python接口
importtflite2onnxtflite_path='/path/to/original/tflite/model'onnx_path='/path/to/save/converted/onnx/model'tflite2onnx.convert(tflite_path,onnx_path)
tflite2onnx
现在支持explicit layout,请检查
test example。在
命令行
^{pr2}$贡献
欢迎对此工具作出任何贡献。在
- 如果你觉得有什么不对劲,report bugs。在
- 如果还不支持某些运算符,则可以request a new operator。在
- 如果您能帮助启用新运算符,那就太好了,请使用How to enable a new operator加入我们。在
- 如果您有任何改进此工具的好主意,请随时展开讨论。在
资源
许可证
Apache许可证2.0版。在
- 项目
标签: