SampleNet Matting解读

论文《Learning-based Sampling for Natural Image Matting》将基于采样的方法与当今的深度学习进行结合,在matting任务中取得了很好的效果。本文是我的阅读笔记,解释以及思考该模型的方法。

模型思路

在matting任务中,认为一张图像是由前景和背景线性组合而成,即下述公式:
I_i = \alpha_iF_i+(1-\alpha_i)B_i
很明显,要将前景背景分开需要知道的是公式中的\alpha,在目前已有的基于学习的方法中,基本上都是通过各种方法直接去推断\alpha。然而本文的方法是先推断出“不确定区域”的前景和背景,再由前景和背景推断出相应的alpha。整体的流程图如下

流程图

可以看到这是一个三阶段的方法,首先并且每一阶段都用到了前一阶段的结果。第一阶段输入trimap和原图得到背景,第二阶段输入trimap和原图和推测的背景得到前景,第三阶段输入trimap和原图和推测的背景与前景得到最终的alpha。

下面分别对这三个部分进行介绍。

背景采样

这一步的目的是对trimap对应的“不确定区域”中的背景进行推测。作者认为这与图像修复这一任务很相似,图像修复是利用已知的纹理与其他高层信息对缺失的区域进行“想象”(hallucinate)。这里的“想象”确实是作者介绍图像修复时的用词,并且这个说法在下面会用到。
由于两个任务的相似性,作者直接使用state-of-the-art的图像修复网络[1],进行些许的改动就用在生成背景的任务中。

第一个改动:
一般的图像修复任务的输入为下图所示,待修复区域是完全不可见的。但是在本文任务中,“不确定区域”并不是不可见的,并且该区域的包含了很多背景相关信息,因此为了利用这些信息这里直接输入了原图而不是遮挡后的图。

图像修复的输入示例

第二个改动:
只对“不确定区域”计算损失函数,这在该任务中是理所当然的,因为该任务就是要预测“不确定区域”的背景。损失函数如下
L_B = \frac{1}{|U|}\sum_{i \in U}{|\hat{B_i}-B_i|}
其中\hat{B_i},B_i分别是预测的背景图和真实的背景图,U代表不确定区域。

这里有一个疑问,损失函数中要求有真实的背景图,那么该模型的训练网络就应该只能用合成的图像进行训练(合成的意思是将某个前景与某个背景组合在一起),因为真实图像没有真实的背景图

前景采样

估计出背景之后再要去估计前景,那么问题就变为估计一个合适的背景,使得它与背景的混合是原图所示的样子。因此在这一步有两个约束,即两个损失函数,如下
\begin{align} &L_{fg} = \frac{1}{|U|}\sum_{i \in U}{|\hat{F_i}-F_i|} \\ &L_{comp}= \frac{1}{|U|}\sum_{i \in U}{|\alpha\hat{F_i}+(1-\alpha)\hat{B_i}-I_i|} \end{align}
前一函数是对比真实前景图的L1损失,后一个是与背景图结合之后的L1损失。

alpha matte 生成

其实得到前景图与背景之后,是可以直接计算出alpha的。但这里作者说为了得到更高质量的alpha,所以这里也使用了神经网络(这里作为博主有些不解)。
这里使用的网络是AlphaGAN[2]中的生成网络,损失函数是关于alpha图和最终合成图的L1损失
L_{dm} = \frac{1}{2|U|}\sum_{i \in U}{|\hat{\alpha_i}-\alpha_i|} + \frac{1}{2|U|}\sum_{i \in U}{|\hat{I_i}-I_i|}
但是L1损失函数存在最终生成图会模糊的问题,因此作者又提出了对于梯度图的L1损失如下
L_{grad} = \frac{1}{|U|}\sum_{i \in U}{|\hat{G_i}-G_i|}

引用

[1]. generative image inpainting with contextual attention
[2]. AlphaGAN: Generative adversarial networks for natural image matting

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

推荐阅读更多精彩内容