25b*_*3nk 1 protocol-buffers onnx
如何从 ONNX 模型获取权重/偏置矩阵值,我目前可以从model.onnx. 我加载模型,然后读取图形节点以获得相同的结果:
import onnx
m = onnx.load('model.onnx')
print(m.graph.node)
Run Code Online (Sandbox Code Playgroud)
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)
Run Code Online (Sandbox Code Playgroud)
| 归档时间: |
|
| 查看次数: |
3086 次 |
| 最近记录: |