图片风格转移Let there be Color!: Joint End-to-end Learning of Global and Local Image Priors for Automatic Image Colorization with Simultaneous Classification

这就是论文的目的,将一张黑白图片着色为一张彩色图片

再看这篇文章前,最好是对神经网络有了初步的了解。文章第三部分,很大篇幅在介绍神经网络的一些基础知识。演示demo

在这篇论文中,他把上述过程描述为:使用现有大量图片数据来训练我们提出的模型,目的是使用 CIE lab颜色空间来预测灰度图像的chrominance(任意一种颜色与亮度相同的一个指定的参考色之间的差异)。 理解一下这个过程,就是我们在L(亮度)相同的点上来预测目标a*b*的值,使得目标a*b*的值与参考a*b*值之间的差异最小,得到我们的模型。训练得到的模型输出是每个点的a*b*值,这个a*b*值图与原来的灰度图L相结合,得到最终的彩色图。


我的理解,一个神经网络很重要的是他的目标函数(损失函数),也就是说,我们想让这个网络达到什么样的效果。这篇论文的目标函数是——将目标图片缩放到网络的输出图片的大小,并计算输出和目标之间的MSE,使得MSE最小。

I,K为两张图片  m,n为图片长宽  (i,j)像素的lab值

接下来,讲一下文章的核心部分,他的网络模型。

网络模型图

3.1 神经网络


公式(1)


公式2

不要被原论文中的公式吓坏了,尤其是(2),完全是(1)的扩展,理解(1)就好了。我们解释下模型图,下面一溜是普通的CNN,只不过这个CNN的前层没用Pooling层。上面的一溜其实是FCN,全连接网络,做语义分割用。

3.2 全局和局部信息结合

为什么要讲这两种信息结合呢?

用全局信息作为一种先验信息存在,来判别这张图片是室内还是室外,局部信息根据这个场景,就不会把家具之类的东西染成天空或者草地的颜色,而是会把他染成棕色。

3.2.1 共享低维特征

网络中的前六层卷积(紫的)是公共的,里面的参数是完全一样的,值得说明的是,这六层网络用Stride=2代替了Pooling层,这样可以增加每层的空间信息。替代详细解释

3.2.2全局图像特征

4层卷积层+3层全连接=256-dimensional vector

3.2.3中层特征

2个卷积层->H / 8 × W / 8 × 256的小长方体

3.2.4融合前两层特征

如何融合呢?

很简单,我们可以把全局的特征看为1×1×256的小长方条,把他叠成H / 8 × W / 8形成一个跟中层特征一样大小的长方体,把这两个长方体串联到一起,形成一个H / 8 × W / 8×521的长方体。【跟论文中一个意思,论文中就像是拿一根串,把两个串起来】

粉色的线表示u,v   局部和全局描述在这个地方串联

3.3分类染色

使用交叉熵和MSE来训练这个分类网络

前面那一项MSE,就是我最先提到的目标函数,为什么还要加上交叉熵呢?交叉熵函数这里还需要再了解下


左边紫色横向的线表示分类网络的输入,右侧竖向的线是输出类别可能性【分类】

4.实验和讨论

进行了以下实验,具体,我们可以看论文中的图片

1.训练时,我们使用一个小的脚本去掉了灰度图和颜色变化较小的图片。

2.显示了下自己的染色结果

3.对比了其他人的染色结果

4.让人民对染色结果进行评判

5.副产品-学习到一张图片的风格->应用于新的图片

6.对老照片的染色

7.网络的分类结果也不错

8.使用CIElab空间效果好

9.计算时间满足realtime

论文的限制-对于一个人穿的衣服颜色,没有一个好的办法去限定,比如:下面的例子说明了错误


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

推荐阅读更多精彩内容