onnx转tflite模型

CV模型

安装依赖包 

pip install onnx2tf
pip install nvidia-pyindex
pip install sng4onnx
pip install onnx-graphsurgeon

转换代码

onnx2tf -i model.onnx

会在当前文件夹生成saved_model，保存saved model和tflite格式文件

记得使用最新版onnx2tf，成功率更高。

ref

https://github.com/PINTO0309/onnx2tf

其他包：

https://github.com/MPolaris/onnx2tflite

Transformer模型

谷歌开始支持Transformer转换：

GitHub - google-ai-edge/ai-edge-torch: Supporting PyTorch models with the Google AI Edge TFLite runtime.

ai-edge-torch/docs/pytorch_converter/README.md at main · google-ai-edge/ai-edge-torch · GitHub

需要安装requirements.txt安装环境。