vanilla_gan.py 文件源码

def generator_model(noise_dim=100, aux_dim=47, model_name="generator"):
    gen_input = Input(shape=(noise_dim,), name="noise_input")

    # Dense Layer 1
    x = Dense(1024)(gen_input)
    x = BatchNormalization()(x)
    x = LeakyReLU(0.2)(x)

    # Dense Layer 2
    x = Dense(1024)(x)
    x = BatchNormalization()(x)
    x = LeakyReLU(0.2)(x)

    # Dense Layer 3
    x = Dense(1024)(x)
    x = BatchNormalization()(x)
    x = LeakyReLU(0.2)(x)

    # Dense Layer 4
    x = Dense(400)(x)
    x = BatchNormalization()(x)
    x = Activation("tanh")(x)

    generator_model = Model(input=gen_input, output=[x], name=model_name)

    return generator_model