4

如何在 onnx 神经网络中获取输出层的维度？

我可以得到 onnx 图，但没有输出维度：

~/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，马库斯

4

4 回答 4

3

你可以简单地使用

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

并在 中找到形状信息inferred_model.graph.value_info。

有关完整文档，请参阅https://github.com/onnx/onnx/blob/master/docs/ShapeInference.md

于 2020-05-25T10:27:13.503 回答
1

嘿，

https://github.com/onnx/onnx-tensorrt 有这个功能。

我做了

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

Terveisin，马库斯

于 2019-08-23T12:56:17.393 回答
1

如果您使用 Java api 加载 ONNX 模型，则从会话中获取元数据信息。元数据包含有关输入和输出维度的详细信息。

于 2021-08-11T12:08:37.553 回答
0

尝试将 .onnx 文件加载到此在线 ONNX 可视化工具 (Netron)

于 2021-01-14T03:38:32.377 回答