Ids-Based-On-GAN-CNN/Network/GAN/Generator.py

from tensorflow.keras import Sequential, Input, Model
from tensorflow.keras.layers import Dense, LeakyReLU, Dropout, BatchNormalization


def build_generator(latent_dim):
    model = Sequential()
    model.add(Dense(64, input_dim=latent_dim, kernel_initializer='glorot_uniform'))
    model.add(LeakyReLU(alpha=0.1))
    model.add(BatchNormalization(momentum=0.8))
    model.add(Dense(128, kernel_initializer='glorot_uniform'))
    model.add(LeakyReLU(alpha=0.1))
    model.add(BatchNormalization(momentum=0.8))
    model.add(Dense(121, kernel_initializer='glorot_uniform'))
    model.summary()

    noise = Input(shape=(latent_dim,))
    fake = model(noise)
    return Model(noise, fake)