尝试使用onnx-tf库将.onnx模型转换为.pb模型

Question

我试图通过使用onnx包将.onnx模型转换为.pb格式，但是在调用onnx_tf.backend.prepare函数之后，python内核就崩溃了。我使用的代码如下：

import onnx

from onnx_tf.backend import prepare
import tensorflow
onnx_model = onnx.load(<path-to-model>)  # load onnx model
tf_rep = prepare(onnx_model)  # <------ That's where python crashes
tf_rep.export_graph(<output-path>)  # export the model

我反复检查了包版本依赖项，因为错误的依赖项在加载.onnx模型时会导致不同的错误，这些错误如下：

onnx==1.7.0
onnx-tf==1.6.0
tensorflow==2.2.0
tensorflow-addons==0.10.0
torch==1.6.0+cu101

然而，据Github页面称，依赖关系似乎是正确的。

Answer 1

ONNX1.6和1.7之间发生了很大的变化，造成了许多依赖问题。尝试使用Onnx 1.6，这可能是问题的根源。你在哪里发现1.7被提及为所需的版本？

此外，您还提到Python内核崩溃吗？你是在木星笔记本里跑步吗？

我发现这个包在Juypter笔记本中更不稳定，没有打印出所有相关的错误。如果您可以使用独立的Python脚本测试您的系统，您可能会得到更多的信息。

Answer 2

在导入onnx和onnx-tf相关库之前导入tf和pytorch应该会有所帮助。这个问题可能是由于版本冲突造成的，因为onnx和onnx-tf都使用tf。

Answer 3

您能否安装此提交的7d8fa7d88f469253d75e5e11cf9cdcb90104c4的onnx-tf？

从进口订单到这次提交，这个问题已经为我解决了。