如何从现有模型向顺序模型添加图层?

时间:2019-04-17 22:54:04

标签: python python-3.x keras

我将VGG16用作基本的CNN模型:

base_model = VGG16(weights='imagenet', include_top=False, input_shape=(64,64,3) )

我想创建一个新模型,然后在添加自己的层之前,将VGG16的层一一添加到我的模型中。

我知道如何使用model.add()命令手动添加新闻图层,但是我不确定如何遍历基本模型图层并将其添加到模型中

1 个答案:

答案 0 :(得分:0)

我找到了问题的答案。

bn = []
for layer in base_model.layers:
    bn.append(layer)
model = Sequential(bn)
model.summary()