- 一、案例描述
- 二、代码详解
-
- 2.1 获取数据
- 2.2 数据集类
- 2.3 构建判别器
-
- 2.3.1 构造函数
- 2.3.2 测试判别器
- 2.4 构建生成器
-
- 2.4.1 构造函数
- 2.4.2 测试生成器
- 2.5 训练GAN
- 2.6 测试生成器的生成效果
在上文利用GAN进行手写数字生成的基础上，本文介绍利用生成对抗网络进行彩色人脸图像生成。
本文选取的数据集是CelebA数据集，其官网地址为：http://mmlab.ie.cuhk.edu.hk/projects/CelebA.html。由于该数据集巨大，有202599张图像，因此我们选择前20000张图像进行训练，这20000张图像可以从下方链接获取：
