小编cwz*_*zat的帖子

keras:使用一个模型输出作为另一个模型输入

我在InceptionResNetV2模型之前添加了一个密集层(预训练)这是InceptionResNetV2输出

model_base = InceptionResNetV2(include_top=True, weights='imagenet')
x = model_base.get_layer('avg_pool').output
x = Dense(3, activation='softmax')(x)
Run Code Online (Sandbox Code Playgroud)

这是将添加的图层

input1 = Input(shape=input_shape1)
pre1 = Conv2D(filters=3, kernel_size=(5, 5), padding='SAME', 
input_shape=input_shape1, name='first_dense')(input1)
pre = Model(inputs=input1, outputs=pre1)
Run Code Online (Sandbox Code Playgroud)

这是两个模型的组合

 after = Model(inputs=pre.output, outputs=x)

 model = Model(inputs=input1, outputs=after.output)

 model.compile(optimizer='sgd', loss='categorical_crossentropy', metrics=['accuracy'])
Run Code Online (Sandbox Code Playgroud)

使用

pre.output
Run Code Online (Sandbox Code Playgroud)

after.input
Run Code Online (Sandbox Code Playgroud)

但它不起作用.我该如何解决?

model computer-vision deep-learning keras keras-2

8
推荐指数
1
解决办法
6011
查看次数

标签 统计

computer-vision ×1

deep-learning ×1

keras ×1

keras-2 ×1

model ×1