[分享]用机器学习生成图片:GAN的局限性以及如何GAN的更爽_AI.人工智能讨论区_Weblogic技术|Tuxedo技术|中间件技术|Oracle论坛|JAVA论坛|Linux/Unix技术|hadoop论坛_联动北方技术论坛  
网站首页 | 关于我们 | 服务中心 | 经验交流 | 公司荣誉 | 成功案例 | 合作伙伴 | 联系我们 |
联动北方-国内领先的云技术服务提供商
»  游客             当前位置:  论坛首页 »  自由讨论区 »  AI.人工智能讨论区 »
总帖数
1
每页帖数
101/1页1
返回列表
0
发起投票  发起投票 发新帖子
查看: 3196 | 回复: 0   主题: [分享]用机器学习生成图片:GAN的局限性以及如何GAN的更爽        下一篇 
huang.wang
注册用户
等级:中将
经验:17623
发帖:407
精华:1
注册:1970-1-1
状态:离线
发送短消息息给huang.wang 加好友    发送短消息息给huang.wang 发消息
发表于: IP:您无权察看 2018-10-4 1:38:16 | [全部帖] [楼主帖] 楼主

本文转自公众号 译智社


计算机视觉(Computer Vision, CV)领域近年来发生了巨大的变化。在 2012 年之前,CV 的主要研究方法是使用人工设计(hand-designed)的特征完成各种任务(见下图)。 

image.png

2012 年使用深度神经网络(Deep Neural Network, DNN)ImageNet 的分类任务上取得了巨大成功(见下图)。 

image.png

从此深度学习(Deep Learning)的相关研究如火如荼地展开了,比如说下面这三个栗子:  

image.png

1、物体识别(Object detection)  [Redmon etal., 2018] 

2、对人体的理解(Human understanding) [Guler et al., 2018]  

3、自动驾驶(Autonomous driving)  [Zhao et al., 2017] 


图形学中的尝试:趁手的武器 or 白费功夫? 

在传统的图形学管线(pipeline)中,输出图像需要经过建模、材质贴图、光照、渲染等一系列繁琐的步骤(见下图)。 

image.png

现在大家看到了 Deep Learning 的潜力,那我们自然的就有个想法:有没有可能使用 Deep Learning 简化计算机图形学(Computer Graphics)的研究呢? 

一个直接的想法是把 DNN 「倒过来用」。之前的 DNN 可能是输入一幅图像,输出一个标签(比如说猫),那我们能不能输入「猫」这个字,输出一张猫的照片呢? 

image.png

很遗憾,答案是 No!因为这种任务实在太复杂啦!我们很难让 DNN 凭空输出图像这样的高维数据(High dimensional data)(这里的「高维」可以理解成数据量大)。实际上,在很长一段时间里,DNN 只能输出数字这种简单的、低分别率的小图像,就像下面这样: 

image.png

而想要生成想游戏场景这类的图片,这种方法根本没用。所以,我们必须得想出更厉害滴东西完成这项任务(使命感爆棚)! 


GAN 就完了?幼稚! 

于是…在月黑风高的某一天(画风逐渐跑偏),一个叫做生成对抗网络 (Generative Adversarial Network),也就是大名鼎鼎的 GAN —— 的东西横空出世。作者是下面这位小哥和他的小伙伴们: 

image.png

那么,我们该怎么 GAN 出图像呢?且听我细细道来~ 

一般来说,GAN 中包含两种类型的网络 GD。其中,G 为 Generator,它的作用是生成图片,也就是说,在输入一个随机编码(random code)之后,它将输出一幅由神经网络自动生成的、假的图片 G(z)。 

另外一个网络 DDiscriminator 是用来判断的,它接受 G 输出的图像作为输入,然后判断这幅图像的真假,真的输出 1,假的输出 0。 

image.png

在两个网络互相博弈(金坷垃日本人:不邀哒架)的过程中,两个网络的能力都越来越高:G 生成的图片越来越像真的图片,D 也越来越会判断图片的真假。到了这一步,我们就能「卸磨杀驴」—— 丢掉 D 不要了,把 G 拿来用作图片生成器。 

正式一点儿讲(上公式啦),我们就是要在最大化 D 的能力的前提下,最小化 DG 的判断能力,这是一个最小最大值问题,它的学习目标是:

 image.png

为了增强 D 的能力,我们分别考虑输入真的图像和假的图像的情况。上式中第一项的 D(G(z)) 处理的是假图像 G(z),这时候评分 D(G(z)) 需要尽力降低;第二项处理的是真图像 x,这时候的评分要高。 


GAN的局限性 

即便如此,传统的 GAN 也不是万能的,它有下面两个不足: 

1. 没有用户控制(user control)能力 

在传统的 GAN 里,输入一个随机噪声,就会输出一幅随机图像。 

image.png

但用户是有想法滴,我们想输出的图像是我们想要的那种图像,和我们的输入是对应的、有关联的。比如输入一只喵的草图,输出同一形态的喵的真实图片(这里对形态的要求就是一种用户控制)。 

image.png

2. 低分辨率(Low resolution)和低质量(Low quality)问题 

尽管生成的图片看起来很不错,但如果你放大看,就会发现细节相当模糊。 

image.png


怎样改善? 

前面说过传统的 GAN 的种种局限,那么现在,我们相应的目标就是: 

1、提高 GAN 的用户控制能力 

2、提高 GAN 生成图片的分辨率和质量 

为了达到这样的目标,和把大象装到冰箱里一样,总共分三步: 

image.png

1、pix2pix:有条件地使用用户输入,它使用成对的数据(paired data)进行训练。 

2、CycleGAN:使用不成对的数据(unpaired data)的就能训练。 

3、pix2pixHD:生成高分辨率、高质量的图像。 

下面分别进行详细叙述~ 


pix2pix 

pix2pix 对传统的 GAN 做了个小改动,它不再输入随机噪声,而是输入用户给的图片: 

image.png

但这也就产生了新的问题:我们怎样建立输入和输出的对应关系。此时的 G 输出如果是下面这样,D 会判断是真图: 

image.png

但如果 G 的输出是下面这样的,D 拿来一看,也会认为是真的图片 QAQ…也就是说,这样做并不能训练出输入和输出对应的网络 G,因为是否对应根本不影响 D 的判断。 

image.png

为了体现这种对应关系,解决方案也很简单,你可以也已经想到了:我们把 G 的输入和输出一起作为 D 的输入不就好了?于是现在的优化目标变成了这样: 

image.png

这项研究还是挺成功的,大家可以去在线体验一下 demo,把草图(sketch)变成图片: 

链接地址:https://affinelayer.com/pixsrv/ 

image.png

当然,有些比较皮的用户输入了奇形怪状的草图,然后画风就变成了这样: 

image.png

pix2pix 的应用 

pix2pix 的核心是有了对应关系,这种网络的应用范围还是比较广泛的,比如: 

1. 草图变图片 [Isola, Zhu, Zhou, Efros, 2016]: 

image.png

2. 灰度图变彩色图 [Isola, Zhu, Zhou, Efros, 2016]: 

image.png

3. 自动着色 Data from [Russakovsky et al. 2015]: 

image.png

4. 交互式着色 [Zhang*, Zhu*, Isola, Geng, Lin, Yu, Efros, 2017]: 

image.png


CycleGAN 

pix2pix 必须使用成对的数据进行训练:

image.png

但很多情况下成对数据是很难获取到的,比如说,我们想把马变成斑马,现实生活中是不存在对应的真实照片的: 

image.png

现在我们就用 Cycle-constraint Adversarial Network 也就是 CycleGAN 解决这个问题。这种网络不需要成对的数据,只需要输入数据的一个集合(比如一堆马的照片)和输出数据的一个集合(比如一堆斑马的照片)就可以了。 

image.png

但是(没错我又要说但是了),直接使用不成对的数据是不奏效的。网络会直接忽略输入,随机产生输出!所以,我们还得对网络增加限制(constraint)才行。 

那怎么加限制呢?我们来思考一个现实问题。马克吐温认为,如果一把一段话从英文翻译成法文,再从法文翻译回英文,那么你应该得到跟之前原始输入的英文一样的内容。这里也是一样,如果我们把马变成斑马,然后再变回马,那么最后的马和开始输入的马应该是一样的。 

image.png

下面讲一下具体技术细节。除了之前提到的把马变成斑马的网络 G,我们还需要一个把斑马变回马的网络 F。 

那么,一匹马 xG 变成斑马 s = G(x),然后再用 F 把它变回马 F(s),得到的马和一开始的马应该是一样的,也就是 x = F(G(x))。 

image.png

反过来,斑马变马再变回斑马也要满足要求,注意这一步不能省(请读者自己想一想为什么)! 

我们同时优化 G 和 F,最后就能拿到一个想要的网络。 

CycleGAN 为什么有效 

CycleGAN 成功的原因在于它分离了风格(Style)和内容(content)。人工设计这种分离的算法是很难的,但有了神经网络,我们很容易让它学习者去自动保持内容而改变风格。 

下面是效果展示环节~ 

马变斑马 

两张图片分别是原来的马和 G,duang 的一下变出的斑马: 

image.pngimage.png

橘子变苹果: 

image.pngimage.png

可以看到,CycleGAN 能够比较准确的找到橘子的位置,并把它变成苹果。 

图像风格的迁移: 

image.png

image.png

游戏场景替换 

这个应用就很酷了,它以一些德国城市的照片作为输入,成功替换了游戏 GTA5 中的场景! 

image.png

失败例子 

在输入骑马的普京大帝照片时,输出图像里把普京也变成了斑马。 

image.png

这是因为,训练图像里并没有骑马的人,所以网络就傻掉了。 

目前暂且的解决办法是先用 Mask R-CNN 做图像分割之后再针对马进行变化,不过这个效果也不好,因为人和马在图像上有重叠的部分。这个问题需要未来解决。 

源代码 

福利来啦!这里给出 CycleGAN 和 pix2pix 的 github 项目: 

福利链接: 

https://github.com/junyanz/pytorch-CycleGAN-and-pix2pix 

这是 2017 年 github 最受欢迎的项目之一,截止到本文写作时间(2018 年 9 月),已经有 5000+ Star 了: 


pix2pix HD 

是的,我们还剩一个悬而未决的问题:分辨率和图像质量。pix2pixHD 就是用来解决这个问题的! 

假设我们输入一张高分辨率的草图: 

image.png

使用 pix2pix,结果很差(之前说过,让网络产生高维数据输出很难): 

image.png

pix2pixHD 采取了金字塔式的方法: 

1、先输出低分辨率的图片; 

2、将之前输出的低分辨率图片作为另一个网络的输入,然后生成分辨率更高的图片。 

image.png

这样,就把一个困难的问题拆分成了两个相对简单的问题~ 

最终的效果是,给定下面的高分辨率草图: 

image.png

pix2pixHD 可以实时(real time)产生这样的效果: 

image.png

pix2pixHD 也支持用户交互,比如加一辆车、添几棵树之类的: 

image.png

image.png

pix2pixHD 还有许多有趣的应用。 

比如用草图生成高分辨率人脸: 

image.png

再比如: 

1、图像增强(Image Enhancement) 

2、图像去雾(Image Dehazing) 

3、非监督动作重定向(Neural Kinematic Networks for Unsupervised Motion Retargetting) 


其他问题 

目前生成的斑马视频帧与帧之间的纹理变化较大,为了解决帧之间的连续性问题,新的研究工作应运而生: 

Video-to-Video Synthesis: 

https://arxiv.org/abs/1808.06601 

它主要的解决思路有下面三个:  

1、输入一段视频中的几帧,检查真假; 

2、把前面的帧当做后面帧的输入; 

3、使用「optical flow」,具体请看 paper。 


总结 

本文介绍了怎样用神经网络生成图片,我们使用 pix2pix 完成了基本任务,使用 CycleGAN 解决了输入数据不成对的问题,最后用 pix2pixHD 解决了图像分辨率和图像质量的问题。 



我超级酷,但是如果你回复我的话我可以不酷那么一小会儿。


——来自logo.png


赞(0)    操作        顶端 
总帖数
1
每页帖数
101/1页1
返回列表
发新帖子
请输入验证码: 点击刷新验证码
您需要登录后才可以回帖 登录 | 注册
技术讨论