Generative Adversarial Networks(GAN)
- Python & TensorFlow 2.0实现。
- 这个GAN用于生成手写数字图像。
import tensorflow as tf
import numpy as np
import matplotlib.pyplot as plt
def generator_network():
model = tf.keras.Sequential([
tf.keras.layers.Dense(256, input_shape=(100,), activation='relu'),
tf.keras.layers.BatchNormalization(),
tf.keras.layers.Dense(512, activation='relu'),
tf.keras.layers.BatchNormalization(),
tf.keras.layers.Dense(784, activation='tanh'),
tf.keras.layers.Reshape((28, 28, 1))
])
return model
def discriminator_network