相关疑难解决方法(0)

onnx图,如何获取输出维度?

如何获取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
Run Code Online (Sandbox Code Playgroud)

特尔维辛,马库斯

python-3.x onnx

6
推荐指数
2
解决办法
6013
查看次数

标签 统计

onnx ×1

python-3.x ×1