基于分层置信度传播的光流估计方法①

张子星,文 颖

(华东师范大学 计算机科学技术系,上海 200062)

1 概述

光流估计是计算机视觉的一个基础研究领域,其计算结果被广泛用于物体分割、跟踪、运动场景分类等许多领域[1].自从 Horn 和 Schunck 的工作[2]以来,光流估计领域的研究已经持续了几十年.传统的方法[3–5]通过变分法框架最小化能量函数来求解光流场,这在小位移场景下能获得较精确的结果,但往往难以解决大位移光流问题.尽管Brox[3]采用由粗到细(coarse-tofine)的策略将高分辨率下的大位移转化为低分辨率下的小位移来计算,但对于位移比物体本身尺寸更大的情况仍然难以适应[6].

为了克服变分法框架处理大位移时的缺陷,Brox等人在变分法模型中集成了稀疏的描述子匹配的结果[6,7],用描述子匹配得到的运动场拉动变分法光流的计算.通过将匹配结果与变分法相融合,可以有效地提高变分法光流的精度[8].Deepflow[9]通过非刚性匹配得到半稠密的运动场,进一步提高了描述子对变分法的拉动能力.文献[10]将图像进行分割并分别求取变换矩阵,在单一运动场景和复杂运动场景下都取得了较好的结果,近年来出现的基于 PatchMatch 的方法[11–13]通过随机搜索和邻域块之间的信息传播[14]加快了寻找匹配对的过程,有效提高了匹配精度.

近年来,马尔科夫随机场也被许多学者用来尝试作为寻找图像间匹配点的模型.求解马尔科夫随机场的推断算法如置信度传播在实际应用中能得到较精确的匹配结果,但是存在着计算量太大的问题.Felzenszwalb等人提出了一种高效的循环置信度传播算法[15],在有限次的消息传递迭代以后就能获得较好的结果,但该方法适用于小位移场景,在大位移场景下仍然过于耗时.SPM-BP方法[16]将消息传递算法与PatchMatch的随机搜索相结合,在减小标签空间的同时获得了亚像素精度的位移场.FullFlow[17]提出了一种全局的消息传递优化方法并将消息传递的计算复杂度从二次降为线性,取得了较为精确的匹配结果.将置信度传播应用于光流估计的关键在于在标签空间和位移场精度之间取得平衡.高精度的大位移估计意味着较大的标签空间,这会导致计算时间及内存上的巨大开销.FullFlow虽然提出了改进方法,但在大位移场景下的精确估计仍然比较耗时,SPM-BP虽然利用随机搜索有效缩短了时间,但是精度不稳定,错误的标签空间选取可能导致严重的误差.

综上所述,以往提出的方法存在的不足之处主要表现在光流计算精度有限和处理速度缓慢的问题,为了解决上述问题,我们提出了分层置信度传播算法用以计算精确光流.图1展示了我们模型的主要流程.我们构建了两层马尔科夫随机场:超像素层与像素层.我们首先在超像素层执行置信度传播得到一个大范围但低精度的位移场,然后利用该位移场限定每个像素点可能的位移范围,从而有效减小像素层置信度传播的标签空间,然后在像素层执行置信度传播得到每个点的高精度位移.通过分层的方式,我们有效地提高了利用置信度传播计算光流的效率.为了提高算法的鲁棒性,我们还在置信度传播模型中加入了相邻多帧的图像信息作为参考,对于存在严重遮挡的运动场景,多帧信息能有效减少误匹配的发生.

图1 本文提出的分层置信度传播光流估计算法的流程示意图

2 相关工作

应用于计算机视觉的置信度传播算法的一般框架是将图像中的每一个像素点作为马尔科夫随机场中的一个节点,并用边连接四邻接的像素点.对每个节点而言,每一种可能的取值都是一个标签.对光流估计问题来说,每一个标签都代表一种可能的位移.目标函数一般定义为:

其中,P 是节点构成的集合,fp是赋给节点p的标签,在光流估计问题中,每个候选标签代表一种候选位移.N代表四邻接节点间互相连接的边.Cp(fp)代表数据项,即节点p在标签fp下的代价,S(fp,fq)代表平滑项,用以衡量两个相邻节点p和q的标签差异.

根据Felzenszwalb等人的工作[15],在带环的马尔科夫随机场中最小化能量函数(1)可以通过循环置信度传播来实现.循环置信度传播通过消息迭代传递来更新每个节点的状态,消息迭代更新的公式如下:

其中,mts−p1(fp)表示在t− 1次迭代时节点s向节点p传递的消息.s是p的邻接节点中除了以外的节点.消息mtpq(fq)是一个向量,维度与节点的候选标签数量相同.经过若干次迭代以后,通过如下公式计算置信度向量:

并选择向量bp中的最小值对应的标签作为节点p最后的标签.

当处理存在大位移场景的高分辨率图片时,过大的节点数量和标签空间使得上述框架计算极为耗时.上文提到的FullFlow虽然可以通过置信度传播取得较为精确的结果,但是其计算极为消耗资源,而SPMBP利用随机选取来获取候选标签的方法并不稳定,错误的标签可能造成较大的误差.本文提出的方法不但能有效提升置信度传播的效率,而且可以获得较高精度的结果,在时间消耗和光流估计的精度上取得了较好的平衡.

3 本文方法

定义输入的两帧图像是I1、I2,将I1视作马尔科夫随机场.L代表候选标签的集合,每个标签都是一个候选的二维位移向量.马尔科夫随机场中的每个节点需要赋予一个合适的标签.

定义fp∈L为赋给节点p的标签,fp=(up,vp)代表通过位移(up,vp)将中的节点p(x,y)映射到I2中的p′(x+up,y+vp)处.本文将能量函数定义为:

其中,p是I1中的一个节点,是两个相邻节点,Ccolor和Cdesc分别代表移动前后的颜色差异、描述子差异.平滑项S(fp,fq)用于惩罚相邻节点p与q运动的不一致性.本文模型中的数据项由Ccolor和Cdesc构成,即:

我们利用公式(2)和公式(3)来最小化能量函数(4),为了克服置信度传播在大位移场景下巨大计算量,我们的模型引入了分层结构.

3.1 分层结构

基于上述模型,较直观的方法是将I1中的每个像素点作为一个节点.但是当进行大位移光流估计时,直接应用该模型会导致 L 包含成千上万的标签.较大的节点数量和过大的标签空间使得计算变得过于耗时.为了克服这个问题,我们提出了分层结构来提高计算效率.

本文对于置信度传播过程的改进基于这样的假设:一个物体表面局部区域内的像素点的运动往往是一致的,因此可以将每一小块这样的区域作为一个节点构建马尔科夫随机场.如果能得到每块区域的大致位移,区域内的每个像素点在进行置信度传播时就能将候选位移控制在较小的范围内.为了得到这些小块的区域,我们将I1、I2进行分割,利用SLIC分割算法[18]将原图分割成若干超像素块.相较于规则分割,超像素分割能更好地贴合物体的边缘,保留边缘细节,有利于保持运动场中物体边缘的清晰.我们在超像素图像和原图像上分别构建马尔科夫随机场,在超像素层和像素层两个层面执行循环置信度传播.节点p在超像素层和像素层分别代表一个超像素块和一个像素点.我们首先在超像素层执行置信度传播,为了进一步加快计算,本文方法在超像素层面获取的是一个低精度的位移场,也就是说限定每个节点的候选位移步长为d,即fp∈R2,R={0,d,−d,2d,−2d,···}.本文利用二维的 distance transform算法[15]来迭代地更新消息.为了计算公式(5),需要确定超像素块p在经过位移fp后在I2上的对应块,为此,将超像素p的中心点x沿着fp移动到I2上的x′处,将x′所属的超像素块p′作为p在fp下的对应超像素块,然后计算p与p′之间的颜色与描述子差异.

在完成超像素层的循环置信度传播后,得到了一个基础的位移场.利用这个基础结果来引导像素层的置信度传播,能够有效地减少每个像素点的标签数量.用代表像素层节点p的一个候选位移,注意如前文所述,节点p在像素层代表一个像素点,代表节点p所在的超像素块在超像素层置信度传播中得到的位移,则up=du+usp,vp=dv+vsp,其中,dv∈ [−δ,δ],δ决定了像素层每个节点的候选标签个数,本文在像素层设置步长为1,则每个节点的标签数量为.像素层置信度传播是在超像素位移场基础上的进一步微调.本文利用min-convolution算法[17]来执行像素层的消息迭代更新.由于相邻的节点可能属于不同的超像素块,因而它们可能拥有不同的基础位移,需要对min-convolution的方法作修改以解决这个问题.min-convolution的消息更新的过程被分解为两个求下包络的过程:

其中 是惩罚函数,本文使用L1范数.p和q是两个邻接节点,ϕpq(up,vp)=Cp(fp)+s∈Npqmts−p1(fp).首先根据vp的每一种可能的取值计算公式(7),然后利用(7)的结果根据vq的 每一种取值计算公式(6).若p与q拥有相同的基础位移,则上述计算过程就是计算下包络的过程.当p与q的基础位移不同时,我们将上述求得的下包络函数图像向外延伸以覆盖p与q标签的取值范围.图2显示了up与uq的基础位移存在1个像素的相对偏移时的情况,对于公式(7),当vp值固定,先假设up与拥有相同的基础位移,计算函数的下包络,即图中加粗黑线部分,然后再将下包络向外延伸1个单位以覆盖uq的取值范围.

图2 min-convolution 算法求取下包络

3.2 数据项定义

在像素层,每个像素点都对应MRF中的一个节点,由于节点数量多,如果计算每个节点在每一种候选位移下对应点间的SIFT特征欧式距离将使得像素层的数据项计算变得极为耗时.为了有效地减少计算时间,本文方法在像素层上使用Census变换[20]提取每个点的局部特征,然后利用汉明距离计算运动前后的Cdesc.相较于 SIFT 特征,Census变换求取方便且特征维数低,通过位运算实现的汉明距离也比欧式距离计算时间短.SPM-BP方法[16]同样利用Census变换来计算数据项,其工作验证了Census变换提取的特征作为数据项可以在较小的时间开销下获得稳定的性能.

3.3 多帧信息

为了解决由于大位移和物体运动到画面以外而导致的遮挡问题,进一步提高结果精度,本文提出的模型考虑了前后多帧图像的信息,由于物体的运动具有惯性,同一物体在相邻多帧图像上的运动往往具有一致性,多帧信息在一定程度上可以弥补前后两帧图像中部分区域丢失的信息[21].我们在计算数据项Cp(fp)时沿着运动方向向前和向后分别映射到相邻多帧图像上,分别计算数据项然后取最小值作为最终的数据项结果,以此提高模型的鲁棒性.

图3显示了使用多帧信息的示意图.It−1、、It+1和It+2是连续的四幅输入图像,我们要计算节点pt在It与It+1之间的光流,(u,v)是一个候选标签,则我们将pt沿移动到It−1上的pt−1处,沿移动到It+1的pt+1处,沿(2u,2v)移动到It+2的pt+2处.根据公式(5)分别计算pt与pt−1、pt+1和pt+2之间的颜色与特征差异,得到,取最小值作为pt在标签下的数据项,即:

3.4 后处理

在完成分层置信度传播以后,通过前后一致性检查来去除错误的位移,得到一个半稠密的光流场.之后我们采用EpicFlow[22]的插值方法,通过现有的半稠密光流场插值填充空白区域的位移,得到最终的亚像素精度的光流场结果.

图3 多帧信息示意图

4 实验及结果分析

实验在常用的光流数据集MPI Sintel数据集[23]上对本文方法进行评估.MPI Sintel数据集提供了包含大位移、运动模糊和大面积遮挡等场景的动画视频序列.数据集包含23个训练集图像序列,12个测试集图像序列,每个图像序列都包含清晰版和加入了运动模糊、气流变化的最终版本,由于最终版本图像更接近于真实场景图像,实验在最终版本的图像上评估算法性能.实验平台为 4 核 Intel i5-4590 3.30 GHz CPU 和 8 GB RAM的个人台式计算机.

4.1 参数分析

我们随机选取MPI Sintel数据集中10%的训练集数据来进行模型参数的评估实验.在公式(4)中,系数α和 β分别控制着颜色和描述子的权重.在3.1节中,步长d影响超像素层置信度传播的标签空间大小,δ决定了像素层标签空间的大小.因为颜色信息和SIFT描述子在寻找点匹配中的作用已被许多研究者论证,我们主要关注 δ和步长d对光流估计结果的影响.实验采用终点误差(EndPoint Error,EPE)来衡量得到的光流场与真值之间的误差,即求每个点的位移向量与真值的欧式距离作为该点的误差,然后求所有点的误差的平均值作为最终的误差.在超像素层,参数设置为 α=1,β=250,在像素层,参数设置为α=0,β= 3.

图4显示了不同d取值对超像素层置信度传播结果的影响,d的取值越大,则候选位移选取时的步长越大,同样的位移范围下标签空间越小.此处的误差是在超像素层完成置信度传播后计算得到的.通过图中的曲线可以看出,在超像素层的置信度传播运算并不需要高精度的候选位移,d= 4、d=5都取得了较好的结果.因为d的取值越大,节点的候选标签空间越小,消息更新的速度更快,在后续实验中设置参数d=5.图5显示了像素点位移范围 δ的不同取值对误差的影响,由于有超像素层的位移对像素点的位移范围进行了限制,在像素层进行置信度传播时,取较小的值就可以得到比较精确的结果,扩大像素点标签空间并不会进一步提升光流估计的精度.通过该实验可知分层结构的确可以有效地减小像素点的标签空间,提高置信度传播的执行效率.在后续试验中设置参数 δ=10.

图4 d的不同取值对结果误差的影响

图5 δ的不同取值对结果误差的影响

4.2 MPI Sintel实验结果

实验模型参数设置为:超像素层上 α=1,β= 250,像素层上 α=0,β= 3,δ= 10.本文模型对原图像进行1/2降采样以后执行分层置信度传播,在超像素层执行2轮消息更新迭代,在像素层执行3轮消息更新迭代.表 1 和表 2 展示了实验结果.“EPE All”表示整幅图像上所有点的位移的误差,“EPE Noc”表示在求取光流的两幅图像上都出现的区域的位移误差.“EPE Occ”表示在第一幅图像中可见而在第二副图像中被遮挡的区域的位移误差.表1显示的是采用了置信度传播的相关方法的实验结果的比较.FullFlow方法效果略优于我们的结果,但是由于使用了全局优化的方法,根据文献中报告的结果,他们的时间开销将近240 s,且是在工作站上运行,时间及内存开销远大于我们的方法.而我们的方法在普通8 GB内存的台式计算机上的运行时间为45 s左右,说明本文的模型能有效提高置信度传播的执行效率.本文模型的时间消耗与SPM-BP接近,但是获得的光流估计的精度更高,说明本文的模型在效率和精度上取得了较好的平衡.HCOF[21]在模型中也加入的多帧信息,但是无论是在处理效率和结果精度上都低于本文方法.

表1 本文方法与相关方法在 Sintel上的结果

表2将本文模型的实验结果与目前前沿的一些方法作了比较.从表中可以看到本文方法的结果优于大部分方法.我们的模型使用了EpicFlow[22]的插值方法,而其最终结果优于EpicFlow,这说明本文提出的模型在精确匹配对应点上比EpicFlow表现更好.从表中结果看出本文方法在精度上具有一定的竞争力.

图6显示了部分实验结果可视化图像的示例,图中将本文提出的方法与EpicFlow与SPM-BP的结果进行了对比.每一列的第一行和第二行为输入图像,第三行为EpicFlow的结果,第四行为SPM-BP的结果,第五行是本文方法的结果,最后一行为真值.图中方框标出的区域显示了本文方法的优势.从第一列我们可以看到EpicFlow的结果在图像右上角出现了明显的错误,SPM-BP则在图像的右下角区域出现明显的错误,本文的方法与真值更接近.第二列的EpicFlow的结果没有捕捉到人物手臂的运动,SPM-BP的结果中手臂的运动方向与真值相比出现了明显的偏差,本文方法捕捉到了手臂及身后木棍的运动.第三列中,EpicFlow和SPM-BP都没有捕捉到左上角手部的运动,而本文提出的方法则成功地捕捉到了手部的运动.第四列中,EpicFlow和SPM-BP在图像左上方的运动方向都出现了明显的错误.

表2 本文方法与部分常用方法在 Sintel上的结果

图6 本文方法与相关方法的可视化结果比较

5 结束语

本文提出了一种分层执行置信度传播的方法来进行光流估计,能在较短的时间内获得高精度的大位移光流场.为了克服将置信度传播应用于大位移场景时巨大的内存与时间开销,本文方法首先在超像素图像上执行置信度传播,将每个超像素块作为节点并扩大候选位移的步长以提高运算效率,之后利用得到的超像素位移场来引导像素层的置信度传播,将每个像素节点的标签空间限制在一个较小的范围内,有效提高了像素层置信度传播的效率.本文模型仅使用了常见的SIFT描述子及Census变换就在大位移场景下取得了较为精确的光流估计结果,这说明本文提出的模型能有效地寻找精确的点对应,适用于高精度的大位移光流估计.

本文模型中使用了SLIC超像素分割算法和稠密SIFT描述子,在之后的工作中,设计更为优秀的分割算法和描述子,将进一步提升模型的性能.此外,利用GPU并行实现本文算法也将进一步提升分层置信度传播的执行效率.虽然本文提出的模型有效地提高了置信度传播的执行效率,但其执行速度距离实时计算仍有差距,且计算精度仍有提升空间.我们相信通过后续的工作改进,置信度传播算法将成为光流估计的一个有力工具.