从零单排让神经网络找到你的style

1. 前言

这段时间城里的小伙伴们都在玩Prisma这个软件,除了在App store大红大紫的ios版本,前两天官方也推出了安卓版了,没有玩过的小伙伴可以试玩一下。

这个 App 的卖点在于:Prisma 能够将普通的照片转换成艺术感爆棚的画作。经过处理的后的照片仿佛是以梵高毕加索等人的笔触“重新”画了一遍。先放两张在Instagram的热门图:

hot pic in instagram

从图片的效果看,Prisma的实现很可能是基于Neural art算法的改进版Perceptual Loss(也有讨论说是基于Texture net),感兴趣的朋友可以阅读。Neural art算法第一次是在论文A Neural Algorithm of Artistic Style中提出。本文默认朋友们了解Convolutional Neural Networks,需要温习的朋友可以看这篇。这篇文章是我对学习Neural art算法做的总结。由于计算的复杂性较高,合理猜想Prisma是在服务器端使用了基于卷积神经网络pretrain的模型。

2. Neural art算法

i. 图像作品的构成

在计算机看来,图片只是像素点的集合,但作为人类而言,一个作品应该分为两个组成部分:内容和风格。艺术家在绘画时,首先看到的实在的物体(内容),然后在脑海中加工后使用更加夸张的色彩和纹理进行二次创作(风格),得到最后的作品。一个很自然的联想,如果我们可以提取出一幅名作的风格,再从我们自己的照片中提取出内容,那么就可以融合成一幅新的作品。

ii. VGGNet

VGGNet是一个非常常用的CNN model。在ILSVRC2012分类任务上,VGGNet模型将top-5的错误率降到了7.3%。VGGNet有不同版本,paper中使用的VGG19包括16层卷积,5层max-pooling,3层全连接。即下文提取内容和风格使用的模型。实际编码中只选部分层作为模型。

iii. 内容提取

我们使用经过了Pre-trained之后的VGG network model的图像数据,用来做object recognition,也就是提取出内容, 用来生成图像的Content Representations。提取图像特征其实是卷积神经网络的拿手好戏,举个栗子:

cnn extract image feature

顺便一提,用随机初始化的filter的效果就和一个edge detector一样识别出物体的轮廓。内容提取的方法如下:通过梯度下降(Gradient Descent)我们把一张白噪声图片生成一个内容接近用户上传的图片。我们定义p向量和x向量分别表示原始图片和生成图片,F^l和P^l对应它们在第l层的数据表示,定义squared-error loss为两种特征表示的error,通过最小化error将内容特征提取出来:

表示内容的差距

iv. 风格提取

对一幅画的“风格”也许画家本人也很难总结,Neural art的核心价值就在于提出了pattern化风格的方法。论文中作者并未给出理由,但从公式中可以反映出作者把风格表示为特征层之间的关系(或者说是同一层不同filter的互相关),把这个关系定义为Gram matrix(Gij^l),下图表示本人对Gij^l的定义和计算方式的理解:

Gij^l的定义

在定义了风格的基础上,提取的过程和内容提取就大同小异了。定义a向量和x向量为原始图和生成图,A^l和G^l对应它们在第l层的数据表示,N,M分别是第l层feature map的个数和feature map的大小,总style loss为每一层的加权和(w是权重):

表示风格的差距

v. Total Loss

定义Total Loss为content loss和style loss的线性组合,α,β分别代表各自的权重(α/β的比例反映更注重content还是更强调style)。

total loss

在实际编码的过程中,由于content和loss的提取都使用的是白噪声图片,还会加入降噪的loss。

3. 进阶

i. Loss代码

主流的深度学习框架几乎都有对Neural art的实现,包括Mxnet,TensorFlow,Caffe,Keras等。选择Keras的实现进行一些解读。上文说过,实际编码要加入降噪的loss,所谓降噪,就是通过保证局部的一致性让图片看上去更和谐。

权值分配
每部分loss的计算

ii. Video

目前算法在图像上的实现效果很好。据说Prisma即将推出移植在视频的尝试。如果只是简单的把视频的每一帧作为独立的图像批处理,视频结果会非常不和谐,因为可能后一帧和前一帧的结果有较大差距。一个可行的解决方案是:对每一帧不是按照白噪声从头开始训练,而是用先前帧的输出初始化,这样与有助于保持视觉的连续性,同时还可以加快渲染时间。

iii. 改进版

有多个在neural art算法基础上改进的其他算法:

Texture net    http://arxiv.org/abs/1603.03417

Real-Time Texture Synthesis    http://arxiv.org/abs/1604.04382 

Perceptual loss    https://arxiv.org/abs/1603.08155

4. 微小的工作


谢谢大家!


                                       

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

推荐阅读更多精彩内容