Kerasで学習済みモデルを初期化できない

kerasを使って学習済みモデルの評価をしようとしています。しかし、重みを設定するとタイトルのようなエラーが出てしまいました。フィルタ数、つまり最初のnOutoutPlaneは32なのでweightsの数も32になると考えましたがなぜ2になるのでしょうか。

エラー

You called `set_weights(weights)` on layer "convolution2d_1" with a  weight list of length 32, but the layer was expecting 2 weights.

コード

    def LoadModel(path, input_shape):
    	params = None
    	with open(path, 'rb') as f:
    		params = json.load(f)
    	model = Sequential()
    	model.add(Convolution2D(
    		params[0]['nOutputPlane'],
    		params[0]['kH'],
    		params[0]['kW'],
    		border_mode='same',
    		weights=params[0]['weight'],
    		bias=params[0]['bias'],
    		input_shape=input_shape))
    	model.add(LeakyReLU(0.1))
    	for param in params[1:]:
    		model.add(Convolution2D(param['nOutputPlane'], param['kH'], param['kW'], border_mode='same', weights=param['weight'], bias=param['bias']))
    		model.add(LeakyReLU(0.1))
    
    	return model

input_shape = (1, [高さ], [幅])
カーネルサイズ3x3です。
最初のweightのサイズ = (32, 3, 3)
最初のaddでエラーが出ます。

行動規範の内容に同意します

回答1件

自己解決

python
1def LoadModel(path, input_shape):
2	params = None
3	with open(path, 'rb') as f:
4		params = json.load(f)
5	model = Sequential()
6	model.add(Convolution2D(
7		params[0]['nOutputPlane'],
8		params[0]['kH'],
9		params[0]['kW'],
10		init='zero',
11		border_mode='same',
12		weights=[np.array(params[0]['weight']), np.array(params[0]['bias'])],
13		bias=True,
14		input_shape=input_shape))
15	model.add(LeakyReLU(0.1))
16	for param in params[1:]:
17		model.add(Convolution2D(
18			param['nOutputPlane'],
19			param['kH'],
20			param['kW'],
21			init='zero',
22			border_mode='same',
23			weights=[np.array(param['weight']), np.array(param['bias'])],
24			bias=True))
25		model.add(LeakyReLU(0.1))
26
27	return model