本发明涉及一种遥感图像重建方法,特别涉及一种双参数贝塔过程联合字典的遥感图像超分辨重建方法,属于遥感图像处理技术领域。
背景技术:
高分辨率遥感图像是遥感应用的重要保证,但获取高分辨率遥感图像并非易事,会受到遥感成像传感器阵列密度、大气扰动、相对运动、环境噪声等多种因素的影响,因此,提高遥感图像的空间分辨率是遥感技术的重要研究方向。获取高分辨率遥感图像最直接的方法是改进遥感成像硬件系统,但会带来成本高昂、周期长、甚至技术无法实现等缺陷。遥感图像超分辨重建(super-resolutionreconstruction,srr)就是一种在不改变现有遥感硬件成像设备及工艺水平条件下,利用软件处理的方法对同一场景的单帧或多帧地物遥感低分辨率图像(low-resolutionimages,lris)进行处理得到一帧高分辨率图像(high-resolutionimages,hri)的技术。这种srr技术可以经济、高效地提高遥感图像空间分辨率,改善图像视觉质量,已成为遥感图像处理领域的研究热点。
随着国内外对超分辨技术研究的深入,srr可总结为基于重建的超分辨和基于学习的超分辨方法两类,近年基于学习的图像超分辨方法已成为研究热点。基于学习的srr方法是利用机器学习的思想,通过对样本图像进行训练和学习,获取hri和lris间的关系,并利用这种关系对类似的lri进行超分辨重建。随着压缩感知、稀疏表示技术的发展,很多学者提出基于字典学习的超分辨方法,如奇异值分解(k-svd)法、在线字典更新(odl)法、有效稀疏编码法等。这种方法采用字典学习技术获取高、低分辨率图像块间的共同先验信息,高分辨率图像块采用高分辨率字典和稀疏系数来恢复,而稀疏系数采用低分辨率特征块和低分辨率特征字典计算得到。这些算法都可以生成过完备字典和稀疏系数,但其中的稀疏系数都是高、低分辨率字典共享的,因此,算法很难拟合两个特征空间的字典和系数。zeyde等人提出了两步字典学习算法,其中低分辨率字典由k-svd算法学习,高分辨率字典通过最小二乘法生成,尽管这种方法大大降低了计算量,但重建高分辨率图像仍然不准确。yang等人提出了该问题的双层优化解决方案,用一种映射函数代表两个特征空间的关系的经典稀疏表示超分辨方法,但这种方法依然是共享两个特征空间的系数。
技术实现要素:
针对上述不足,本发明提供一种准确、运算量少的双参数贝塔过程联合字典的遥感图像超分辨重建方法。
本发明的双参数贝塔过程联合字典的遥感图像超分辨重建方法,所述方法包括:
s1、输入待重建的低分辨率图像l、高分辨率图像字典d(x)、低分辨率图像字典d(y)和映射矩阵a;
s2、根据低分辨率训练图像字典d(y),对输入的低分辨率图像l进行稀疏编码,获得低分辨率稀疏系数,然后,利用映射矩阵a映射出低分辨率稀疏系数对应的高分辨率稀疏系数,最后,利用所述高分辨率稀疏系数和高分辨率训练图像字典d(x)重建高分辨率图像;
所述高分辨率图像字典d(x)、低分辨率图像字典d(y)和映射矩阵a的获取方法为:
根据成对的高、低分辨率训练图像,利用双参数贝塔过程,获得高分辨率图像字典d(x)、低分辨率图像字典d(y)和映射矩阵a,所述映射矩阵a为低分辨率图像字典d(y)的稀疏系数到高分辨率图像字典d(x)的稀疏系数的映射矩阵,所述稀疏系数为系数权值和字典原子的乘积。
优选的是,所述映射矩阵a:
a=[(α(y)α(y)t)-1α(y)α(x)t]t
其中,α(y)表示低分辨率图像字典d(y)的稀疏系数矩阵,α(x)表示高分辨率图像字典d(x)的稀疏系数矩阵。
优选的是,所述双参数贝塔过程包括:
s31:以像素重叠分别从高、低分辨率训练图像中提取训练图像块xi和yi,i=1,2,...,n,n表示高/低分辨率训练图像块的数量,
s32:初始化字典d(x)、字典d(y)及参数πk、zi、
πk表示贝塔过程函数,zi表示字典原子,
s33:由吉布斯采样更新πk;
πk~beta(a/k,b(k-1)/k),参数a,b>0,beta表示贝塔分布,k=1,2,...k,k表示字典原子的数量;
s34:根据
px和py分别表示高、低分辨率训练图像块耦合特征空间的维数;
s35:根据
s36:根据
s37:根据
s38:根据
s39:执行s33至s38,直至达到循环次数,根据n个图像块xi、yi、
优选的是,所述s1中,还包括:
将输入待重建的低分辨率图像l从rgb色彩空间转换到ycbcr色彩空间,并提取亮度空间特征,根据提取的特征编辑索引;
所述s2中重建高分辨率图像的过程还包括:利用所述索引重建高分辨率图像的亮度空间,并对另外两个色度空间进行双三次插值,将插值后的图像从ycbcr色彩空间转换到rgb色彩空间,完成重建。
优选的是,所述方法还包括采用全局约束和非局部相似性约束重建高分辨率图像:
h*表示约束后的重建高分辨率图像,h表示步骤s2获得的重建高分辨率图像,h0表示初始高分辨率图像,hi和hi(0)表示分别从h和h0中提取的图像块,
本发明的有益效果,本发明解决了耦合特征空间中的双参数联合字典的训练问题,可自适应地训练成更小尺寸的字典,使得字典维数更低。与当前最先进的稀疏字典学习srr方法相比较,超分辨结果图像的主观效果更好,恢复出更多的纹理细节;客观参数psnr提高约1.5db,ssim高出约0.02;本发明利用人眼视觉对亮度空间更敏感的特性,将待重建的遥感图像从rgb空间转换至ycbcr空间,超分辨率重建时只对人眼较为敏感的亮度空间进行的超分辨重建,而对其色度空间仅采用双三次插值处理,这样在保障超分辨重建效果的同时大大减少运算量。超分辨重建时间降低约5秒。
附图说明
图1为本发明具体实施方式中获取高分辨率图像字典d(x)、低分辨率图像字典d(y)和映射矩阵a的流程示意图;
图2为本发明具体实施方式中重建高分辨率图像的流程示意图。
图3为第1组遥感图像的实验结果对比,图3(a)至图3(d)分别表示低分辨率图像、双三次插值法重建结果、经典稀疏表示超分辨方法重建结果和本发明方法超分辨重建结果;
图4为第2组遥感图像的实验结果对比,图4(a)至图4(d)分别表示低分辨率图像、双三次插值法重建结果、经典稀疏表示超分辨方法重建结果和本发明方法超分辨重建结果;
图5为第3组遥感图像的实验结果对比,图5(a)至图5(d)分别表示低分辨率图像、双三次插值法重建结果、经典稀疏表示超分辨方法重建结果和本发明方法超分辨重建结果;
图6为第4组遥感图像的实验结果对比,图6(a)至图6(d)分别表示低分辨率图像、双三次插值法重建结果、经典稀疏表示超分辨方法重建结果和本发明方法超分辨重建结果;
图7为靶标图像实验结果对比,图7(a)至图7(d)分别表示低分辨率图像、双三次插值法重建结果、经典稀疏表示超分辨方法重建结果和本发明方法超分辨重建结果。
具体实施方式
下面将结合本发明实施例中的附图,对本发明实施例中的技术方案进行清楚、完整地描述,显然,所描述的实施例仅仅是本发明一部分实施例,而不是全部的实施例。基于本发明中的实施例,本领域普通技术人员在没有作出创造性劳动的前提下所获得的所有其他实施例,都属于本发明保护的范围。
需要说明的是,在不冲突的情况下,本发明中的实施例及实施例中的特征可以相互组合。
下面结合附图和具体实施例对本发明作进一步说明,但不作为本发明的限定。
本实施方式的双参数贝塔过程联合字典的遥感图像超分辨重建方法,包括:
s1、输入待重建的低分辨率图像l、高分辨率图像字典d(x)、低分辨率图像字典d(y)和映射矩阵a;
s2、根据低分辨率训练图像字典d(y),对输入的低分辨率图像l进行稀疏编码,获得低分辨率稀疏系数,然后,利用映射矩阵a映射出低分辨率稀疏系数对应的高分辨率稀疏系数,最后,利用所述高分辨率稀疏系数和高分辨率训练图像字典d(x)重建高分辨率图像;
所述高分辨率图像字典d(x)、低分辨率图像字典d(y)和映射矩阵a的获取方法为:
根据成对的高、低分辨率训练图像,利用双参数贝塔过程,获得高分辨率图像字典d(x)、低分辨率图像字典d(y)和映射矩阵a,所述映射矩阵a为低分辨率图像字典d(y)的稀疏系数到高分辨率图像字典d(x)的稀疏系数的映射矩阵,所述稀疏系数为系数权值和字典原子的乘积。
本实施方式考虑字典学习算法应该兼顾两个特征空间(低分辨率图像和高分辨率图像)的学习误差,利用双参数贝塔过程,利用先验模型,将字典学习过程中的稀疏系数分解为系数权值和字典原子的乘积,根据字典原子指标缩减字典,得到自适应大小的字典,能以更小的字典重建高质量的超分辨遥感图像;与当前最先进的稀疏字典学习srr方法相比较,超分辨结果图像的主观效果更好,恢复出更多的纹理细节;
优选实施例中,本实施方式的映射矩阵a:
a=[(α(y)α(y)t)-1α(y)α(x)t]t
其中,α(y)表示低分辨率图像字典d(y)的稀疏系数矩阵,α(x)表示高分辨率图像字典d(x)的稀疏系数矩阵。
学习得到稀疏矩阵α(x)和α(y)之后,通过最小二乘法计算得到a,映射矩阵a即为双参数联合训练字典。
双参数贝塔过程具体包括:假设有两个耦合特征空间
其中,
d(x)和d(y)分别为每个空间中学习得到的含有k个原子的字典,k∈(1,px)(n是个不定未知数,根据图像大小而变换);
w由无穷多的从h0中采样得到的
其中,字典原子服从正态高斯分布:
稀疏系数权值服从正态高斯分布:
字典原子服从伯努利分布:
重建误差服从正态分布:
c和d分别表示
以上设计的双参数贝塔过程中,选择相同的zi为xi和yi中的相对应的字典原子,但稀疏系数不同,构成稀疏系数的权重
其中,○表示元素依次相乘。由于
a=[(α(y)α(y)t)-1α(y)α(x)t]t(15)
优选实施例中,本实施方式的双参数贝塔过程包括:
首先生成模拟训练样本图像,选定10帧不同类别、纹理丰富的图像作为hri,分别将hri进行模糊和下采样的降质操作,得到lri,并进行吉布斯采样(gibbssampling),得到字典训练样本。如图1所示,具体步骤如下:
输入:成对的高、低分辨率训练图像
输出:学习得到的字典d(x)、d(y)和矩阵a
s31:设置实验参数,以像素重叠分别从高、低分辨率训练图像中提取训练图像块xi和yi,i=1,2,...,n,n表示高/低分辨率训练图像块的数量,筛选图像块,特征归一化图像块;
s32:初始化稀疏先验:初始化字典d(x)、字典d(y)及参数πk、zi、
πk表示贝塔过程函数,zi表示字典原子,
s33:由吉布斯采样更新πk;
πk~beta(a/k,b(k-1)/k),参数a,b>0,beta表示贝塔分布,k=1,2,...k,k表示字典原子的数量;
s34:根据
px和py分别表示高、低分辨率训练图像块耦合特征空间的维数;
s35:根据
s36:根据
s37:根据
s38:根据
s39:执行s33至s38,直至达到循环次数,根据n个图像块xi、yi、
优选实施例中,如图2所示,所述s1中,还包括:
将输入待重建的低分辨率图像l从rgb色彩空间转换到ycbcr色彩空间,并提取亮度空间特征,根据提取的特征编辑索引;
所述s2中重建高分辨率图像的过程还包括:利用所述索引重建高分辨率图像的亮度空间,并对另外两个色度空间进行双三次插值,将插值后的图像从ycbcr色彩空间转换到rgb色彩空间,完成重建。
具体遥感图像超分辨重建过程中,由于人眼对图像的亮度信息更敏感,所以先对遥感图像进行色彩空间的变换,由rgb彩色空间变换到ycbcr彩色空间,仅对其亮度空间使用双参数联合字典超分辨算法,而对色度空间直接采用双三次插值法。
本实施方式的亮度空间的srr步骤如下:
·输入:低分辨率图像l、学习得到的字典d(x)、d(y)和映射矩阵a
·输出:高分辨率图像h*
s21:以ω像素重叠从l中提取低分辨率图像块li,用四个特征提取算子f1=[-1,0,1],f2=f1,f3=[1,0,-2,0,1],f4=f3构造yi,构造依据xi=h;yi=[f1l;f2l;f3l;f4l],最后,用稀疏编码计算得到
s22:用a将稀疏系数
s23:用
恢复所有高分辨率图像块之后,初始高分辨率图像h0可以由ω重叠像素重建。
s24:采用全局约束和非局部相似性约束进一步提高重建准确性:
h*表示约束后的重建高分辨率图像,h表示步骤s2获得的重建高分辨率图像,h0表示初始高分辨率图像,hi和hi(0)表示分别从h和h0中提取的图像块,
利用本实施方式的方法、双三次插值及背景技术中提到的用映射函数代表两个特征空间的关系的经典稀疏表示超分辨方法对四张遥感和一张靶标的低分辨图像进行超分辨重建实验,获得如图3至图7所示的实验结果,五组的峰值信噪比psnr和结构相似度ssim统计对比如表1所示。客观参数psnr提高约1.5db,ssim高出约0.02;本发明超分辨重建时间降低约5秒。
表1遥感超分辨结果图像的psnr和ssim统计对比
虽然在本文中参照了特定的实施方式来描述本发明,但是应该理解的是,这些实施例仅仅是本发明的原理和应用的示例。因此应该理解的是,可以对示例性的实施例进行许多修改,并且可以设计出其他的布置,只要不偏离所附权利要求所限定的本发明的精神和范围。应该理解的是,可以通过不同于原始权利要求所描述的方式来结合不同的从属权利要求和本文中所述的特征。还可以理解的是,结合单独实施例所描述的特征可以使用在其他所述实施例中。