百行内实现首个生成对抗网络及DCGAN入门
1. 准备工作
在开始训练生成对抗网络(GAN)之前,需要完成一系列准备步骤:
- 在 full-gan 文件夹中创建 train.py 和 run.py 文件。
- 若还未创建,在 full-gan 文件夹内创建 data 文件夹。
- 确保 full-gan 文件夹位于 $HOME/full-gan 位置,否则可能需要更新 run.sh 脚本。
2. 训练GAN的两个主要部分
训练GAN主要包含两个部分:训练类的定义和运行该训练类的脚本。
2.1 训练类定义
训练类的实现可分为以下几个关键组件:
- 导入模块 :
#!/usr/bin/env python3
from gan import GAN
from generator import Generator
from discriminator import Discriminator
from keras.datasets import mnist
from random import randint
import numpy as np
import matplotlib.pyplot as plt
百行代码实现GAN与DCGAN入门
超级会员免费看
订阅专栏 解锁全文

被折叠的 条评论
为什么被折叠?



