APP下载

基于自适应符号函数的主动轮廓模型∗

2019-10-26翁桂荣何志勇

软件学报 2019年12期
关键词:轮廓线轮廓边缘

翁桂荣,何志勇

(苏州大学 机电工程学院,江苏 苏州 215021)

图像分割技术在计算机视觉、场景识别、医学图像分析、移动目标跟踪等领域具有非常重要的意义.基于曲线演化模型和偏微分方程等的分割方法在理论上已经取得新的突破,基于水平集方法的主动轮廓模型已成为图像分割的热点[1−4].1988年,Osher等人[5]在分析流体力学问题时,创造性地提出水平集方法,主要思想是:把平面上的闭合曲线的演化过程上升为高一维的函数与水平面交集的演变过程,即零水平集的变化过程,通过求解零水平集函数来获得运动的边界,用于界面追踪和形状建模.但是传统的水平集演化由于采用逆向差分的数值解法,计算效率不高.Chop[6]首先提出了窄带方法,同时给出了重新初始化的概念,就是在数值演化过程中只更新零水平集附近带状区域里的水平集数值,计算量就大幅减少.但多次重新初始化,计算量也随之增加,快速地生成符号距离函数,对于提高水平集方法的速度至关重要.

1989年,Mumford和Shah提出了一个经典的基于能量最小化的分割模型,即Mumford-Shah(M-S)模型[7].该模型的能量泛函由保真项、平滑项和长度约束项组成.所谓保真项显示原图像与之分割结果的相似性;平滑项的作用是使得分割结果光滑;长度约束项用于约束模型的曲线长度.其中,保真项和平滑项利用区域信息特征,长度约束项基于边缘梯度信息.

边缘型的方法适用于目标边缘清楚的图像,对于没有明显梯度变化的边缘则逼近效果不理想.最早的水平集模型是由Malladiet[8]和Caselles[9]等人提出的几何活动轮廓模型,该模型基于曲线的平均曲率运动和图像的梯度信息,构建了基于水平集方法的速度函数.Siddiqis[10]提出在测地线活动轮廓模型中增加面积项功能,目的是提高变形曲线跨越轮廓线中较小缝隙的能力.

2001年,Chan-Vese提出了基于简化M-S能量泛函的CV模型[11],该模型不依赖于图像的梯度信息,对噪声不敏感,较为适用于分割梯度无意义或边缘模糊的图像.Kimmel在模型[12]中集成了边缘的方向性信息,并通过最小化边缘和图像梯度间夹角的方式对测地线活动轮廓模型进行了改良.

保证演化方程处于稳定状态是几何主动轮廓模型初始化的标准方法,理想状态下使得演化方程稳定的状态解是一个符号距离函数.几何主动轮廓模型的问题主要有两个,一是通过重新初始化得到规则的水平集函数并不能保证零水平集函数可以运动至目标边界,二是在实际操作过程中如何选择重新初始化的时间.

Li针对上述问题,提出了解决重新初始化的方法[13],即距离规则的水平集(DRLSE)方法,通过正规项的作用,保证水平集函数是符号距离函数.其构造初始超平面也非常简洁,提高了分割速度.为了进一步提高模型的运行速度,并克服单阱势函数的缺点,Li构建了一个双阱函数,发展出快速距离规则水平集[14]以及区域拟合驱动的主动轮廓模型(RSF)[15,16],利用势函数的特点,主动轮廓模型在演化过程中就不需要重新初始化.

但是,DRLSE,RSF模型未能解决另一个核心问题,即对初始轮廓敏感.此外,DRLSE模型还存在下述不稳定的问题,如抗噪声能力差,收敛速度较慢,在演化过程中容易陷入虚假边界,会从弱边缘处泄露[17].

Wang提出了一种基于边缘的自适应水平集演化方程[18],将面积项系数修改为可变符号函数,达到改善对初始轮廓敏感的目的,克服了DRLSE单向运动的缺点.该模型具有较好的操作性,分割速度大幅提高,特别是在边界较清晰或背景噪声低的情况下,能达到很好的分割效果.但是该模型有抗噪声能力较差、容易陷入虚假边界以及从弱边缘处泄露等不稳定问题[19,20].

本文提出了一种致力于提高初始轮廓鲁棒性的主动轮廓模型,设计了一个自适应符号函数结合自适应边缘指示函数的能量泛函,使得主动轮廓演化能根据自适应符号函数的方向从初始轮廓开始自动进行膨胀及收缩.本文模型提高了对初始轮廓的鲁棒性,同时解决了主动轮廓模型收敛速度慢以及在弱边缘处分割错误的问题.此外,为了演化的稳定性,还改进了距离规则项.

1 经典的主动轮廓模型

1.1 DRLSE模型

设置一个灰度图像I:Ω→R,其图像范围为Ω⊂R2.

距离规则水平集[14]模型在传统的模型上添加了距离规则功能,其主要作用是解决水平集函数与符号距离函数之间的偏差,使得水平集函数在迭代过程中不再考虑重新初始化的问题.该模型的能量泛函定义为

公式(1)中,p(⋅)为势阱函数,∇为梯度算子,φ为水平集函数,δε(⋅)是Dirac函数,Hε(⋅)是Heaviside函数,g(⋅)为边缘指示函数,μ,λ,α为常数.

其中,c0为正值常数,R0代表图像区域中的某块区域.

DRLSE模型为水平集方法提供了新的研究思路,其存在如下问题.

1) 用于加快零水平集演化速率的面积项为单一值(正或负值),在能量泛函进行最小化中,只能是从正(负)趋向0的单向选择,即没有自我调整能力;

2) 面积项及长度项高度依赖边缘指示函数,而边缘指示函数由图像的梯度构建,当梯度值很大时,其函数值越接近0.经过高斯滤波后的目标边界会变得模糊变宽,导致目标边界区域变宽.当目标之间的距离很近时,目标边界可能会因高斯平滑而相互连接,导致分割失败;

3) 在实验中常数α需要人为设置,其取值大小对分割的结果影响很大.α过小会使演化速度降低,从而陷入虚假边界,导致演化停止;α过大,会使边界处的分割出现泄漏,导致分割失败;

4) DRLSE中的双阱势dp(s)是一个三段式函数,当分母s为0时,无法计算;该双阱势函数在零点附近演化速度接近为1,为演化最大值,使得水平集迅速向前进行演化,从而让图像里的弱边界s急剧减小至0,导致演化曲线侵入被分割目标内部,分割可能失败[17].另外,在|∇φ|=1附近时,其斜率小,演化速度慢.

1.2 RSF模型

Li等人提出了一种区域拟合驱动的主动轮廓(RSF)模型[16],也称局部二值拟合(LBF)模型[15].RSF模型可以在较小邻域内精确找出细小目标,并加快大目标在大邻域处的演化速度.RSF模型的核心思想是利用图像局部区域像素的灰度信息,用局部二值拟合能量项代替全局二值拟合能量项,通过调节核函数的方差σ控制中心像素的邻域.根据轮廓定义了一个数据拟合能量公式:

其中,Kσ(⋅)为Gaussian函数,y−x代表x和y像素点的距离,H(⋅)是Heaviside函数,φ(⋅)是Lipschitz函数,拟合值f1(⋅),f2(⋅)局部地逼近轮廓外、内图像灰度的平均值,I(⋅)是灰度值,λ1,λ2为常数.

公式(8)中,H(⋅)不是一个连续函数,用H(⋅)代替[11].RSF模型的梯度下降流方程如下:

其中,μ,ν为常数,e1(⋅),e2(⋅)表达如下:

其梯度下降流方程中:第1项表示拟合部分;第2项对零水平轮廓有缩短长度或者光滑的效果,对于维持轮廓的规则性是必不可少的;第3项是内部能量项,由单势阱距离规则组成,主要作用是矫正水平集与符号距离间的误差,避免曲线在演化过程中重复初始化.

RSF模型对于灰度不均匀的图像分割效果显著,而对于亮度不均匀的图像却无法进行分割.另外,如果初始轮廓设置不当,RSF模型往往会陷入局部极小.

1.3 自适应水平集演化模型

Wang和He[18]提出了轮廓提取的自适应水平集演化模型(ALSE)解决初始轮廓敏感性的方法,其水平集函数的偏微分方程即其梯度下降流方程如下[18]:

分析文献[18]我们可以发现,ALSE模型引入由图像信息调整的权系数v(I)代替常值权系数α.在DRLSE模型的基础上增加了一个符号函数,符号函数的方向由I(x,y)与轮廓线外、内图像的平均值的均值差值决定,克服了DRLSE模型中面积项为单一方向运动的问题.因此,偏微分方程能根据初始轮廓内外的图像灰度信息调整运动方向,改善模型对初始轮廓的鲁棒性.

该模型具有较好的操作性,分割速度大幅提高,特别是在边界清晰或背景噪声低的情况下,能发挥很好的分割效果.但是该模型仅仅修正了DRLSE模型中的方向,而DRLSE模型中的抗噪声能力差、容易陷入虚假边界以及从弱边缘处泄露等不稳定问题,没有得到进一步解决.

2 自适应符号函数的主动轮廓模型

针对DRLSE,RSF初始轮廓位置敏感性,本文设计了一个自适应符号函数用于提高初始轮廓的鲁棒性.为了进一步克服ALSE及上述DRLSE的几个缺点,改进了边缘指示函数,加快了演化速度.此外,还修正了距离正则项,使得演化更稳定.其梯度下降流方程为

2.1 自适应边缘指示函数分析

公式(18)中,图像的标准差(std2)反映了图像像素值与均值的离散程度[21].公式(19)中β的大小表达了目标相距平均值的远近关系.

公式(19)能根据目标图像信息自适应调整gβ函数的曲线.公式(18)中的β如同公式(6)中α的作用:当图像目标相距平均值较远时,图像的方差S较大,β,gβ对应增大,面积项能量随之提高,可以使演化速度变快,并解决了分割时演化曲线进入虚假边界的问题;图像的方差S较小时,边缘处会模糊,β,gβ随之减少,使得面积项能量降低,降低演化速度且不会出现边界泄漏现象.图1所示为两种边缘指示函数的对比图(两种边缘指示函数的对比:蓝色线为DRLSE方法;红色为本文方法,共3条:β=0.75,β=1.0,β=1.25),自适应边缘指示函数对远离边界(梯度值小)的区域给予较大的值,此区域由边界阈值β自适应根据图像特征决定,有利于加快演化速度,同时避免陷入虚假边界中;接近边界(梯度值大)的区域给予较小的值,当零水平集运动到目标边缘时,演化速度迅速减少到0,避免出现边界泄漏现象.该函数能够使零水平集保持较好的光滑性,从而加快演化速度.β又叫边界阈值,是调节gβ函数的重要参数.

Fig.1 Adaptive boundary indicator function图1 自适应边缘指示函数

为验证本文提出的自适应边缘指示函数的有效性,根据DRLSE模型,仅改变gβ函数,得到如图2(d)~图2(f)中所示的分割结果,其中,图2(a)~图2(c)是DRLSE分割结果(图2(a)~图2(c)为DRLSE中的边缘指示函数方法,图2(d)~图2(f)为自适应边缘指示函数方法,绿线为初始轮廓,青线为迭代过程轮廓线,红线为分割结果).

Fig.2 Segmentation results with two boundary indicator functions图2 两种边缘指示函数方法对比分割图

每隔10次迭代计算一次分割面积,当前后两次面积相对误差小于1%时,自动停止迭代,N为迭代次数.DRLSE实验参数:Δt=1,μ=0.2,λ=5,ε=1.0,α=1.5,c0=2,σ=1.5.实验结果表明:本文自适应边缘指示函数模型在保持较快演化速度的同时,可以有效避免陷入虚假边界中.

2.2 双阱势演化速度dp(s)

DRLSE的dp(s)必须考虑s=0的特殊情况,且在零点附近演化速度接近为1,达到演化最大值.如图3所示,此情况下水平集迅速向前演化,有可能导致零水平集进入目标内部,而本文的dp(s)降低了在零阱势处的演化速度,提高了对弱边界的鲁棒性.同时,从图3看出:本文的双阱势在s=1的邻域内比DRLSE的双阱势更加陡峭,这就使得保持符号距离函数的作用更加灵敏.

Fig.3 Comparison between theevolution speedof two double-well evolution speed functions图3 两种双阱势函数演化速度的对比

对比实验条件相同,参数同上,用DRLSE的方法仅改变双阱势函数,进行分割比较,结果如图4所示(第1排为原图,第2排为DRLSE的双阱势函数,第3排为本文的双阱势函数;绿线为初始轮廓,青线为迭代过程轮廓线,红线为分割结果).

Fig.4 Segmentation results withtwo double-well evolution speed functions图4 两种双阱势函数演化对比效果图

第1列为了说明两种双阱势函数的穿透边界的情况,将α从1.5增加到2.5,加快演化速度,可以看到,DRLSE模型出现了边界泄露的情况,而运用了本文双阱势函数的模型则得到了理想的结果;第2列说明在正常演化速度(α=1.5)的情况下,避免了目标遗漏;第3列说明本文方法的dp(s)可以加快演化速度.以上结果说明,本文的双阱势函数对于曲线的演化有较好的稳定性.

2.3 自适应符号函数分析

DRLSE模型及相关论文中涉及的模型,在初始轮廓线处的曲线迭代过程具有单向性[20,21].α>0时为收缩运动,α<0时为膨胀运动,直至梯度下降流方程为0时停止.因此,初始曲线必须完全设置在目标内部或目标外部.为了克服DRLSE模型对初始轮廓较为敏感的缺点,本文增加一个图像自适应符号函数ψ(Iσ,c1,c2),使模型能根据图像目标自适应改变水平集的运动方向.(c1+c2)/2为轮廓线内、外图像均值的平均值,偏黑区域< (c1+c2)/2<偏白区域,Iσ−(c1+c2)/2表示偏黑的区域为负值,而偏白的区域为正值,这就产生了正负号,其绝对值大小反映了偏离均值的强度,但要对此强度值进行进一步处理以便加入模型中.如图5所示,arctan((Iσ−(c1+c2)/2)/τ)函数平滑了各像素点偏离均值的强度,限定了最大幅度,使目标边界(零点)两侧的符号相反,从而让零水平集可以自适应地选择演化方向,当轮廓线嵌入目标内时,轮廓线开始进行膨胀运动;当轮廓线在目标外时,轮廓线开始进行收缩运动.不仅如此,arctan((Iσ−(c1+c2)/2)/τ)在边界(零点)附近具有很强的灵敏度,可以较精确地表达边界的能量变化.在远离边界的区域,自适应符号函数绝对值较大,可以加快演化速度;在接近边界的区域,自适应符号函数快速下降,演化速度迅速降低,避免出现边界泄漏现象.其中,τ(tau)可以调整强度大小.

Fig.5 Adaptive sign function图5 自适应符号函数

3 实验分析

自适应符号函数的主动轮廓模型算法如下.

1) 参数常规设置如下:α=15,Δt=5,μ=0.008,ε=1.0,λ=5,σ=1.5,c0=2,τ=35.内迭代次数=5,外迭代次数=80.另外,如果目标太小,Gaussian滤波会让边界模糊,就可能丢失细小的目标,可以取σ=0.7;

2) 实验图像进Gaussian滤波后,计算其边缘指示函数gβ(I)、自适应符号系数ψ(I)及β(I);

3) 设置初始轮廓线参数;

4) 根据公式(16)进行参数计算;

5) 每迭代5次,计算水平集函数面积变化,如果变化小于1%,则停止迭代并输出分割参数;

6) 检测是否完成总迭代次数?是,输出分割参数.否则,返回到步骤4);

第5)步仅仅提供一个客观的评判标准,并非必要的步骤.

3.1 模型中各重要函数的实验分析

公式(16)中,本文修改了面积项的能量公式,面积项的能量ψ(Iσ,c1,c2)⋅gβ的最小值由当中的两个函数共同决定.当演化轮廓曲线C运动到边界区域时,自适应符号函数ψ(Iσ,c1,c2)到达最小值.边缘指示函数gβ与ψ(Iσ,c1,c2)同样具有引导主动轮廓曲线向同质边界运动的功能,在边界处也达到最小值,提高了模型的鲁棒性.同时,由于ψ(Iσ,c1,c2)⋅gβ具有不同质地区域符号相反,因此能改变运动方向,很好地提高了模型对初始轮廓的鲁棒性.图6用图解的方式表达了重要函数的演化.图6(a)为分析原图像及初始轮廓线.图6(b)为自适应边缘指示函数效果图.图6(c)为ψ(Iσ,c1,c2)在初始轮廓下的效果图,白色区域为正值,黑色区域为负值,红色线为正负区域的边界.图6(d)为ψ(Iσ,c1,c2)⋅gβ在初始轮廓下的效果图,在目标边缘趋近于0,两侧符号相反.此系数在目标边缘附近比图6(c)更平滑,红色线为正负区域的边界.图6(e)为在初始轮廓下,第1次迭代后的面积项β(I)ψ(Iσ,c1,c2)gβδε(φ)的能量图,红色线为正负区域能量的边界,绿线为初始轮廓线.初始轮廓线切割目标区(负能量区域)的线段向外膨胀,以求能量趋于零方向,初始轮廓线在目标区(负能量区域)外的线段为正能量,因此,此部分线段就做收缩运动,向能量减小的方向运动,以求能量趋于零方向.另外,由于δε(φ)函数的因素,仅仅考虑了初始轮廓线附近的能量,远离轮廓曲线的能量就大幅减弱.图6(f)为第1次迭代分割结果.

Fig.6 The segmentation results withparameters in the iteration图6 迭代前后各参数的效果图

图7为几次迭代前后,轮廓曲线变化的效果图(第1行为第2次迭代前后的效果图;第2行为第5次迭代前后的效果图;第3行为最后一次(25)迭代前后的效果图).

Fig.7 The segmentation process of the proposed model图7 迭代效果图

图7(a)为自适应符号函数ψ(Iσ,c1,c2)迭代前后的正负区域的变化,绿色线为迭代前的,红色线为迭代后;图7(b)为第n次迭代后的面积项能量图,红色线为正负区域能量的边界,绿色线为前一次迭代结果,同时作为下次迭代的初始轮廓线;图7(c)为迭代分割结果,绿色线为初始轮廓,红色线为分割结果.

3.2 初始轮廓敏感性实验

本文模型的首要目的就是要提高水平集模型对初始轮廓的鲁棒性,如图8所示,选取4副灰度图像,每个灰度图像设置3个不同形状或位置的初始轮廓线,分别得到演化过程及分割结果(绿线为初始轮廓,青色线为迭代过程的轮廓线(子图),红线为分割结果,N为迭代次数).实验说明:本文模型可以选取不同的初始轮廓线,可以从不同的位置嵌入或包含目标边界,对于有空洞的目标,也能正确有效地分割.

Fig.8 The evolution process and segmentation results with the proposed model under different initial contours图8 使用本文模型,选取不同的初始轮廓、所得到的演化过程及分割结果

3.3 评价实验

为说明本文模型的速度及精确性,引入一个标准图库,分割评价数据库[24].本文模型及ALSE模型采用固定的边框初始轮廓.由于RSF模型的初始轮廓必须要精心选取才能正确分割,否则分割就会失败,考虑到DRLSE模型的初始轮廓要包含目标且初始轮廓线的位置会严重影响速度,故本节不参与比较.图9所示是采用分割评价数据库图像选取固定的初始轮廓(95:M-70,100:N-80,M及N为图像的大小)及分割结果.绿线为初始轮廓,红线为分割结果.第1列为评价数据库原图;第2列为数据库分割图;第3列为RSF模型分割图;第4列为ALSE模型分割图;第5列为本文模型分割图:第4列、第5列的初始轮廓为95:M-70,100:N-80(M及N为图像的大小).

Fig.9 The segmentation results with three different models图9 3种模型评价数据库的分割结果

为了更加准确地将本模型的分割结果与专家分割进行定量[25]比较,本文采用基于面积的dice similarity coefficient(DSC)[23]来度量模型算法的性能,定义如下:

其中,N1和N2分别代表得到的分割和真实的分割区域,S(Ni)是指分割的区域面积.DSC的值越接近1,其分割效果越好.表1显示了分别用RSF模型、ALSE模型及本文模型对分割评价数据库[25]中的图像进行分割所得结果的DSC计算以及分割时间记录.表1数据表明:3种模型的准确度均较高且基本相同,但在计算时间上有很大差别.本文模型速度上远远优于RSF模型,也优于ALSE模型.

Table 1 The number of time spent and DSC in Fig.9 under the same initialcontour (s)表1 比较RSF、ALSE及本文模型的分割图像相似系数值及分割时间(DSC/Time(s)) (秒)

3.4 实验比较

为了说明本文模型对亮度不均及弱边界图像的分割效果,采用固定初始轮廓线(均采用距边框14个像素),选取了6副图像对DRLSE,RSF,ALSE及本文模型进行比较,如图10所示(第1列:原图;第2列:DRLSE模型;第3列:RSF模型;第4列:ALSE模型;第5列:本文模型.绿线为初始轮廓(均采用距离边框14个像素为初始轮廓),青色线表示其迭代过程中轮廓线的演化(子图),红线为分割结果,N为迭代次数).

Fig.10 Segmentation results with four models under the same initial contour图10 4种模型在相同初始轮廓下的分割效果图

Fig.10 Segmentation results with four models under the same initial contour (Continued)图10 4种模型在相同初始轮廓下的分割效果图(续)

表2数据及图10的实验结果表明:DRLSE模型分割速度慢,如果初始轮廓与目标交叉,分割就失败,但对弱边界图像的处理较好;RSF模型对初始轮廓及亮度不均的图像敏感;ALSE模型分割速度比较快,提高了初始轮廓的鲁棒性,对亮度不均及弱边界的图像比较敏感;本文模型分割速度较快,同时提高了对初始轮廓、亮度不均及弱边界的图像的鲁棒性.

Table 2 The number of time spent with four models (s)表2 DRLSE模型、RSF模型、ALSE模型及本文模型的分割速度 (秒)

为了进一步分析本文模型对亮度不均及弱边界图像的分割效果的鲁棒性,如图11所示,将本文模型与Canny,Sobel及Laplacian of a Gaussian这3种经典的边缘检测模型进行分割比较(第1列:原图;第2列:Canny模型,参数T=0.3,sigma=1.5;第3列Sobel模型,采用系统默认值;第4列:Laplacian of a Gaussian模型,参数T=0.3,sigma=2.8;第5列:本文模型.红色线为分割结果).图11表明:本文模型的分割结果具有较高的精确且光滑的曲线及较好的鲁棒性,其分割效果优于其他几种经典模型.

Fig.11 Contrast experiments results of the proposed model and three classical edge-based models图11 本文模型与3种经典的边缘检测模型的分割效果图

Fig.11 Contrast experiments results of the proposed model and three classical edge-based models (Continued)图11 本文模型与3种经典的边缘检测模型的分割效果图(续)

综上所述可以发现:本模型采用自适应符号函数ψ(I),通过计算轮廓线内、外的灰度平均值,从而大致计算出目标区域,在迭代后不断改善目标区域.函数在目标区域边缘趋近于0,两侧符号相反,使得面积项能量有正负之值.因此,构建的面积项自适应地实现收缩及膨胀,控制零水平集曲线向ψ=0处演化,从而正确识别目标的边界.

4 总结

本文提出了一种边缘型主动轮廓模型.通过轮廓线内、外的灰度平均值,构建了一个自适应符号函数.该函数不仅在目标两侧具有相反的符号,而且能够反映出演化曲线距离目标边界的远近程度并自动据此调节演化速度,可以使曲线在迭代过程中自适应地调整运动的方向.因此,可以设置不同的初始轮廓线参数.利用本文提出的自适应边缘指示函数,进一步使得演化曲线改善容易陷入虚假边界及易在边界泄漏的问题,同时加快了演化速度.改进了距离规则项,实现对弱边界的准确检测,提高了演化的稳定性.通过与分割评价数据库进行对比可以看出,本文模型有较高的分割精度.通过与3个著名的模型对比更进一步说明:本文模型分割速度较快,提高了对初始轮廓的鲁棒性,抗噪声能力强.

猜你喜欢

轮廓线轮廓边缘
立体图像任意剖面轮廓线提取方法仿真研究
跟踪导练(三)
一张图看懂边缘计算
墨戏
儿童筒笔画
创造早秋新轮廓
在边缘寻找自我
走在边缘
边缘艺术