onnx模型转tensorflow的pb格式
遇到了很多坑,最常见的就是少了什么模块,或某个模块没有啥属性,
如: module ‘tensorflow’ has no attribute ‘ceil’
记录一下现在转换成功的方式,
首先:git clone https://github.com/onnx/onnx-tensorflow.git
之后在终端安装:
pip install -e.
pip install tensorflow-addons
都安装好了后就可以直接在终端转换了:
onnx-tf convert -i /path/to/input.onnx -o /path/to/