n12_pepe_zoo.py 文件源码

def build_mod12(opt=adam()):
    n = int(2 * 1024)
    in1 = Input((128,), name='x1')
    x1 = fc_block1(in1, n, d=0.2)
    x1 = fc_identity(x1, n, d=0.2)
    x1 = fc_identity(x1, n, d=0.2)

    in2 = Input((1024,), name='x2')
    x2 = fc_block1(in2, n, d=0.2)
    x2 = fc_identity(x2, n, d=0.2)
    x2 = fc_identity(x2, n, d=0.2)

    x = merge([x1, x2], mode='concat', concat_axis=1)

    x = fc_identity(x, n, d=0.2)
    x = fc_identity(x, n, d=0.2)
    x = fc_identity(x, n, d=0.2)
    x = fc_block1(x, n)

    out = Dense(4716, activation='sigmoid', name='output')(x)

    model = Model(input=[in1, in2], output=out)
    model.compile(optimizer=opt, loss='categorical_crossentropy')

    model.summary()
    # plot(model=model, show_shapes=True)
    return model