onnxファイルからパラメータ(weight, bias)を抽出する
やりたいこと
- ニューラルネットワークの表現形式の一つであるonnxファイルからパラメータを取り出したい。
- PyTorchなどのフレームワークで一旦読み込んでからパラメータを取り出すこともできるかと思いますが、onnxだけで完結する方法を共有します。
実装
- サンプルとしてonnxのリポジトリにあるyolo v3のonnxを使いました。
- onnxをNetronで可視化した一部が以下の通り。最初のConvレイヤーのweightの名前は「W74」らしいです。
- 以下のsnippetでW74のweightをprintしています。Netronで見た値と同じであることが確認できます。
