一种基于自适应切换分析稀疏与合成稀疏正则化的图像复原方法

文档序号:10471898阅读:256来源:国知局
一种基于自适应切换分析稀疏与合成稀疏正则化的图像复原方法
【专利摘要】本发明提供一种基于自适应切换分析稀疏与合成稀疏正则化的图像复原方法,首先,对高质量的训练样本构成的集合聚类并为每一类样本训练一个合成稀疏子字典和一个分析稀疏子字典;然后,将训练的字典分别用于复原低质量的训练图像样本集合,统计各类采用每一种子字典的复原误差,根据误差大小自适应切换选择稀疏正则化;最后,在切换准则基础上建立了图像重建复原模型,并采用交替优化策略求解该模型。本发明有益的效果:本发明考虑了图像子块对稀疏子字典的适应性,为图像子块自适应的选择稀疏正则化,提高了图像的复原质量;复原图像子块时每次只选择一种稀疏正则化。
【专利说明】
-种基于自适应切换分析稀疏与合成稀疏正则化的图像复原 方法
技术领域
[0001] 本发明属于图像处理技术领域,设及一种图像复原的方法,具体是一种基于自适 应切换分析稀疏与合成稀疏正则化的图像复原方法,用于对已知模糊类型的模糊图像进行 复原和对已知低分辨率图像复原为高分辨率图像。
【背景技术】
[0002] 图像在传输、处理、记录等过程中,由于模糊、下采样、噪声等影响导致图像质量下 降。图像质量下降的过程可W表示为:
[0003] y = SBx+n
[0004] 上式中,X为原始高质量图像,B为模糊算子,S是下采样矩阵,η是加性高斯白噪声, y为退化图像。图像复原就是根据退化图像y求解未知高质量图像X的过程。当S为单位矩阵, B为模糊算子时,上述问题就成为图像去模糊;当S为下采样矩阵,B为模糊算子时,上述问题 就成为图像超分辨率问题。
[0005] 图像复原是指从低质量图像中重建出高质量图像的过程,该过程是一个病态反问 题,在求解该问题时往往需要引入图像先验知识,W先验作为反问题模型的正则项。常用的 两种稀疏正则化是合成稀疏正则化和分析稀疏正则化,很多文献显示运两种正则化用于图 像复原都取得了较好的效果。但是,运些文献在应用运两种稀疏正则化时都没有考虑图像 块的特点,一些图像块更适合于采用合成稀疏正则化进行复原,一些图像块更适合于采用 分析稀疏正则化进行复原,因此没有为图像块找到最佳的稀疏正则化。

【发明内容】

[0006] 本发明的目的是克服现有技术中的不足,而提供一种基于自适应切换分析稀疏与 合成稀疏正则化的图像复原方法。
[0007] 本发明的目的是通过W下技术方案实现的。运种基于自适应切换分析稀疏与合成 稀疏正则化的图像复原方法,该方法首先对高质量的训练样本构成的集合聚类并为每一类 样本训练一个合成稀疏子字典和一个分析稀疏子字典;然后将训练的字典分别用于复原低 质量的训练图像样本集合,统计各类采用每一种子字典的复原误差,根据误差大小自适应 切换选择稀疏正则化;最后在切换准则基础上建立了图像重建复原模型,并采用交替优化 策略求解该模型。
[000引该发明技术方案中各个细节详细说明如下:
[0009] 步骤1:训练K类合成稀疏子字典和分析稀疏子字典,进一步包括:
[0010] 将高质量的训练样本图像进行分块,分成大小为VFxV?的相互重叠若干像素的 图像子块,重叠像素个数小于P,子块向量化后表示为XiERP。首先,对输入的Μ个图像子块进 行Κ均值聚类,采用主分量分析法为每一类训练一个合成稀疏子字典,采用序列最小化特征 值方法为每一类训练一个分析稀疏子字典,得到Κ个类中屯、、合成稀疏子字典和分析稀疏子 字典,分力[]记为1^111、巫111, Qm,m二 1...K。
[0011] 步骤2:确定稀疏正则化切换准则,进一步包括:
[0012] 输入训练低质量图像集合,对于图像去模糊,低质量图像是模糊图像;对于超分辨 率重建,低质量图像是低分辨率图像经插值放大后与目标图像相同大小的图像。对每一幅 图像分成大小为y/iT X 的子块,由运些子块构成子块集合,将子块与K个类中屯、具有最小 欧氏距离的类作为该子块的类索引41,山=1-'1(,为子块集合中的每个子块确定类索引。对 类索引为ki的子块分别按合成稀疏子字典>分析稀疏子字典复原该子块,分别计算 并记录合成稀疏、分析稀疏复原误差,为子块集合中的每个子块记录复原误差。统计子块集 合中具有相同类索引的子块的合成稀疏、分析稀疏复原误差;如果该类索引的合成稀疏复 原误差更小,则选择合成稀疏正则化复原该类子块,否则选择分析稀疏正则化复原该类子 块。
[0013] 步骤3:图像复原重建,进一步包括:
[0014] 在合成稀疏正则化和分析稀疏正则化的共同作用下,图像的复原重建模型表示 为:
[0015]
[0016] 式(1)中,ki是子块i所属类的索引,曰1和馬京分别是子块i的合成稀疏编码与分析 稀疏编码,RiX表示从图像X中提取子块i,e,、,丫 1是正则化系数,II · I lq(q是0含q含1的整 数)是〇范数。5*^为切换开关,当^=1>式(1)变成合成稀疏正则化模型;当^=〇1式(1) 变成分析稀疏正则化模型。在切换机制下,对于给定的子块,式(1)中的两个正则项只有一 个被激活。采用交替优化策略近似求解式(1)。求解过程为:
[0017] 1)输入:图像y
[001引2)初始化:迭代次数S = 0,图像子块数化,图像中的像素个数NI,最大外循环次数 o_Max_iter,最大内循环次数i_Max_iter,外循环迭代终止误差o_e,内循环迭代终止误差 i_e,期望的协稀疏度/,初始值xW。
[0019] 3)当1< 化时
[0020] (a)令yi = RixW。找出与yi具有最短欧氏距离的类中屯、A,k功yi的类索引,得到对 应的合成与分析子字典^&, ^
[0021] (b)在当前类的切换函数基础上执行下列情况之一:
[0025]采用贪婪梯度投影法求解。
[0026] 当&-咐,取q = l,
[0027]
[002引其中soft(>,>)是软阔值运算。
[00巧]循环变量i增1。
[0030] 4)置 u = 0,x(u) = x(s)
[0031] 5)更新 x(u)
[0032]
[0033] 6)判If
I否成立,不成立则返回步骤5),否 则到步骤7)
[0034] 7)更新 x(s") = x(u+i)
[0035] 8)判断或s<o_Max_ite;r是否成立,不成立则返回步骤3),否 则到步骤9)
[0036] 9)输出图像 x = x(s+i)。
[0037] 本发明有益的效果:本发明考虑了图像子块对稀疏子字典的适应性,为图像子块 自适应的选择稀疏正则化,提高了图像的复原质量;复原图像子块时每次只选择一种稀疏 正则化,
[0038] 因此复原模型的计算复杂度不高于单独采用合成稀疏正则化或分析稀疏正则化 时的最高计算复杂度。
【附图说明】
[0039] 图1是本发明图像复原模型建立过程示意图;
[0040] 图2是本发明与其他几种去模糊方法对比图;
[0041 ]图3是本发明与其他几种超分辨率方法对比图。
【具体实施方式】
[0042] 下面结合附图和实施例对本发明做进一步描述。
[0043] 如图1所示,步骤1:训练K类合成稀疏子字典和分析稀疏子字典,具体是:将高质量 的训练样本图像进行分块,分成大小为、屯x、/F=7X7的相互重叠5X7个像素的图像子块, 子块向量化后表示为xiERP。首先,对输入的M=727615个图像子块进行K均值聚类,Κ = 200, 采用主分量分析法为每一类训练一个合成稀疏子字典,采用序列最小化特征值方,为每一 类训练一个分析稀疏子字典,得到Κ个类中屯、、合成稀疏子字典和分析稀疏子字典,分别记 为^!!!、巫m, Qm,rn_1...K。
[0044] 步骤2:确定稀疏正则化切换准则,具体是:
[0045] 输入训练低质量图像集合,对于图像去模糊,低质量图像是模糊图像;对于超分辨 率重建,低质量图像是低分辨率图像经插值放大后与目标图像相同大小的图像。对每一幅 图像分成大小为7X7的子块,由运些子块构成子块集合,将子块与Κ个类中屯、具有最小欧氏 距离的类作为该子块的类索引41^1=1-乂,为子块集合中的每个子块确定类索引。对类索 引为ki的子块分别按合成稀疏子字典Φν、分析稀疏子字典Q*,.复原该子块,分别计算并记录 合成稀疏、分析稀疏复原误差,为子块集合中的每个子块记录复原误差。统计子块集合中具 有相同类索引的子块的合成稀疏、分析稀疏复原误差;如果该类索引的合成稀疏复原误差 更小,则选择合成稀疏正则化复原该类子块,否则选择分析稀疏正则化复原该类子块。
[0046] 步骤3:图像复原重建,如图1所示,具体是:
[0047] 由于人眼对于亮度分量Υ更为敏感,对于彩色的RGB图像,先转换成YUV图像。当复 原模型为超分辨率重建时,在Y分量上进行超分辨率重建,而UV分量采用双Ξ次插值放大; 当复原模型为去模糊时,只在Y分量上进行去模糊,UV分量不做处理。然后将YUV图像再转换 成RGB图像;对于灰度图像,直接在灰度图上进行超分辨率重建或者去模糊。
[0048] 在合成稀疏正则化和分析稀疏正则化的共同作用下,图像的复原重建模型表示 为: Γ00491
[0050] 式(1)中,ki是子块i所属类的索引,曰1和%&分别是子块i的合成稀疏编码与分析 稀疏编码,RiX表示从图像X中提取子块i,e,、,丫 i是正则化系数,I I · llq(q是0含q。的整 数)是q范数。λι= 丫 i = 20。当超分辨率重建时,β = 2000;当去模糊时,0=15〇〇。&,为切换开 关,当&, =1,式(1)变成合成稀疏正则化模型;当A =0,式(1)变成分析稀疏正则化模型。在 切换机制下,对于给定的子块,式(1)中的两个正则项只有一个被激活。采用交替优化策略 近似求解式(1)。求解过程为:
[0051] 1)输入:图像y;
[0052] 2)初始化:迭代次数s = 0,图像子块数Np,图像中的像素个数化,最大外循环次数〇_ Max_ite;r,最大内循环次数;!_]/[曰又_;^61·,外循环迭代终止误差o_e,内循环迭代终止误差i_ e,期望的协稀疏度J,初始值xW。取/"=70,0_e = i_e = 2X 10-6,当超分辨率重建时,o_Max_ iter = 500, i_Max_iter= 10,xW为输入低分辨率图像插值放大的结果;当去模糊时,0_ Max_iter = 200,i_Max_iter = 1,x(0)为输入模糊图像;
[0化3] 3)当1<咕时
[0054] (a)令yi = RixW。找出与具有最短欧氏距离的类中屯、從,'ki为yi的类索引,得到 对应的合成与分析子字典Ω巾。
[0055] (b)在当前类的切换函数基础上执行下列情况之一:
[0化6] 当鸣=〇时,取9 = 0,
[0059] 采用贪婪梯度投影法求解。
[0060] 当&=1财,取9 = 1,
[0061]
[0062] 其中soft(>,>)是软阔值运算。
[0063] 循环变量i增1。
[0064] 4)置 u = 0,x(u) = x(s)。
[00化]5)更新 x(u)。
[0066]
[0067] 6)判断
或s>i_Max_ite;r是否成立,不成立则返回步骤 5),否则到步骤7)。
[006引 7)更新 x(s") = x(u")。
[0069] 8)判l|F
或s>o_Max_ite;r是否成立,不成立则返回步骤 3),否则到步骤9)。
[0070] 9)输出图像 x = x(s")。
[0071] 如图2,图3所示,本发明采用的方法较W往几种方法,更提高了图像的复原质量。
[0072] 最后说明的是,W上优选实施例仅用W说明本发明的技术方案而非限制,尽管通 过上述优选实施例已经对本发明进行了详细的描述,但本领域技术人员应当理解,可W在 形式上和细节上对其作出各种各样的改变,而不偏离本发明权利要求书所限定的范围。
【主权项】
1. 一种基于自适应切换分析稀疏与合成稀疏正则化的图像复原方法,其特征在于,首 先,对高质量的训练样本构成的集合聚类并为每一类样本训练一个合成稀疏子字典和一个 分析稀疏子字典;然后,将训练的字典分别用于复原低质量的训练图像样本集合,统计各类 采用每一种子字典的复原误差,根据误差大小自适应切换选择稀疏正则化;最后,在切换准 则基础上建立了图像重建复原模型,并采用交替优化策略求解该模型。2. 根据权利要求1所述的基于自适应切换分析稀疏与合成稀疏正则化的图像复原方 法,其特征在于,所述合成稀疏子字典和分析稀疏子字典,进一步包括:将高质量的训练样 本图像进行分块,分成大小为,的相互重叠若干像素的图像子块,重叠像素个数小于 P,子块向量化后表示为XleRp;首先,对输入的Μ个图像子块进行K均值聚类,采用主分量分 析法为每一类训练一个合成稀疏子字典,采用序列最小化特征值方法为每一类训练一个分 析稀疏子字典,得到Κ个类中心、合成稀疏子字典和分析稀疏子字典,分别记为μ?、Φ?,Ω η,πι = 1.·_Κ〇3. 根据权利要求1所述的基于自适应切换分析稀疏与合成稀疏正则化的图像复原方 法,其特征在于,所述确定稀疏正则化切换准则,进一步包括:输入训练低质量图像集合,对 于图像去模糊,低质量图像是模糊图像;对于超分辨率重建,低质量图像是低分辨率图像经 插值放大后与目标图像相同大小的图像;对每一幅图像分成大小为的子块,由这些 子块构成子块集合,将子块与Κ个类中心具有最小欧氏距离的类作为该子块的类索引knh = 1···κ,为子块集合中的每个子块确定类索引;对类索引为ki的子块分别按合成稀疏子字 典?^、分析稀疏子字典^\复原该子块,分别计算并记录合成稀疏、分析稀疏复原误差,为 子块集合中的每个子块记录复原误差;统计子块集合中具有相同类索引的子块的合成稀 疏、分析稀疏复原误差;如果该类索引的合成稀疏复原误差更小,则选择合成稀疏正则化复 原该类子块,否则选择分析稀疏正则化复原该类子块。4. 根据权利要求1所述的基于自适应切换分析稀疏与合成稀疏正则化的图像复原方 法,,其特征在于,图像复原重建,进一步包括:在合成稀疏正则化和分析稀疏正则化的共同 作用下,图像的复原重建模型表示为:式(1)中,h是子块i所属类的索引,αι和砵Λ分别是子块i的合成稀疏编码与分析稀疏 编码,RlX表示从图像X中提取子块i,β,λ1,γl是正则化系数,||·|| q(q是0<q<l的整数) 是q范数;为切换开关,当及=1,式(1)变成合成稀疏正则化模型;当\ =0,式(1)变成分析 稀疏正则化模型;在切换机制下,对于给定的子块,式(1)中的两个正则项只有一个被激活; 采用交替优化策略近似求解式(1);求解过程为: 1) 输入:图像y; 2) 初始化:迭代次数s = 0,图像子块数Np,图像中的像素个数NI,最大外循环次数〇_ Max_iter,最大内循环次数;[_]\&?_;^61·,外循环迭代终止误差o_e,内循环迭代终止误差i_ e,期望的协稀疏度?,初始值x(()); 3) 当i <Np时 ⑷令Υι=ΚιΧω;找出与71具有最短欧氏距离的类中心義,kl为 71的类索引,得到对应的 合成与分析子字典Ω . (b)在当前类的切换函数基础上执行下列情况之一: 当\ = 0:时,取q = 〇,采用贪婪梯度投影法求解; 当.\ =1时,取q=l,其中soft( ·,·)是软阈值运算; 循环变量i增1; 4) 置 u = 0,x(u) = x(s); 5) 更新 x(u);'是否成立,不成立则返回步骤5),否则 到步骤7); 7)更新 X(s+1) = x(u+1);、是否成立,不成立则返回步骤3),否则到步 骤9); 9)输出图像x = x(s+1)。
【文档编号】G06T5/00GK105825473SQ201610153994
【公开日】2016年8月3日
【申请日】2016年3月17日
【发明人】陈华华, 吴志坚, 严军荣
【申请人】三维通信股份有限公司
网友询问留言 已有0条留言
  • 还没有人留言评论。精彩留言会获得点赞!
1