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

最新推荐文章于 2024-08-02 16:43:37 发布

原创

最新推荐文章于 2024-08-02 16:43:37 发布

· 861 阅读

11 ·

版权

文章标签：

#生成对抗网络 #机器学习 #深度学习

一个简单的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