2016-07-07 4 views
0

prototxtモデルの説明とcaffemodelチューニングパラメータファイル(約16 Mbのサイズ)で取得したCaffeプロジェクトがあります。静的解析や実行時のモデルの中にいくつのニューロンや重みがあるかを簡単に調べる方法はありますか?Caffeソリューションでのニューロンとウエイトのカウント数

prototxtファイルをNetscopeに視覚化することはできますが、畳み込みレイヤーは24個あり、プールレイヤーは7個あるため、どこからカウントを開始するのかはわかりません。私もthis Google Groups questionを見ましたが、私の問題にどのように適用するのか分かりません。

答えて

2

はい、カフェモデルを見なくても、prototxtファイルから重量パラメータを数えることができます。畳み込みレイヤの場合は、いくつのフィルタが存在し、どのフィルタサイズがあるかを見ることができます。例えば、畳み込みサイズが3x3であり、フィルタの数が64である場合、そのレイヤのパラメータ数は576となる。完全に接続されたレイヤの場合、重みパラメータの数は、前のレイヤのサイズのサイズと等しい現在のレイヤー、例えば前のレイヤーの出力がサイズ200で、現在のレイヤーが100個の隠れユニットを持つ場合、200x100。前のレイヤが畳み込みレイヤの場合、前のレイヤのサイズは入力イメージのサイズにも依存します。完全に接続されたレイヤを有するネットワークは、固定サイズの入力をとるので、最後の畳み込みレイヤの出力サイズは、入力サイズおよび他のレイヤパラメータhttp://cs231n.github.io/convolutional-networks/に基づいて計算することができる。

バイアスパラメータがある場合、そこにあるフィルタの数と完全に接続されたレイヤの数を追加する必要があります。プーリングレイヤーにはウェイトパラメーターがないため、これらは無視することができます。最後に、floatのサイズである4でサイズを掛けることができます。

関連する問題