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



我正在使用VGG16作为我的基本CNN模型:

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

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

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

我找到了问题的答案。

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

最新更新