我使用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()