如何将Onnx模型(.onnx)转换为Tensorflow (.pb)模型

Question

我正在尝试将.onxx模型转换为.pb模型。我已经编写了代码，但是我收到了错误：

@tf_func(tf.ceil)AttributeError:模块'tensorflow‘没有属性'ceil’

代码：

import onnx
from tensorflow.python.tools.import_pb_to_tensorboard import import_to_tensorboard
from onnx_tf.backend import prepare
onnx_model = onnx.load("original_3dlm.onnx")
tf_rep = prepare(onnx_model)
tf_rep.export_graph("model_var.pb")
import_to_tensorboard("model_var.pb", "tb_log")

如何解决这一问题？是否有其他方法将Onxx转换为Tensorflow？

Answer 1

我用这个来解决这个问题。ONNX的Tensorflow后端。如果你有什么问题请告诉我。从tensorflow 2.0改为1.14。也许可以解决这个问题。

Answer 2

据我所知，你的代码应该没问题。问题可能在于您目前使用的onnx-tf版本。pip当前安装的版本只支持TensorFlow <= 1.15。在终端中运行这个程序来安装一个最新版本的onnx-tf。

pip uninstall onnx_tf
pip install git+https://github.com/onnx/onnx-tensorflow.git

有关详细信息，请参阅本期