系统: 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 .