tensorflow - 如何将 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？

标签： tensorflowkerasdeep-learningneural-networkpytorch

Answer 1

我用这个解决了这个问题。 用于 ONNX 的 TensorFlow 后端。如果您有任何问题，请告诉我。从 tensorflow 2.0 更改为 1.14。也许可以解决问题。