如何在Keras TensorFlow中组合两个预定义模型?

2024-10-01 04:57:46 发布

您现在位置:Python中文网/ 问答频道 /正文

我想构建一个具有两个输入的神经网络,并使用EfficientNetB1提取特征并在新层上进行微调,因此我编写了以下代码:

def createNet(self,shape):
    FE1 = K.applications.EfficientNetB1(include_top=False, input_shape=shape)
    FE2 = K.applications.EfficientNetB1(include_top=False, input_shape=shape)
    inp1 = FE1.input
    out1 = FE1.layers[-1].output
    inp2 = FE2.input
    out2 = FE2.layers[-1].output

    merged_out = K.layers.concatenate((out1, out2))
    # .... other layers
    self.model = K.models.Model(inputs=[inp1, inp2], outputs=[merged_out])
    
    self.model.summary()

但我有一个错误:

ValueError: The name "stem_conv_pad" is used 2 times in the model. All layer names should be unique.

那么我如何建立我的模型呢


Tags: selffalseinputoutputmodelincludelayerstop