为了账号安全,请及时绑定邮箱和手机立即绑定

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

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

猛跑小猪 2023-08-08 10:35:06
我试图使用onnx-tf包将.onnx模型转换为.pb格式,但是,在调用onnx_tf.backend.prepare函数后,python 内核崩溃了。我正在使用下面的代码:import onnxfrom onnx_tf.backend import prepareimport tensorflowonnx_model = onnx.load(<path-to-model>)  # load onnx modeltf_rep = prepare(onnx_model)  # <------ That's where python crashestf_rep.export_graph(<output-path>)  # export the model我仔细检查了包版本依赖关系,因为错误的依赖关系在加载.onnx模型时导致了不同的错误,如下所示:onnx==1.7.0onnx-tf==1.6.0tensorflow==2.2.0tensorflow-addons==0.10.0torch==1.6.0+cu101然而,根据 Github 页面,依赖关系似乎是正确的。
查看完整描述

3 回答

?
慕运维8079593

TA贡献1876条经验 获得超5个赞

ONNX 1.6 和 1.7 之间存在较大变化,导致了许多依赖性问题。尝试使用 Onnx 1.6,这可能是您问题的根源。您在哪里找到提到的所需版本 1.7?

另外,你提到Python内核崩溃了?您在 Jupyter 笔记本中运行吗?

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



查看完整回答
反对 回复 2023-08-08
?
慕的地8271018

TA贡献1796条经验 获得超4个赞

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



查看完整回答
反对 回复 2023-08-08
?
牛魔王的故事

TA贡献1830条经验 获得超3个赞

您可以安装此提交7d8fa7d88fab469253d75e5e11cf9cdcb90104c4的onnx-tf吗

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


查看完整回答
反对 回复 2023-08-08
  • 3 回答
  • 0 关注
  • 221 浏览
慕课专栏
更多

添加回答

举报

0/150
提交
取消
意见反馈 帮助中心 APP下载
官方微信