1 Star 0 Fork 0

tom/pytorch_to_tflite

加入 Gitee
与超过 1200万 开发者一起发现、参与优秀开源项目,私有仓库也完全免费 :)
免费加入
文件
该仓库未声明开源许可证文件(LICENSE),使用请关注具体项目描述及其代码上游依赖。
克隆/下载
贡献代码
同步代码
取消
提示: 由于 Git 不支持空文件夾,创建文件夹后会生成空的 .keep 文件
Loading...
README

Sample Conversion Example

This example will try to convert Mobilenetv2 model.

Conversion

PyTorch to ONNX

python torch_to_onnx.py

ONNX to TF

python onnx_to_tf.py

TF to TFlite

python tf_to_tflite.py

Run Inference

TF Model Inference

python run_tf_model.py

TFLite Model Inference

python run_tflite_model.py
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
1
https://gitee.com/tom_ud/pytorch_to_tflite.git
git@gitee.com:tom_ud/pytorch_to_tflite.git
tom_ud
pytorch_to_tflite
pytorch_to_tflite
main

搜索帮助