onnx graph, how to get output dimensions?

Question

How to get dimensions of output layers in onnx neural net?

I can get onnx graph, but there is no output dimensions:

~/onnx-tensorrt/third_party/onnx/onnx/tools/net_drawer.py --input ./weights/tiny_3l_v5_11_608.onnx  --output ./weights/tiny_3l_v5_11_608.dot --embed_docstring

Terveisin, Markus

score 5 · Answer 1 · answered May 25 '20 at 10:27

5

You can simply use

from onnx import shape_inference
inferred_model = shape_inference.infer_shapes(original_model)

and find the shape info in inferred_model.graph.value_info.

For full doc, see https://github.com/onnx/onnx/blob/master/docs/ShapeInference.md

answered May 25 '20 at 10:27

gizzmole

1,437
18
26

the value_info contains all output, rather than ops wanted whole model final output. this can be retrieved by another method. – Nicholas Jela Jan 23 '22 at 15:02
Can we know the onnx model expected input shape, e.g BCHW or BHWC – Wade Wang Apr 13 '22 at 08:28

score 1 · Answer 2 · answered Aug 23 '19 at 12:56

1

Hei,

https://github.com/onnx/onnx-tensorrt has this feature.

I did

onnx2trt yolo_tiny_3l.onnx  -t yolo_tiny_3l.onnx.txt -l -v

Terveisin, Markus

answered Aug 23 '19 at 12:56

Markus Kaukonen

334
4
10

score 1 · Answer 3 · answered Aug 11 '21 at 12:08

1

If you are using Java apis to load ONNX model, then get the metadata information from session. Metadata contains the details about input and output dimensions.

answered Aug 11 '21 at 12:08

Ambuj Kumar

71
1
6

score 0 · Answer 4 · answered Jan 14 '21 at 03:38

0

Try loading the .onnx file into this online ONNX visualiser tool (Netron)

answered Jan 14 '21 at 03:38

omatai

3,448
5
47
74

onnx graph, how to get output dimensions?

4 Answers4

Linked