首页
学习
活动
专区
圈层
工具
发布
社区首页 >问答首页 >keras:使用一个模型输出作为另一个模型输入

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

Stack Overflow用户
提问于 2017-12-06 23:40:47
回答 0查看 12.6K关注 0票数 15

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

代码语言:javascript
复制
model_base = InceptionResNetV2(include_top=True, weights='imagenet')
x = model_base.get_layer('avg_pool').output
x = Dense(3, activation='softmax')(x)

这是要添加的图层

代码语言:javascript
复制
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)

这是两个模型的结合

代码语言:javascript
复制
 after = Model(inputs=pre.output, outputs=x)

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

 model.compile(optimizer='sgd', loss='categorical_crossentropy', metrics=['accuracy'])

使用

代码语言:javascript
复制
pre.output

作为

代码语言:javascript
复制
after.input

但它不起作用。我怎么解决它呢?

EN

回答

页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/47678108

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档