我正在尝试将 .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")
Run Code Online (Sandbox Code Playgroud)
如何解决这个问题?还有其他方法可以将 Onxx 转换为 Tensorflow 吗?