如何将Onxx模型（.onxx）转换为tensoreflow（.pb）模型

Question

我正在尝试将.onxx模型转换为.pd模型，我已经编写了代码，但出现错误：@@ 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

我用这个解决了这个问题。Tensorflow Backend for ONNX。让我知道您是否有任何问题。