Trying to convert .onnx model to .pb model using onnx-tf library

Question

I am trying to convert .onnx model to .pb format by using onnx-tf package, however, after invoking onnx_tf.backend.prepare function, the python kernel crashes. I am using the code below:

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

I have double-checked package version dependencies, as wrong dependencies caused different errors while loading the .onnx model, and those are as follows:

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

However, the dependencies seem to be correct, according to Github pages.

Prunus Persica · Accepted Answer · 2020-09-14 14:32:38Z

1

There were large changes between ONNX 1.6 and 1.7 that caused many dependency issues. Try with Onnx 1.6, which may be the source of your issue. Where did you find 1.7 mentioned as the required version?

Additionally, you mention the Python kernel crashing? Are you running in a Jupyter notebook?

I've found this package to be a lot more unstable in Juypter notebook, and not print out all of the relevant errors. If you could test your system in a standalone Python script you may get more information.

answered Sep 14, 2020 at 14:32

Prunus Persica

1,23313 silver badges30 bronze badges

Sign up to request clarification or add additional context in comments.

2 Comments

Prunus Persica Over a year ago

Great! Could you confirm for future readers, was it Onnx 1.6 that solved the issue, or running in a standalone script?

wprazuch Over a year ago

Downgrading from 1.7 to 1.6 solved the issue! Indeed I was running it in Jupyter Notebook previously, however, the same crash happened while running it from the Python interpreter

Ruli · Accepted Answer · 2020-09-15 11:34:54Z

0

Importing onnx and onnx-tf related libraries before tf and pytorch should help. The problem is probably caused by a version clash since both onnx and onnx-tf use tf.

edited Sep 15, 2020 at 11:34

Ruli

2,83113 gold badges36 silver badges48 bronze badges

answered Sep 14, 2020 at 14:21

Prasanna Kumar R

11 bronze badge

1 Comment

wprazuch Over a year ago

As you can see from my code, the import order is as you suggested

Prasanna Kumar R · Accepted Answer · 2020-09-15 15:05:34Z

0

Can you install onnx-tf of this commit 7d8fa7d88fab469253d75e5e11cf9cdcb90104c4

Between order of imports and this commit the issue was solved for me

answered Sep 15, 2020 at 15:05

Prasanna Kumar R

11 bronze badge

Collectives™ on Stack Overflow

Trying to convert .onnx model to .pb model using onnx-tf library

3 Answers 3

2 Comments

1 Comment

Comments

Your Answer

Hot Network Questions

Collectives™ on Stack Overflow

3 Answers 3

2 Comments

1 Comment

Comments

Your Answer

Sign up or log in

Post as a guest

Related