Warning: file_get_contents(/data/phpspider/zhask/data//catemap/2/python/339.json): failed to open stream: No such file or directory in /data/phpspider/zhask/libs/function.php on line 167

Warning: Invalid argument supplied for foreach() in /data/phpspider/zhask/libs/tag.function.php on line 1116

Notice: Undefined index: in /data/phpspider/zhask/libs/function.php on line 180

Warning: array_chunk() expects parameter 1 to be array, null given in /data/phpspider/zhask/libs/function.php on line 181
Python 正在尝试使用onnx tf库将.onnx模型转换为.pb模型_Python_Tensorflow - Fatal编程技术网

Python 正在尝试使用onnx tf库将.onnx模型转换为.pb模型

Python 正在尝试使用onnx tf库将.onnx模型转换为.pb模型,python,tensorflow,Python,Tensorflow,我试图通过使用onnx tf包将.onnx模型转换为.pb格式,但是,在调用onnx\u 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)

我试图通过使用onnx tf包将.onnx模型转换为.pb格式,但是,在调用onnx\u 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

然而,据Github pages称,依赖关系似乎是正确的。

tf
pytorch
之前导入
onnx
onnx-tf
相关库应该会有所帮助。该问题可能是由版本冲突引起的,因为onnx和onnx tf都使用了
tf

在onnx 1.6和1.7之间有很大的变化,导致了许多依赖性问题。尝试使用onnx1.6,这可能是问题的根源。您在哪里发现1.7被称为所需版本

另外,您提到Python内核崩溃了吗?你是在用笔记本电脑跑步吗


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

您能在NX tf上安装这个commit 7d8fa7d88fab469253d75e5e11cf9cdcb90104c4吗


在导入顺序和此提交之间,问题已为我解决

从我的代码中可以看出,导入顺序与您建议的一致!你能为未来的读者确认一下,是NX1.6解决了这个问题,还是在独立脚本中运行?从1.7降级到1.6解决了这个问题!事实上,我以前在Jupyter笔记本上运行它,但是,在Python解释器中运行它时,同样的崩溃发生了
onnx==1.7.0
onnx-tf==1.6.0
tensorflow==2.2.0
tensorflow-addons==0.10.0
torch==1.6.0+cu101