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。如果您有任何问题，请告诉我。从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

有关更多详细信息，请参见this issue

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

2 个答案: