如何將 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

有關更多詳細信息，請參閱此問題