protocol-buffers - 如何从 ONNX 模型中读取各个层的权重和偏差值？

Question

如何从 ONNX 模型中获取权重/偏置矩阵值，我目前可以从model.onnx. 我加载模型，然后读取图形节点以获得相同的结果：

import onnx
m = onnx.load('model.onnx')
print(m.graph.node)

标签： protocol-buffersonnx

Answer 1

from onnx import numpy_helper
MODEL_PATH = "....../resnet50"
_model = onnx.load(MODEL_PATH + "/model.onnx")
INTIALIZERS=_model.graph.initializer
Weight=[]
for initializer in INTIALIZERS:
    W= numpy_helper.to_array(initializer)
    Weight.append(W)