首页 > 解决方案 > vgg net的输入层在逐层复制时不会复制到新的顺序模型中

问题描述

我正在尝试将 vgg16 层(最后一层除外)复制到新的 Sequential 模型中。我遍历 vgg16 的每一层,除了最后一层,并将它们添加到我的新模型中。但是当我打印模型的摘要时,我看到 vgg16 的输入层没有被复制。我正在关注本教程https://www.youtube.com/watch?v=oDHpqu52soI,其中使用了与我相同的代码,但也复制了输入层。我想知道是否有人可以帮助我解释为什么我的代码不复制输入层。

我正在使用 keras 2.2.4-tf

这是我的代码:

vgg16_model = keras.applications.vgg16.VGG16(weights='imagenet')
model = Sequential()
for layer in vgg16_model.layers[:-1]:
   model.add(layer)
model.summary()

这些是“模型”的前几层(缺少输入层):

在此处输入图像描述 这些是“vgg16_model”的前几层,也有输入层。

在此处输入图像描述

标签: keras-layervgg-net

解决方案


推荐阅读