Keras：有没有办法“pop（）”顶层？

Question

在Keras中有一个名为pop（）的功能，可以删除模型的底层。有没有办法删除模型的顶层？

我有一个完全保存的预先训练的变分自动编码器，我试图只加载解码器（底部四层）。

我正在使用带有Tensorflow后端的Keras。

Answer 1

Keras pop（）删除最后一层（也称为顶层），而不是底层。

我建议您使用model.summary（）打印出图层列表，然后使用pop（）直到只剩下必要的图层。

Answer 2

pop（0）对我有用

from keras.applications import vgg16
vgg = vgg16.VGG16(include_top=False, input_shape=(604,604,3))
vgg.summary()
vgg.layers.pop(0)
vgg.summary()
vgg.layers.pop()
vgg.summary()