Concatenating or cascading multiple pretrained keras models

让人想犯罪 __ 提交于 2019-12-04 11:44:03

You need to use the functional API model for that. This kind of model works with tensors.

First you define a common input tensor:

inputTensor = Input(inputShape)

Then you call each model with this input to get the output tensors:

outputTensors = [m(inputTensor) for m in models]

Then you pass these tensors to the concatenate layer:

output = Concatenate()(outputTensors) 
output = Dense(...)(output)    
#you might want to use an Average layer instead of these two....

output = Activation('softmax')(output)

Finally, you define the complete model from start tensors to end tensors:

fullModel = Model(inputTensor,output)
易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!