專利名稱:一種基于二維分析稀疏模型及其訓(xùn)練字典的圖像重建方法
技術(shù)領(lǐng)域:
本發(fā)明屬于信號建模的技術(shù)領(lǐng)域,具體地涉及一種基于二維分析稀疏模型及其訓(xùn)練字典的圖像重建方法。
背景技術(shù):
信號模型對于處理很多問題起著很重要的作用,比如壓縮,采樣,重建,等等。目前對于信號進行建模的很重要的一種方法就是基于合成模型的稀疏表不方法。合成模型如下所不D α =χ,其中X e Rd, D e Rdxn0彳目號x e Rd被認(rèn)為是給定字典下某些稀疏基a e Rn的線性組合。這種模型在過去的近幾年里,引起了廣泛的關(guān)注。特別是其在圖像壓縮,超分辨率重建,圖像去噪領(lǐng)域應(yīng)用十分廣泛。對于這種合成模型的稀疏表示方法,其研究重點集中在從采樣信號中得到字典,一些稀疏字典的學(xué)習(xí)方法包括K-奇異值分解(K-SVD),稀疏編碼sparse coding,等等,都已經(jīng)比較成熟?;蛘呔褪窃诮o定字典的情況下,信號可以得
到精確的稀疏解,即求解如下問題toy = Dx,其中I I X I I。表示X中非零元素
的個數(shù),用來表示該信號的稀疏度和該信號所屬空間的維度。對于合成模型的稀疏表不方法已經(jīng)有很大的應(yīng)用,但是對于稀疏表不的分析模型則一直被擱置。分析模型表示為如下形式I I Ωχ| Ici = p-ι,其中Ω e Rpxd是一個線性算子,I表征稀疏信號X的聯(lián)合稀疏度。對比與合成模型中使用非零個數(shù)來表示信號的稀疏度,而這里使用領(lǐng)的個數(shù)來表征稀疏度和定義信號所屬空間維度。在合成模型中D為稀疏字典,而分析模型中Ω為字典,對于分析模型中相應(yīng)的重建逆問題為min||x — y\2subject to| | Ωχ| |0 = ρ-1。不過,近年來分析模型的稀疏表示在圖像處理方面也被成功地應(yīng)用于圖像去噪,圖像模糊,以及壓縮感知領(lǐng)域。因此基于分析模型的稀疏表示方法也引起了廣泛關(guān)注。針對該模型的研究主要集中于兩方面,一方面是該模型的應(yīng)用,如Figueiredo提出的利用過完備小波變換進行圖像的重建,分析模型的應(yīng)用主要體現(xiàn)在先對信號應(yīng)用前向變換來得到稀疏系數(shù)。并且他們把分析模型與I1范數(shù)規(guī)范下過完備字典變換應(yīng)用于去噪和圖像復(fù)原。而Portilla將分析模型應(yīng)用于圖像的解卷積和圖像重建,他們把稀疏系數(shù)定義為一個稀疏矢量和其高斯矯正項。最終利用迭代邊緣最小化來解決逆問題。而該模型的另一個研究方面就是分析稀疏字典的學(xué)習(xí)方法。Ophir提出的訓(xùn)練方法,主要是每次訓(xùn)練Ω的一行,來說明和一部分訓(xùn)練集正交的向量的方向。然而,他們的方法是隨機進行初始化,當(dāng)需要訓(xùn)練的字典增大時,則訓(xùn)練的效率會大大提高。Rubinstein提出的訓(xùn)練算法主要包含兩種貪
婪追蹤算法-Back Greedy Algorithm(BGA 反向貪婪算法)和 Optimized Back Greedy
AlgorithmCOBGA,最優(yōu)反向貪婪算法),而且可以使用聯(lián)合稀疏度和誤差兩種方式控制迭代次數(shù),從而完成字典的訓(xùn)練。另外Rubinstein提出的算法是采用字典的支撐集的聯(lián)合秩而不是利用聯(lián)合稀疏度來控制迭代條件,因此輸出稀疏表示的實際為零的個數(shù)大于聯(lián)合秩的個數(shù)。但是,基于傳統(tǒng)分析模型的稀疏表示的建立和求解,都是將圖像塊重新排列為一個一維信號,然后利用上述分析模型來求解重建信號,以及利用大量的重建信號來不斷更新字典的每一行,從而實現(xiàn)分析稀疏字典的學(xué)習(xí),以及完成圖像的重建。然而實際中圖像塊的相關(guān)性是存在方向性的,如果直接將圖像塊按列或者按行進行重排,則會造成不必要的相關(guān)性的引入,如第一列信號的最后一個像素與第二列的第一個像素的相關(guān)性在圖像中是相對比較弱的,但是當(dāng)圖像塊按列重排成一維信號,在訓(xùn)練過程中,必然會認(rèn)為這兩個像素間是強相關(guān)的。這種不必要的相關(guān)性的引入普遍存在于傳統(tǒng)的字典訓(xùn)練中,無法反映圖像的時間空間特性、重建效果差。
發(fā)明內(nèi)容
本發(fā)明的技術(shù)解決問題是克服現(xiàn)有技術(shù)的不足,提供一種更能反映圖像空間特性、重建效果好的基于二維分析稀疏模型及其訓(xùn)練字典的圖像重建方法。本發(fā)明的技術(shù)解決方案是這種基于二維分析稀疏模型及其訓(xùn)練字典的圖像重建方法,包括以下步驟(I)構(gòu)造訓(xùn)練樣本集
權(quán)利要求
1.一種基于二維分析稀疏模型及其訓(xùn)練字典的圖像重建方法,其特征在于包括以下步驟 (1)構(gòu)造訓(xùn)練樣本集//=[y'/2),···,/'· 嚴(yán),其中y⑴表示對圖像進行采樣得到的第i個Cl1Xd1的圖像塊,及##表示實數(shù)域,其維度為屯,M0 =表示圖像塊樣本數(shù)量; (2)構(gòu)造并訓(xùn)練第一個方向的字典Q1:構(gòu)造第一個新的訓(xùn)練樣本集Y=Vyf yf i ^ j ^ d1; Y中每一個元素都是原始圖像塊中的每一列,j是列數(shù),對該訓(xùn)練樣本集采用分析稀疏字典訓(xùn)練方法K-SVD在給定聯(lián)合秩Γι = I約束下訓(xùn)練字典Q1,即求解稀疏系數(shù)以及不斷更新字典Ω i的每一行14^ , 表示Ω i的第Ic1行; (3)構(gòu)造并訓(xùn)練第二個方向的字典Ω2:利用Q1對樣本集II中的每一個圖像塊求解ΩιΥα),得到第二圖像塊,對每一個第二圖像塊進行轉(zhuǎn)置,得到ζα) = (ΩιΥα))τ,即Ζ = [ )……zf],I彡j彡P(guān)l,zf表示第z⑴塊的第j列,j可取的范圍為I彡j彡P(guān)l,即有ζω塊有P1列,對訓(xùn)練集Z采用分析稀疏字典訓(xùn)練方法K—SVD在給定聯(lián)合秩r2 = I約束下訓(xùn)練字典ω2; (4)利用ΩρΩ2和公式(I)求解原稀疏求解中的字典Qci,以便進行一維分析稀疏重建
2.一種基于二維分析稀疏模型及其訓(xùn)練字典的圖像重建方法,其特征在于包括以下步驟 (I)構(gòu)造訓(xùn)練樣本集,其中y(i)表示對圖像進行采樣得到的第i個Cl1Xd1的圖像塊,。表示實數(shù)域,其維度為屯,M0 =表示圖像塊樣本數(shù)量;(2)構(gòu)造并訓(xùn)練第一個方向的字典Q1:構(gòu)造第一個新的訓(xùn)練樣本集F = …ΜΜ)],I ^ j ^ d1; Y中每一個元素都是原始圖像塊中的每一列,j是列數(shù),對該訓(xùn)練樣本集采用分析稀疏字典訓(xùn)練方法K-SVD在給定聯(lián)合秩Γι = I約束下訓(xùn)練字典Q1,即求解稀疏系數(shù)以及不斷更新字典Q1的每一行Hf表示Ωι的第Ic1行; (3)構(gòu)造并訓(xùn)練第二個方向的字典Ω2:利用Q1對樣本集II中的每一個圖像塊求解ΩιΥα),得到第二圖像塊,對每一個第二圖像塊進行轉(zhuǎn)置,得到ζα) = (ΩιΥα))τ,即Z = [z]!)zf-zf-^1], I彡j彡P(guān)1,^ )表示第z(i)塊的第j列,j可取的范圍為I彡j彡P(guān)1,即有z⑴塊有P1列,對訓(xùn)練集Z采用分析稀疏字典訓(xùn)練方法K-SVD在給定聯(lián)合秩r2 = I約束下訓(xùn)練字典ω2; (4)利用Q1和Ω2構(gòu)造二維分析稀疏求解中的Ω
3.根據(jù)權(quán)利要求2所述的基于二維分析稀疏模型及其訓(xùn)練字典的圖像重建方法,其特征在于所述步驟(7)中的一維分析稀疏重建方法為反向貪婪算法BGA或最優(yōu)反向貪婪算法 OBGA。
全文摘要
本發(fā)明公開一種更能反映圖像空間特性、重建效果好的基于二維分析稀疏模型及其訓(xùn)練字典的圖像重建方法,包括(1)構(gòu)造訓(xùn)練樣本集;(2)構(gòu)造并訓(xùn)練第一個方向的字典;(3)構(gòu)造并訓(xùn)練第二個方向的字典;(4)求解原稀疏求解中的字典Ω0,以便進行一維分析稀疏重建;(5)利用Ω0和一維分析稀疏重建方法求解重建值;(6)對重建值進行反操作,得到對應(yīng)于N個圖像塊的重建值,從而得到重建圖像。
文檔編號G06T11/00GK102930573SQ201210433389
公開日2013年2月13日 申請日期2012年11月2日 優(yōu)先權(quán)日2012年11月2日
發(fā)明者施云惠, 齊娜, 尹寶才, 丁文鵬 申請人:北京工業(yè)大學(xué)