Keras 汽车旅馆 Theano Fynstion

Question

我正在尝试将经过训练的模型（下面给出的代码）转换为 theano 函数。但是我收到以下错误：AttributeError: 'Dense' object has no attribute 'output'。

我的模型代码：

model = Sequential()

model.add(Convolution2D(32, 3, 3, border_mode='same',
                        input_shape=(img_channels, img_rows, img_cols)))
model.add(Activation('relu'))
model.add(Convolution2D(32, 3, 3))
model.add(Activation('relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Dropout(0.25))

model.add(Convolution2D(64, 3, 3, border_mode='same'))
model.add(Activation('relu'))
model.add(Convolution2D(64, 3, 3))
model.add(Activation('relu'))
model.add(MaxPooling2D(pool_size=(2, 2)))
model.add(Dropout(0.25))

model.add(Flatten())
model.add(Dense(512))
model.add(Activation('relu'))
model.add(Dropout(0.5))
model.add(Dense(nb_classes))
model.add(Activation('softmax'))

# let's train the model using SGD + momentum (how original).
sgd = SGD(lr=0.01, decay=1e-6, momentum=0.9, nesterov=True)
model.compile(loss='categorical_crossentropy',
              optimizer=sgd,
              metrics=['accuracy'])

我用来将 Keras 模型转换为 theano 函数的代码遵循 this tutorial:

 from keras import backend as K
    get_last_layer_output = K.function([model.layers[0].input],
                                       [model.layers[-1].output])
    y=f(patches)

谁能告诉我该怎么做？

Answer 1

尝试model.layers[-1].get_output(train=False)。 original Keras tutorial 可能已过时。

Keras 汽车旅馆 Theano Fynstion

Keras model to Theano function

python

deep-learning

keras