如何找到 onnx 模型的输入大小?我最终想从 python 编写脚本。
使用 tensorflow 我可以恢复图定义,从中找到输入候选节点,然后获取它们的大小。我可以用 ONNX(甚至更简单)做类似的事情吗?
谢谢
如何从 ONNX 模型获取权重/偏置矩阵值,我目前可以从model.onnx. 我加载模型,然后读取图形节点以获得相同的结果:
import onnx
m = onnx.load('model.onnx')
print(m.graph.node)
Run Code Online (Sandbox Code Playgroud)