以下就是keras中Model模型的定义与使用的一个简单模板,注意:传入多个参数要用中括号括起来
def Model_A(self, name=None):
layer3 = Input(shape=(128,36,36))
layer4 = Input(shape=(256,18,18))
layer5 = Input(shape=(512,9,9))
//一系列操作
return Model(inputs=[layer3,layer4,layer5], outputs=[D_mid,D_layer33], name=name)
self.G_A2B_no = self.GeneratorA2B_without_att(name='G_A2B_no_model')
D_mid = self.G_A2B_no([layer3,layer4,layer5])[0]