相关性矩阵及特征匹配

  • PSGAN: Pose and Expression Robust Spatial-Aware GAN for Customizable Makeup Transfer
  • Cross-domain Correspondence Learning for Exemplar-based Image Translation
  • Convolutional neural network architecture for geometric matching

相关性矩阵(correlation matrix)是一个简单的范式,矩阵中的每个元素 Cij 用来表示一个特征图(feature map)中的第 i 个局部特征向量和另一个特征图中的第 j 个局部特征向量之间的相关性,通常是由两个局部特征向量的点积来表示(包含一个 L2 归一化)。两个尺寸为 H * W * d 的特征图的相关性矩阵的尺寸即为(H * W)*(H * W)。相关性是衡量特征匹配程度的依据,特征则会根据不同的任务有着不同的表示,通常是基于形状、颜色、纹理的语义特征。

语义特征相关性

语义特征与结构位置信息结合

PSGAN 在妆容迁移(makeup transfer)时会对人脸中相应位置进行妆容迁移,为此 PSGAN 使用了 Attentive Matrix 来衡量两个人脸局部位置之间的语义信息和人脸空间结构位置的相似性,并引入了 Attentive Makeup Morphing(AMM)模块来围绕 Attentive Matrix 进行妆容的迁移。

PSGAN的AMM模块

在计算 Attentive Matrix 时,使用到的不仅是局部的特征向量,还有基于当前局部位置到以68个人脸关键点(landmark)作为锚点(anchor)的坐标位置的偏差信息。特征图与坐标偏差合并(concat)成一个张量进行 Attentive Matrix 的计算。

由Attentive Matrix得到的Attention Map

从生成的 Attention Map 上看,相应的人脸位置区域有着很好的响应,对关注对应的人脸位置也使得 PSGAN 相对于其他妆容迁移算法对姿态变化更为鲁棒。

跨域语义对应

CoCosNet 是基于范例(exemplar-based)的图像翻译(image translation)技术,网络在接受输入域图片的同时,也接受一张目标域的范例图片,该范例图片与输入图有相似的语义且具有用户期望的目标风格。CoCosNet 提出建立两个域图片之间的密集语义对应(dense correspondence),进而依赖这样的对应,定位输入图在范例中相应位置的颜色、纹理信息,使得生成图片风格精细匹配范例中相同语义的物体。

跨域语义对应

于是,CoCosNet 的生成任务归结于,如何精准地找到输入及范例之间的语义对应。这里有两大挑战:一是输入图和范例来自不同的域,它们的图像块(image patch)有着巨大的图片域差异(domain gap);另一方面,不同域图片之间的语义对应的标注信息很难取得,并拿来作为网络训练的监督信息。如上图,建立自然人脸与油画人像之间的密集对应,事实上是困难的,而这样的跨域语义对应(cross-domain semantic correspondence)问题也是首次被提出。

CoCosNet结构图

CoCosNet 联合训练跨域对应和图像生成,其中跨域对应通过弱监督学习(weakly supervised learning)的方式建立。整个网络由两个子网络构成。第一部分为跨域对应(cross-domain correspondence)子网络,其用于找到来自于不同的域的输入图 x_A 与范例图 y_B 之间的密集对应。两个域的特征提取器 (F_(A→B) 和 F_(B→A))利用训练集中 x_A 和它在目标域的配对 x_B 来训练,使它们提取到的特征可以适应到一个对齐的隐空间,这里的域自适应损失函数定义为:

之后,再用处于共同域上的特征进行相关性矩阵的计算,进行语义匹配。相关性矩阵中的元素为对应特征的向量的余弦相似度。

几何特征匹配

在 Convolutional neural network architecture for geometric matching 这篇论文中提出了一个用于几何匹配的 CNN 架构


图像几何匹配

几何匹配的过程是将两张图像A、B输入进CNN模型后得到一个变换参数 theta,然后图A根据参数 theta 进行变换后即和图B的几何特征相匹。

模型结构图

在该 CNN 模型中,ImageA 和 ImageB 两张图像输入到同一个特征提取的 CNN 里得到 feature map A 和 feature map B。之后两个 feature map 进行相关性匹配(matching)得到 feature map AB,再经过一个 CNN 回归得到用以几何匹配的仿射变换参数 theta。
输入图像经过CNN提取特征后得到的 feature map 的尺寸是 w * h * d,相当于是由 w * h 个 d 维的局部特征向量组成的。

相关性匹配过程

在相关性匹配时,两个 feature 中的任意两个局部特征向量的点积用来表示这两个局部特征之间的相关性。最后可以看作是产生了一个(w * h)*(w * h)的相关性矩阵 cAB,cAB实际表示为一个 w * h *(w * h)的三维张量相关性图。cAB 中某个位置的向量中的第 k 个元素表示的是 feature map B 中对应位置的向量和 feature map A 中的第 k 个向量的相似度。

归一化的相关性图

这个相关性图(correlation map)中的其中一个向量表示为 feature map A 中的所有特征向量与 feature map B 中的某个向量的相关程度。最后,相关性图中的每个向量还会进行 L2 归一化得到归一化的相关性图,即为 feature map AB。

用以回归 theta 的CNN结构

feature map AB 再经过一个回归 CNN 得到变换参数 theta。

两阶段匹配示意图

考虑到复杂的几何对齐可能会难以一步到位的实现,论文中还提出了一个两阶段的匹配流程:先在第一个阶段使用仿射变换进行粗略匹配,然后在此基础上,第二阶段再使用 tps 变化进行一个精细的匹配。

两阶段中 theta 参数的维度

两阶段中回归 CNN 的输出维度需要根据变换的自由度进行调整。

阶段一示意

简单的仿射变换。

阶段二示意

复杂的 tps 变换。

由于缺少公开数据,训练数据将从公开数据中生成。

从公开数据集中生成图像对
从公开数据集中生成图像对

生成随机的 theta GT
损失函数定义为,在gird上衡量

损失函数定义

在gird上衡量

训练

该自监督模型作为weakalign的预训练模型。

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