获取keras中间层的输出会引发名称错误

Question

我想想象胶囊网络中的图层。为此，我需要中间层的形状。代码如下：

from keras.models import Model
model = Sequential()
conv1 = model.add(Conv2D(256, (9, 9), padding='valid', strides = 1, input_shape = (28, 28, 1), activation = 'relu', name = 'conv1'))
model.add(Conv2D(256, (9,9), padding='valid', strides = 2, name = 'primarycaps_conv2d'))
model.add(keras.layers.core.Reshape([-1,8]))
model.add(keras.layers.core.Lambda(squash, name = 'hello'))

layer_name = 'conv1'
intermediate_layer_model = Model(inputs=model.input,
                             outputs=model.get_layer(layer_name).output)
intermediate_output = intermediate_layer_model.predict(data)

引发以下错误：

NameError: name 'data' is not defined

怎么克服这个？

Answer 1

参考this，我得到的解决方案如下：

model = Sequential()
conv1 = model.add(Conv2D(256, (9, 9), padding='valid', strides = 1, input_shape = (28, 28, 1), activation = 'relu', name = 'conv1'))
model.add(Conv2D(256, (9,9), padding='valid', strides = 2, name = 'primarycaps_conv2d'))
model.add(keras.layers.core.Reshape([-1,8]))
model.add(keras.layers.core.Lambda(squash, name = 'hello'))

idx = 3  # index of desired layer
input_shape = model.layers[idx].get_output_shape_at(0)
print(input_shape)

获取输出形状如下：

(None, 1152, 8).

获得特定图层输出的问题已得到解决。