GAN-生成式对抗网络

生成式对抗网络(GAN)是近年来大热的深度学习模型。最近正好有空看了这方面的一些论文,跑了一个 GAN 的代码,于是写了这篇文章来介绍一下 GAN。

本文主要分为三个部分:

介绍原始的 GAN 的原理 

同样非常重要的 DCGAN 的原理 

如何在 Tensorflow 跑 DCGAN 的代码,生成如题图所示的动漫头像,附送数据集哦 :-)

  一、GAN 原理介绍

说到 GAN 第一篇要看的 paper 当然是 Ian Goodfellow 大牛的 Generative Adversarial Networks,这篇 paper 算是这个领域的开山之作。

GAN 的基本原理其实非常简单,这里以生成图片为例进行说明。假设我们有两个网络,G(Generator)和 D(Discriminator)。正如它的名字所暗示的那样,它们的功能分别是:

G 是一个生成图片的网络,它接收一个随机的噪声 z,通过这个噪声生成图片,记做 G(z)。

D 是一个判别网络,判别一张图片是不是 “真实的”。它的输入参数是 x,x 代表一张图片,输出 D(x)代表 x 为真实图片的概率,如果为 1,就代表 100% 是真实的图片,而输出为 0,就代表不可能是真实的图片。

在训练过程中,生成网络 G 的目标就是尽量生成真实的图片去欺骗判别网络 D。而 D 的目标就是尽量把 G 生成的图片和真实的图片分别开来。这样,G 和 D 构成了一个动态的 “博弈过程”。

最后博弈的结果是什么?在最理想的状态下,G 可以生成足以 “以假乱真” 的图片 G(z)。对于 D 来说,它难以判定 G 生成的图片究竟是不是真实的,因此 D(G(z)) = 0.5。

这样我们的目的就达成了:我们得到了一个生成式的模型 G,它可以用来生成图片。

以上只是大致说了一下 GAN 的核心原理,如何用数学语言描述呢?这里直接摘录论文里的公式:

简单分析一下这个公式:

整个式子由两项构成。x 表示真实图片,z 表示输入 G 网络的噪声,而 G(z) 表示 G 网络生成的图片。

D(x) 表示 D 网络判断真实图片是否真实的概率(因为 x 就是真实的,所以对于 D 来说,这个值越接近 1 越好)。而 D(G(z)) 是 D 网络判断 G 生成的图片的是否真实的概率。

G 的目的:上面提到过,D(G(z)) 是 D 网络判断 G 生成的图片是否真实的概率,G 应该希望自己生成的图片 “越接近真实越好”。也就是说,G 希望 D(G(z)) 尽可能得大,这时 V(D, G) 会变小。因此我们看到式子的最前面的记号是 min_G。

D 的目的:D 的能力越强,D(x) 应该越大,D(G(x)) 应该越小。这时 V(D,G) 会变大。因此式子对于 D 来说是求最大 (max_D)

下面这幅图片很好地描述了这个过程:

那么如何用随机梯度下降法训练 D 和 G?论文中也给出了算法:

这里红框圈出的部分是我们要额外注意的。第一步我们训练 D,D 是希望 V(G, D) 越大越好,所以是加上梯度 (ascending)。第二步训练 G 时,V(G, D) 越小越好,所以是减去梯度 (descending)。整个训练过程交替进行。

  二、DCGAN 原理介绍

我们知道深度学习中对图像处理应用最好的模型是 CNN,那么如何把 CNN 与 GAN 结合?DCGAN 是这方面最好的尝试之一(点击查看论文)

DCGAN 的原理和 GAN 是一样的,这里就不在赘述。它只是把上述的 G 和 D 换成了两个卷积神经网络(CNN)。但不是直接换就可以了,DCGAN 对卷积神经网络的结构做了一些改变,以提高样本的质量和收敛的速度,这些改变有:

取消所有 pooling 层。G 网络中使用转置卷积(transposed convolutional layer)进行上采样,D 网络中用加入 stride 的卷积代替 pooling。

在 D 和 G 中均使用 batch normalization

去掉 FC 层,使网络变为全卷积网络

G 网络中使用 ReLU 作为激活函数,最后一层使用 tanh

D 网络中使用 LeakyReLU 作为激活函数

DCGAN 中的 G 网络示意:

  三、DCGAN in Tensorflow

好了,上面说了一通原理,下面说点有意思的实践部分的内容。

DCGAN 的原作者用 DCGAN 生成 LSUN 的卧室图片,这并不是特别有意思。之前在网上看到一篇文章 Chainer で顔イラストの自動生成 - Qiita ,是用 DCGAN 生成动漫人物头像的,效果如下:

这是个很有趣的实践内容。可惜原文是用 Chainer 做的,这个框架使用的人不多。下面我们就在 Tensorflow 中复现这个结果。

1. 原始数据集的搜集

首先我们需要用爬虫爬取大量的动漫图片,原文是在这个网站:http://safebooru.donmai.us / 中爬取的。我尝试的时候,发现在我的网络环境下无法访问这个网站,于是我就写了一个简单的爬虫爬了另外一个著名的动漫图库网站:konachan.net。

爬虫代码如下:

这个爬虫大概跑了一天,爬下来 12 万张图片,大概是这样的:

可以看到这里面的图片大多数比较杂乱,还不能直接作为数据训练,我们需要用合适的工具,截取人物的头像进行训练。

2. 头像截取

截取头像和原文一样,直接使用 github 上一个基于 opencv 的工具:nagadomi。

简单包装下代码:

截取头像后的人物数据:

这样就可以用来训练了!

如果你不想从头开始爬图片,可以直接使用我爬好的头像数据(275M,约 5 万多张图片):百度云盘 提取码:g5qa

3. 训练

DCGAN 在 Tensorflow 中已经有人造好了轮子:carpedm20/DCGAN,我们直接使用这个代码就可以了。

不过原始代码中只提供了有限的几个数据库,如何训练自己的数据?在 model.py 中我们找到读数据的几行代码:

if config.dataset == 'mnist':

data_X, data_y = self.load_mnist()

else:

data = glob(os.path.join("./data", config.dataset, "*.jpg"))

这样读数据的逻辑就很清楚了,我们在 data 文件夹中再新建一个 anime 文件夹,把图片直接放到这个文件夹里,运行时指定 --dataset anime 即可。

运行指令(参数含义:指定生成的图片的尺寸为 48x48,我们图片的大小是 96x96,跑 300 个 epoch):

python main.py --image_size 96 --output_size 48 --dataset anime --is_crop True --is_train True --epoch 300

4. 结果

第 1 个 epoch 跑完(只有一点点轮廓):

第 5 个 epoch 之后的结果:

第 10 个 epoch:

200 个 epoch,仔细看有些图片确实是足以以假乱真的:

题图是我从第 300 个 epoch 生成的。

  四、总结和后续

简单介绍了一下 GAN 和 DCGAN 的原理。以及如何使用 Tensorflow 做一个简单的生成图片的 demo。

一些后续阅读:

Ian Goodfellow 对 GAN 一系列工作总结的 ppt,确实精彩,推荐:GAN 之父 NIPS 2016 演讲现场直击:全方位解读生成对抗网络的原理及未来(http://it.sohu.com/20161210/n475485860.shtml)

GAN 论文汇总,包含 code:zhangqianhui/AdversarialNetsPapers(https://github.com/zhangqianhui/AdversarialNetsPapers)

©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 199,393评论 5 467
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 83,790评论 2 376
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 146,391评论 0 330
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 53,703评论 1 270
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 62,613评论 5 359
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 48,003评论 1 275
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 37,507评论 3 390
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 36,158评论 0 254
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 40,300评论 1 294
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 35,256评论 2 317
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 37,274评论 1 328
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 32,984评论 3 316
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 38,569评论 3 303
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 29,662评论 0 19
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 30,899评论 1 255
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 42,268评论 2 345
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 41,840评论 2 339

推荐阅读更多精彩内容