2017-02-28 13 views
5

事前にトレーニングされたcaffeモデルをkerasに変換したいので、レイヤーごとにレイヤーを初期化する必要があります。 私はウェイトとバイアスをマットファイルに保存し、それらをPythonワークスペースにロードしました。 私は "重み"パラメータを知っていますが、どのようにnumpyの配列を取得? ありがとうケラスのnumpy配列でレイヤーを初期化する方法

+0

がhttp://stackoverflow.com/questions/42211619/how-to-set-weights-for-convolution2d/42212349#42212349 – maz

+0

@mazをチェックし、それが有用でした –

答えて

2

Keras Layers Documentationでモデルの重量を設定する方法について詳しく知ることができます。基本的には使用:

layer.set_weights(weights):(get_weightsの出力と同じ形状の)numpyのアレイのリストからの層の重みを設定します。

また、レイヤーの作成時に直接初期化することもできます。すべてのレイヤーにnum34の配列で設定できるパラメータweightsがあります。 each layer's documentationを読んで、右の重みの形式を入力してください。例えば、Dense()層パラメータweightsこの形式を受け入れる:numpyのアレイの

リストとして初期重みを設定します。このリストには、それぞれ重みと偏りの2つの要素(input_dim、output_dim)と(output_dim)が必要です。畳み込み層のためのsource

関連する問題