1. 引言
遥感图像在采集、传输过程中容易受到大气或传感器等诸多因素的影响,导致获取的遥感图像受到模糊和噪声的污染。被污染的图像会影响人们对图像信息的识别和判断,获得到图像信息就无法应用在实际工作当中。近年来随着遥感技术的飞速提升,遥感图像所应用的领域的也在不断扩大。所以对遥感图像复原、提高遥感图像的质量 [1] [2] [3] [4] 的研究是具有现实意义的。
图像复原问题是一个典型的不适定 [5] [6] 的反问题,一幅退化的图像可以看作是由退化算子和噪声共同作用在清晰图像的结果。从数学的角度来说,描述退化图像f与清晰图像u的关系表达式为
,其中K为退化算子,
为噪声。根据附加噪声的种类不同,图像复原的相应的方法也有差异。脉冲噪声是较为常见的一种噪声,它是由于图像信号采集的过程中受到强烈干扰、传感器故障或类比数位转换器错误 [7] [8] [9] 等诸多因素产生的。由于噪声的像素位置成一定密度的随机分布并且通常与临近像素的强度是接近的,导致复原含有噪声的退化图像是不容易的。
目前大多数针对含有脉冲噪声的图像复原模型是以Rudin等人 [10] 提出的基于全变分(TV, Total variation)算法模型为框架加以改进的。Zhong等人 [11] 提出了一种自适应二阶变分模型,该模型对复原脉冲噪声下的模糊图像,可以很好的抑制复原所产生的阶梯效应;Liu [12] 提出了一种混合正则化算子的脉冲噪声图像复原方法,该方法在广义变分框架的基础上融合了小波框架模型克服了原始单一正则化模型因恢复含有脉冲噪声的退化的图像所产生的跳跃的不连续性;Cui等人 [13] 提出了“非凸 + 非凸”的图像复原模型,即模型同时具有非凸性的数据拟合项和非凸性的正则项,该模型具有对恢复图像边缘信息的具有很好的效果。Liu等人 [14] 构造了一种基于Lp范数的非凸模型用于去模糊去噪,保证复原的同时对增强边缘细节也有良好的效果。Luo等人 [15] 在正则项中引入对数函数,并采用平滑裁剪绝对偏差(SCAD)函数作为数据拟合项,所提出的模型可以有效地恢复含有高密度脉冲噪声图像;Yuan等人 [16] 根据脉冲噪声的特性,结合贝叶斯定理构造了L0-TV的稀疏优化模型,并且提出了一种近似交替乘子法对模型进行优化求解,证实了该方法框架是对于复原含有脉冲噪声的退化图像很有效的一种框架模型。
近年来组稀疏 [17] [18] [19] 方法在图像复原中的应用得到了广泛的应用,通过将图像分若干块从而进行稀疏建模,可以达到更好的复原效果。Selesnick等人 [20] 将全变分框架与组稀疏相结合,该方法可以减轻在基于全变分模型的图像复原中经常出现的阶梯伪影。Luo等人 [21] 提出了组稀疏与双重L1范数相结合的图像复原模型,该方法容易获得更具有鲁棒性的稀疏系数,能够有效地减少振铃效应,保留图像的本身特征。Yi等人 [22] 构造由重叠组稀疏约束项和非凸性二阶Lp范数约束项的复原模型,具备非常好的边缘保持能力和纹理细节保留效果;Adam等人 [23] 提出了一种基于混合二阶非凸全变分和重叠组稀疏正则项的模型,该模型保证了消除阶梯伪影,同时保持清晰的恢复图像;Yin等人 [24] 在L0TV模型的中融合了组稀疏正则化器,既保持L0范数可以有效地去除脉冲噪声,又有效的抑制了复原过程中阶梯伪影的产生。
根据上述诸多学者的研究为启发,结合多尺度分析工具能够对图像稀疏表示,本文设计了一种基于小波变换与组稀疏相结合的复原含有脉冲噪声的遥感图像的算法。采用重叠组稀疏正则化器与小波变换下的L0范数对梯度图像进行联合稀疏建模构建先验信息,从而达到在保证复原效果的同时抑制复原过程中产生的阶梯伪影。
2. 预备知识
2.1. 脉冲噪声与L0-TV模型
常见的脉冲噪声主要包含随机值脉冲噪声(Random-valued impulse noise)与椒盐脉冲噪声(Salt-and-pepper impulse noise)。设原始图像的像素值的范围为[umin, umax],i为像素的位置,ui与
分别表示为原始像素值与退化损坏的像素值,则含有两种脉冲噪声的退化图像的数学表达式分别如下:
随机值脉冲噪声:一定百分比的像素被改变为均匀随机数
:
(1)
椒盐脉冲噪声:一定百分比的像素被改变为umin或umax:
(2)
其中K为退化算子,prv与psp分别为随机值脉冲噪声与椒盐脉冲噪声的噪声密度。
L0-TV图像复原模型的数学表达式为:
(3)
其中
表示哈达玛乘积,
为平衡正则项与数据拟合项的平衡参数;
由使用者指定,当
时,表示对应第i个位置的像素值为异常值;当
时,表示对应第i个位置的像素值为潜在的异常值。在本文中,我们选取椒盐脉冲噪声,当含噪图像中的像素ui取umin或umax时,对应
设置为0;当含噪图像中的像素ui取其他值时,对应
设置为1。
问题(3)可以等价地表述成具有平衡约束的数学规划问题(MPEC, Mathematical Program with Equilibrium Constraints),其数学表达式如下:
(4)
2.2. 重叠组稀疏正则化器
TV正则化器在稀疏先验信息建模的过程中容易产生块状伪影,重叠组稀疏正则化器(OGS, Overlapping group sparsity) [25] 在抑制块状伪影具有良好的效果,对于任意的图像,二维点群组稀疏正则化算子的数学表达式如下:
(5)
其中,算子
的中心为(i,j),
,
。设向量
是通过堆叠矩阵
的列得到的,二维重叠组稀疏正则化算子就可以紧凑的写为:
(6)
2.3. 小波域稀疏表示
小波变换 [26] [27] [28] 是一种重要的信号多尺度分析工具,在图像处理中,小波变换可以将图像信号按不同尺度分解出多个不同频子带的小波系数,这些小波系数可以分析图像在不同尺度下的边缘细节信息。利用小波变换的这一特性,小波变换可以应用于特征提取、图像分解与重构等领域。相比较去空间域表示,小波变换对像素进行空间去相关,从而允许更简化和准确的建模。图1为小波变换的分解结构图:
(a) 原图像u(b) 第一级小波分解 (c) 第二级小波分解
Figure 1. The structure diagram of Wavelet decomposition
图1. 小波分解结构图
其中黄色区域为低频信息,表示原图像u的近似分量;绿色区域为高频信息,表示原图像u的细节分量。
3. 本文算法
3.1. 基于小波变换与组稀疏相结合的遥感图像复原算法
利用小波变换后高频图像具有稀疏性的特征再结合组稀疏正则化器对梯度图像进行双重约束,构造基于小波变换与组稀疏相结合的遥感图像复原模型:
(7)
其中K为模糊算子,
组稀疏正则化算子,W为小波变换,
为梯度算子,
,
为正则化器的平衡参数。
将模型(7)等价地表述为具有平衡约束的数学规划问题,其数学表达式如下:
(8)
3.2. 模型求解
对于极小化问题(8)的求解,首先使用变量分裂方法,引入辅助变量d、z、c,则上述问题变为如下约束优化问题:
(9)
其中
、
、
。
下面对问题(9)进行求解,首先构造增广拉格朗日函数
(10)
其中
、
、
、
是朗格朗日乘数,
、
、
、
是各二次罚函数项的罚参数。
根据交替方向乘子法,问题(10)可以交替得到下列各子问题
(11)
(12)
(13)
(14)
(15)
(16)
(17)
(18)
(19)
接下来各个子问题进行求解:
1) 首先是u-子问题,(11)式等价于
(20)
根据一阶最优性条件,问题(20)需要求解下面关于u的线性方程组
(21)
其中,
表示小波变换的逆变换,
表示梯度的逆变换。那么u-子问题的闭合解的形式为:
(22)
其中,F表示为二维离散傅里叶变换,F-1表示为二维离散傅里叶变换的逆变换。
2) 求解v-子问题,(12)式根据投影计算得出v的最终解形式:
(23)
3) 对于d-子问题,该问题是重叠稀疏组问题,问题本身存在内部迭代,采用极小化方法对问题进行求解:
(24)
其中,i表示内部迭代次数,
。
4) 对于c-子问题,合并(14)式中的同类项得到下式:
(25)
根据硬阈值算法,上述问题的近似解是:
(26)
其中硬阈值算子的表达式为:
(27)
5) 求解z-子问题,合并(15)式中的同类项后得到下式:
(28)
最后舍弃掉常数项,通过四维收缩算子计算(28)式,得到z-子问题的最终形式
(29)
6) 根据(16)式、(17)式、(18)式、(19)式中更新拉格朗日乘数
、
、
、
。
3.3. 算法流程
算法:基于小波变换与组稀疏相结合的遥感图像复原算法
输入:含有脉冲噪声的模糊图像f,模糊算子K,正则化参数
输出:复原后的图像u
初始化:
,拉格朗日乘子
、
、
、
步骤1:设置迭代次数k = 1:1000,设置
为输出条件;
步骤2:利用(22)式更新u;
步骤3:利用(23)式更新v;
步骤4:利用(24)式更新d;
步骤5:利用(26)式更新c;
步骤6:利用(29)式更新z;
步骤7:利用(16)、(17)、(18)、(19)式更新
、
、
、
。
4. 实验结果与分析
本节在开放的遥感卫星图像数据集中选取六幅大小为512 × 512的图像(ImageA, ImageB, ImageC, ImageD, ImageE, ImageF)进行仿真试验。清晰图像见图2。本文算法实验在MATLAB R2020b上进行,且电脑配置为Windows10,Intel(R) Core(TM) i5-6300 HQ CPU @ 2.30GHz。
实验选用高斯模糊、平均模糊、运动模糊,噪声均选取噪声密度为30%的椒盐噪声模拟模糊噪声图像,其中高斯模糊核的宽度尺寸为7,滤波器的标准差为6;平均模糊核大小为7 × 7;运动模糊核的运动的角度为3,运动的长度为30。选择L0-TV算法 [16] 、HNHOTV-OGS算法 [23] 、L0-OGSTV算法 [24] 与本文提出的模型算法进行实验对比。每种模糊环境下的复原效果分别选取两种场景进行展示,本文与对比试验的图像复原的结果的对比图为图3~8所示,作为复原效果的主观评价指标。根据六组对比图可以观察,本文算法在复原的过程中可以有效地复原出清晰图像,并且可以很好的抑制振铃效应和阶梯效应。
选择指标峰值信噪(PSNR)和图像相似度(SSIM)作为对比试验的客观评价指标,各算法的PSNR与SSIM的数值结果在表1中展示,从表1中的数据得知,本文算法相比较三种对比实验算法,对同一幅含有模糊和脉冲噪声的退化图像的复原结果,PSNR与SSIM均有提升。通过表1可以看出,本文算法的客观评价指标峰值信噪(PSNR)和图像相似度(SSIM)的数值结果均高于其它复原算法。
Table 1. Evaluation index results of restoration using different algorithms for remote sensing images under impulse noise
表1. 脉冲噪声下的遥感图像使用不同算法复原的评价指标结果
综合上述可以得出结论,本文提出的遥感图像复原算法,对含有脉冲噪声的遥感图像的复原结果,在主观视觉效果与客观指标上均优于三种对比试验算法,针对含有脉冲噪声的遥感图像复原问题,本文算法是一种有效的解决方案。
5. 结束语
本文对开放的遥感卫星图像数据集中不同场景的图像进行图像复原研究,设计了小波变换与组稀疏相结合的遥感图像复原算法。该算法模型采用L0范数作为数据保真项,可以有效地去除脉冲噪声;在正则项上,本文采用重叠组稀疏正则化器与梯度图像在小波变换下的L0范数进行稀疏建模。使用优化最小化方法分别与交替方向乘子法对算法模型进行求解,从而复原出清晰图像。本文方法有效复原了脉冲噪声下的高斯模糊、平均模糊、运动模糊的遥感图像。
基金项目
吉林省教育厅科学技术研究项目,JJKH20230788KJ。