我是靠谱客的博主 结实紫菜,这篇文章主要介绍安装ONNX-TF记录,现在分享给大家,希望可以做个参考。

系统: ubuntu 16.04

tensorflow版本: 2.3.0

onnx-tf : https://gitee.com/Yexiaopei/onnx-tensorflow

 

pip install tensorflow-addons

然后按照官网步骤走:

git clone xxxxx

cd xxxxx

pip install -e .

 

cmd输入:

onnx-tf -h 

出现信息,成功

 

转换代码:

onnx-tf convert -i ./centerface.onnx -o ./outgraph.pb

 

================

然而tf 2.3.0转换出来的pb是权重和图分离的,过于麻烦。于是换成了tf1.15操作了一遍

============

转出的pb结构很乱,尝试转keras

https://github.com/nerox8664/onnx2keras/

安装onnx :

pip install onnx==1.5.0   (搭配tf2.0.0)

#conda install -c conda-forge onnx

安装onnx2keras:进入onnx2keras文件夹:

pip install -e .

 

最后

以上就是结实紫菜最近收集整理的关于安装ONNX-TF记录的全部内容,更多相关安装ONNX-TF记录内容请搜索靠谱客的其他文章。

本图文内容来源于网友提供,作为学习参考使用,或来自网络收集整理,版权属于原作者所有。
点赞(140)

评论列表共有 0 条评论

立即
投稿
返回
顶部