APP下载

图像去模糊的自适应交替方向乘子重叠组稀疏方法

2018-09-13李胜光宋一帆马天磊

郑州大学学报(工学版) 2018年5期
关键词:变差正则复原

王 杰, 李胜光, 宋一帆, 白 珂, 马天磊

(郑州大学 电气工程学院,河南 郑州 450001)

0 引言

随着科学与技术的发展,图像成为越来越重要的信息传播媒介.然而图像在形成、传送和储存的过程中,由于一些不可抗拒的因素,会产生图片污染进而造成图片退化,即模糊图片[1-2].通常,图像在退化过程中可用下面模型进行近似,

g=h⊗f+n,

(1)

式中:f为清晰图片;g为观察到的图片;n是零均值高斯白噪声;h为点扩散函数;⊗是卷积算子.图像复原技术是一个病态问题,通常使用正则化技术去解决这个问题,通过最小化下式的变分问题来复原清晰图像f,

(2)

式中:第一项通常被称为保真项,用来保存图像的最大信息,此部分可以通过最大后验估计推导出来[3];第二项被称为正则化项,通过改善正则化项使复原的图像更清晰.正则化参数α>0.

正则化项φ(f)的选取对复原出的图像质量有着至关重要的影响.Rudin等[4]提出一种全变差方法,用φ(f)=‖f‖1表示正则化项.这种方法能够比较好地保存图像的边缘信息,然而会产生阶梯效应.为了减弱复原图片的阶梯效应并且保存图像的边缘信息,Lysaker等[5]提出了一种用二阶次全变差正则化项取代原始全变差正则化项的方法.Chan等[6]提出了一种混合的全变差公式,将一阶和二阶全变差混合使用.也有其他学者采用更高阶的全变差公式,然而阶次越高,改进后的模型就越复杂,计算难度也会随之增加,产生一些不良效应.Huang等[7]通过引入辅助变量来代替真实图片,提出了一种快速总变差(Fast-TV)最小化方法.Liu等[8]和Shi等[9]采用重叠组稀疏正则化项恢复噪声损坏图像,在减轻阶梯效应方面非常有效.Bai等[10]提出了一种基于交替方向乘子法求解全变差正则化的模型,该模型对去除椒盐噪声非常有效,但对于随机噪声效果一般.赖明倩等[11]提出了一种全范数全变差的算法,既保留了边缘效果又减弱了阶梯效应.

全变差正则化方法能够较好地保留图像的边缘信息,但会产生阶梯效应,所以要对正则化项进行改进以便减弱阶梯效应.采用重叠组稀疏全变差函数(OGSTV)代替正则化项,该方法曾被用来去除一维噪声[12],能够保留边缘特性,并且减弱阶梯效应.重叠组稀疏全变差通常是利用交替方向乘子(ADMM)方法来进行求解.但在求解过程中,其惩罚因子对复原效果有较大影响,且一般由经验选取,不易调出最佳效果,所以笔者提出了一种自适应交替方向乘子法来求解该模型,在复原图片的过程中根据复原情况能够实时地调节ADMM的惩罚因子.该算法不仅能够保存图像的边缘特性,并且能够克服全变差的阶梯效应.此外,自适应交替乘子法比经典的交替乘子法更具有鲁棒性,第3节的试验证明了该算法的高效性.

1 预备知识

1.1 重叠组稀疏

(3)

所以对应的全变差公式的正则化项变为:

φ(s)=φ(xs)+φ(ys).

(4)

1.2 OGSTV

通过引入定义的重叠组稀疏正则化项,图像复原公式(2)可转换为:

(5)

对于图片中的每个像素值都有一定的范围约束PΩ=[pl,pu],这种约束被称为黑箱约束[13].通过引入一些辅助变量vx、vy、z,将式(5)转化为有约束问题,

s.tvx=xf,vy=yf,z=f.

(6)

相应的增广拉格朗日函数为:

φ(vy))+PΩ(z)+μT{(xf-vx)+

(7)

式中:μ是拉格朗日乘数;σ>0是惩罚因子.式(7)可以用ADMM算法进行求解.

2 自适应ADMM-OGSTV

在式(7)中,σ通常是由经验选取.但试验证明其值选取的大小对图片恢复质量有着至关重要的影响.所以提出了一种自适应正参数σ的选取方法.对式(7)进行变形得,

(8)

利用ADMM算法对式(8)进行求解可得复原图片f的迭代公式为(文献[14]中给出了相应的求解过程),

(9)

由式(9)可以容易地看出图片复原迭代公式是以Si,i=1,2,3为步长的梯度下降.为了证明σ值变化对f的影响,令σ为自变量,其他参数为常量,可得,

f=argminΦ(σ),

(10)

(11)

为了在图片复原过程中根据复原情况进行自适应的迭代,以空间欧式距离作为复原前后相似度度量得,

(12)

当Δk+1<ε,ε为任意小常数,图像复原迭代过程中图片再无变化,迭代终止.在迭代过程中根据Δk实时调整σk值,做如下定义,

(13)

3 试验与结果

为了更加充分地验证本算法,笔者选取不同尺度的图片进行测试,所有测试图片如图1所示,其中图Goldhill尺寸为512×512,图Lena尺寸为256×256.

图1 测试图片Fig.1 The picture for test

在试验过程中,分别对测试图片进行了两种高斯模糊,模糊核为7×7记为模糊核1;模糊核为21×21记为模糊核2,高斯函数标准差均为2.两种运动模糊,位移像素为10,偏移角度为30,记为模糊核3;位移像素20,偏移角度为45,记为模糊核4,偏移角度均为逆时针方向,论文以下内容采用此简记方式.笔者方法和最新的Fast-TV算法[7]、TVFN算法[11]和OGSATV-ADM4算法[8]进行了对比.为了保证对比的公平,所有算法的迭代终止条件为ε=1×10-5,其他条件为其论文中证明的最优条件.笔者方法经试验证明参数γ取1.01,参数η取1.09;重叠组稀疏窗,选取W=3,文献[15]已验证其最优性.实验环境为Windows 7, 64位Intel Core i5-3230 CPU 2.5 GHz,MATLAB版本为2014 a.

笔者对恢复出来的图片进行质量评价时,使用的是3种常用的评价方法,相对误差(RE)、峰值信噪比(PSNR)和信噪比(SNR),使用这3种方法[7]比较容易和其他算法进行对比.模糊信噪比[7](BSNR)为试验附加噪声.

3.1 ADMM计算步长的影响

采用交替方向乘子法去优化重叠组稀疏正则化模型时,为了验证正项惩罚参数σ对复原图片性能的影响,笔者选取图片Goldhill在模糊核1附加BSNR=40的零均值高斯噪声情况下,绘制了信噪比在不同σ值的曲线,具体如图2所示.

图2 不同σ值复原图片的SNR值Fig.2 SNR values of the picture at different σ values

由图2可知,不同σ对复原图片的SNR影响是至关重要的,即使σ发生非常微小的变化也能对复原结果产生巨大的影响.通过手动调节σ不仅费时,而且不易找出最佳σ值.所以笔者提出一种自适应调节方法,在迭代过程中根据复原图片fk和fk+1进行实时的调节,这样不仅易于操作,而且复原效果更好.

3.2 自适应步长复原效果

为了验证自适应步长的复原效果,笔者对不同尺寸图片进行了仿真,模糊和复原效果如图3所示.图3(a)、3(c)中依次为模糊核1至4,Gold-hill、Lena的退化图像,对于每种模糊类型都加入了BSNR=40的零均值高斯噪声.图3(b)、3(d)为相应的复原效果图.从人眼感官上,复原出的图片既保留了边缘特性又消除了阶梯效应,对于不同尺寸的图片在不同模糊类型和不同模糊核上均取得了较好结果.

3.3 算法对比

为了验证笔者所提算法的优越性,依次用算法Fast-TV、TVFN、OGSATV-ADM4和本文算法对不同尺度图片在不同模糊类型和不同模糊核下,对复原图片进行了对比,对比结果如表1所示.由表1可知,笔者所提算法的PSNR、SNR和RE均优于其他算法,不过在复原时间上略长于Fast-TV和OGSATV-ADM4,但保证了复原效果.

图3 Goldhill和Lena不同模糊和复原图片Fig.3 Different blur and restore pictures of Goldhill and Lena

为进一步验证笔者所提方法的优越性,选取Goldhill在模糊核1情况下,各算法的迭代结果对比如图4所示.笔者算法在获取最佳结果时迭代次数最少,证明了所提算法加速过程的有效性.由于Fast-TV、TVFN、OGSATV-ADM4和笔者算法辅助变量储存空间[16]依次为O(5mn)、O(6mn)、O(6mn)和O(8mn),所以笔者算法在自适应迭代过程中消耗一些时间.

各算法的计算时间迭代结果如图5所示.由图5可知,笔者算法获得最佳结果的计算时间略长,然而所提算法本身具有加速过程且复原结果具有很大提升,即使相比原算法多用0.14 s也是具有研究价值的.

各算法在不同模糊情况下的复原效果如图6所示.图6选取Goldhill在模糊核1和Lena在模糊核4情况下的模糊图片.在图6第一行中,我们可以观察到Fast-TV和TVFN在箭头所指处都产生了块状效应即阶梯效应,OGSATV-ADM4在一定程度上避免了阶梯效应.笔者所提算法不仅避免了阶梯效应,而且相比OGSATV-ADM4更符合人眼观测效应.

图4 Goldhill SNR在迭代次数上的迭代结果Fig.4 The SNR iteration results of Goldhill over iterations

图5 Goldhill SNR在计算时间上的迭代结果Fig.5 The SNR iteration results of Goldhill over time

图片模糊核GoldhillLena模糊核1模糊核2模糊核3模糊核4模糊核1模糊核2模糊核3模糊核4Fast-TVTVFNOGSATV-ADM4笔者算法PSNR/dB30.7730.0534.1631.0830.2229.4135.0932.19SNR/dB24.4123.9427.7924.4423.0622.2327.4024.93RE0.0630.0670.0450.0600.0700.0770.0430.057时间/s2.963.551.742.661.231.791.131.25PSNR/dB30.9530.3234.3731.2430.4729.5035.3432.31SNR/dB24.6823.9927.9324.5923.2522.3027.6925.02RE0.0600.0650.0430.0590.0690.0750.0400.056时间/s3.915.352.614.082.113.371.741.86PSNR/dB31.1430.6034.5831.3330.6330.1235.7832.38SNR/dB24.7724.2328.2224.9623.7923.2228.8725.47RE0.0580.0610.0390.0560.0670.0690.0360.053时间/s3.073.751.813.011.492.081.351.61PSNR/dB32.0030.7735.4632.2431.4630.5236.3533.47SNR/dB25.6324.4029.1025.8724.5523.6229.4426.56RE0.0520.0600.0350.0510.0590.0660.0340.047时间/s3.213.961.933.141.582.151.461.68

在图6第二行中,我们可以观察到,在箭头所指处笔者所提算法更加清晰,并且在对图片放大时对比算法都不同程度地产生了阶梯效应.通过试验证明,所提算法对不同尺度的图片在不同模糊类型和不同的模糊核下复原效果均优于对比算法,不仅恢复出了很好的边界,而且能够很好地避免阶梯效应.

4 结论

笔者研究了基于重叠组稀疏全变差正则化的图像去模糊算法.为了解决相应的最小化问题,提出了一种自适应交替方向乘子算法来求解该模型,该方法在复原图片的过程中根据复原情况能够实时地调节ADMM的惩罚因子.通过试验证明该方法的有效性.与其他算法相比具有较高的峰值信噪比和信噪比,并且具有较小的相对误差.笔者提出的算法不仅能够保存边缘特性,而且能够避免阶梯效应.由于笔者所研究算法是非盲去模糊方法,在未来工作中可能把此算法扩展到盲去模糊问题中.

猜你喜欢

变差正则复原
半群的极大正则子半群
温陈华:唐宋甲胄复原第一人
献血后身体会变差?别信!
π-正则半群的全π-正则子半群格
Virtually正则模
滞后型测度泛函微分方程的Φ-有界变差解*
浅谈曜变建盏的复原工艺
毓庆宫惇本殿明间原状陈列的复原
任意半环上正则元的广义逆
双次幂变差与价格跳跃的分离