ONNXモデルから重み/バイアスマトリックス値を取得する方法。現在、から入力、カーネルサイズ、ストライド、パッドの値を取得できますmodel.onnx
。モデルをロードしてから、グラフノードを読み取って同じものを取得します。
import onnx
m = onnx.load('model.onnx')
print(m.graph.node)
公式のgitリポジトリに問題を投稿した後、私は質問に対する答えを得ました。の初期化子から重み値にアクセスできますm.graph
。
weights = m.graph.initializer
重み行列を取得するにはnumpy_helper
、onnxから使用する必要があります。
from onnx import numpy_helper
w1 = numpy_helper.to_array(weights[0])
この記事はインターネットから収集されたものであり、転載の際にはソースを示してください。
侵害の場合は、連絡してください[email protected]
コメントを追加