Python实现的一个简单的GAN（生成对抗网络）例子

KENYCHEN奉孝

于 2023-12-07 10:38:55 发布

阅读量855

点赞数 9

文章标签：生成对抗网络机器学习深度学习

本文链接：https://blog.csdn.net/keny88888/article/details/134849192

版权

一个简单的GAN（生成对抗网络）例子

以下是使用Python实现的一个简单的GAN（生成对抗网络）例子，它可以生成手写数字图像

```python
# Importing libraries
import numpy as np
import matplotlib.pyplot as plt
from tensorflow.keras.datasets import mnist
from tensorflow.keras.layers import Input, Dense, Reshape, Flatten, Dropout
from tensorflow.keras.layers import BatchNormalization, Activation, ZeroPadding2D
from tensorflow.keras.layers import LeakyReLU
from tensorflow.keras.models import Sequential, Model
from tensorflow.keras.optimizers import Adam

# Loading the dataset
(X_train, _), (_, _) = mnist.load_data()

# Rescaling the images to [-1, 1] range
X_train = X_train / 127.5 - 1.
X_train = np.expand_dims(X_train, axis=3)

# Defining the generator network
generator = Sequential()
generator.add(Dense(256, input_dim=100))
generator.add(LeakyReLU(alpha=0.2))
generator.add(BatchNormalization(momentum=0.8))
generator.add(Dense(512