按模型层划分Keras Init顺序模型层

时间:2018-10-19 04:59:18

标签: python-3.x machine-learning keras computer-vision keras-layer

我正在尝试使用Transfer Learning构建一些应用程序。我想使用Vgg16,所以我做了这样的事情: vgg16_model = keras.applications.vgg16.VGG16(),但我想将图层从Vgg16传输到我的模型。 model = Sequential(layers=vgg16_model.layers)(我见过此here) 但这会导致我出错

  

TypeError:添加的图层必须是类Layer的实例。发现:    tensorflow.python.keras.engine.input_layer.InputLayer

如何通过vgg16层初始化我的顺序模型?

谢谢。

1 个答案:

答案 0 :(得分:0)

尝试一下:

vgg = VGG16()

model = Sequential()
model.add(vgg)
model.add(...) # add additional layers
相关问题