快速方向字典插值的中低倍率超分辨率算法*

王鐘斐，王 彪

寶雞文理學(xué)院 數(shù)學(xué)與信息科學(xué)學(xué)院，陜西 寶雞 721013

1 引言

隨著人們對(duì)視頻和顯示質(zhì)量需求的逐漸增加，作為視頻質(zhì)量的重要特性，視頻的圖像分辨率已經(jīng)逐漸過(guò)渡到目前主流市場(chǎng)的4K水平[1-2]。為了滿足越來(lái)越高的圖像分辨率，終端顯示設(shè)備的分辨率也逐漸加大。但是由于陳舊視頻源的分辨率較低，且低分辨率（low resolution，LR）視頻源在高分辨率（high resolution，HR）顯示設(shè)備上不能得到較好的顯示效果，因此視頻源從LR向HR轉(zhuǎn)化的超分辨率[3-4]（super resolution，SR）技術(shù)的研究被廣泛關(guān)注。

基于當(dāng)前市場(chǎng)需求，SR應(yīng)用主要分為兩類：一類為視頻源從LR直接轉(zhuǎn)化為HR后，通過(guò)網(wǎng)絡(luò)傳輸且顯示[5-7]；另一類為L(zhǎng)R視頻源通過(guò)網(wǎng)絡(luò)傳輸后，在顯示端轉(zhuǎn)化為HR后顯示。可以看出，后者在提高視頻分辨率的同時(shí)可以節(jié)約大量的網(wǎng)絡(luò)帶寬。因此顯示端的實(shí)時(shí)SR技術(shù)成為研究的重中之重[8-10]。

目前SR技術(shù)主要從現(xiàn)實(shí)應(yīng)用角度分為兩種，分別為實(shí)時(shí)系統(tǒng)中SR和非實(shí)時(shí)系統(tǒng)中SR，非實(shí)時(shí)系統(tǒng)中SR代表算法為基于深度學(xué)習(xí)的方法[11-12]，實(shí)時(shí)系統(tǒng)中SR代表算法為基于方向插值的方法[13-16]。在基于學(xué)習(xí)的方法中，典型算法為稀疏矩陣的字典算法，該方法認(rèn)為L(zhǎng)R和HR存在映射關(guān)系，通過(guò)大量LR和HR數(shù)據(jù)的訓(xùn)練學(xué)習(xí)，找到最優(yōu)的映射矩陣。圖像由LR到HR的轉(zhuǎn)化過(guò)程中，通過(guò)查找對(duì)應(yīng)類型的映射矩陣，實(shí)現(xiàn)圖像的SR重建。該方法為了獲取最優(yōu)的圖像質(zhì)量，需要存儲(chǔ)海量的映射矩陣和邊信息，并且運(yùn)算復(fù)雜度極高。由于該方法的海量存儲(chǔ)導(dǎo)致實(shí)現(xiàn)成本增加，且運(yùn)算復(fù)雜度過(guò)大不能達(dá)到實(shí)時(shí)的要求，因此該方法目前不適于工業(yè)量產(chǎn)。在基于插值的方法中，典型算法為方向插值算法，該方法首先判斷紋理方向，然后根據(jù)紋理方向進(jìn)行插值，得到SR重建圖像。該方法通過(guò)運(yùn)算復(fù)雜度和圖像質(zhì)量的均衡，在控制成本的前提下，追求圖像質(zhì)量更優(yōu)。

由于目前片源和顯示設(shè)備的分辨率倍數(shù)差異不會(huì)過(guò)大，并且考慮到成本，大多實(shí)時(shí)顯示系統(tǒng)的SR分為2～3倍的低倍率和4～5倍的中倍率。

2018年，Kim等提出了基于線性映射的邊緣定位超分辨率算法（super resolution using edge orientation based on linear mapping，SREO）[17]。目前該算法為實(shí)時(shí)中低倍率SR系統(tǒng)的最優(yōu)算法。該算法首先在LR圖像中根據(jù)紋理邊緣的計(jì)算得到大量的紋理方向，然后利用這些紋理方向和HR圖像進(jìn)行映射，訓(xùn)練得到對(duì)應(yīng)的映射矩陣，最終通過(guò)映射矩陣查表的方式進(jìn)行SR的圖像重建。雖然SREO在圖像質(zhì)量和成本復(fù)雜度之間進(jìn)行平衡，以達(dá)到實(shí)時(shí)性和實(shí)現(xiàn)量產(chǎn)的目的，但是該方法仍存在以下缺點(diǎn)：

（1）較經(jīng)典的稀疏矩陣的字典算法，由于SREO的映射矩陣數(shù)量過(guò)少，導(dǎo)致圖像質(zhì)量下滑嚴(yán)重。

（2）較方向插值算法，SREO的映射矩陣需要大量的存儲(chǔ)空間，且浮點(diǎn)型矩陣映射運(yùn)算的復(fù)雜度極高，因此實(shí)現(xiàn)成本大幅度增加。

基于以上問(wèn)題，針對(duì)實(shí)時(shí)顯示系統(tǒng)，本文提出了一種紋理字典和全方向插值的中低倍率快速超分辨率算法（directional dictionary and fitting interpolation，DDFI）。該算法通過(guò)金字塔紋理字典和方向矯正，為每個(gè)待插值像素（interpolated pixel，IP）的位置計(jì)算插值方向，并通過(guò)單邊擬合的方向插值得到IP的像素值，最終在進(jìn)一步提高圖像質(zhì)量的同時(shí)，減少運(yùn)算復(fù)雜度。

2 SREO算法

2.1 算法介紹

以2倍的低倍率SR為例，實(shí)際工作之前，SREO在先驗(yàn)的原始HR圖像和LR圖像之間，計(jì)算得到625組映射矩陣模型。實(shí)際工作之后，SREO首先對(duì)LR圖像的每個(gè)像素，在625組映射矩陣模型中查找對(duì)應(yīng)的模型；然后采用該模型，對(duì)該LR像素進(jìn)行映射得到對(duì)應(yīng)的HR像素；依次循環(huán)，最終得到SR重建圖像。具體步驟如下：

步驟1將先驗(yàn)的HR圖像，以固定倍率進(jìn)行下采樣得到LR圖像。

研究表明，草33薄層稠油油藏在水平井井網(wǎng)條件下具備蒸汽驅(qū)可行性。但為了穩(wěn)妥起見(jiàn)，建議實(shí)施井組礦場(chǎng)試驗(yàn)，在取得經(jīng)驗(yàn)的基礎(chǔ)上大規(guī)模推廣，同時(shí)在試驗(yàn)過(guò)程中需要加大監(jiān)測(cè)資料的錄取和分析，及時(shí)配套工藝技術(shù)解決現(xiàn)場(chǎng)的開(kāi)發(fā)問(wèn)題。

步驟2如圖1所示，通過(guò)HR圖像中IP位置反推，得到LR圖像中基準(zhǔn)像素位置。將基準(zhǔn)像素和其周圍共9個(gè)像素抽取，形成1個(gè)3×3的大塊后，分割為4個(gè)2×2的子塊。

Fig.1 Direction calculation diagram using 3×3 pixels in SREO圖1 SREO中3×3像素方向計(jì)算示意圖

步驟3根據(jù)式（1）對(duì)每個(gè)子塊進(jìn)行水平梯度gh和垂直梯度gv的計(jì)算。與圖1對(duì)應(yīng)，式中表示第i個(gè)子塊中第x列第y行的像素。

將計(jì)算結(jié)果帶入式（2），計(jì)算邊緣方向。式中，mi和di分別是第i個(gè)子塊的梯度權(quán)重和梯度斜率。

根據(jù)計(jì)算結(jié)果和預(yù)定義閾值，在表1中查詢得到每個(gè)子塊的紋理方向和索引。

Table 1 SREO subpatch direction map table表1 SREO子塊方向映射表

步驟4將4個(gè)子塊的索引組成2×2矩陣X。設(shè)HR圖像的2×2矩陣為Y，映射矩陣為M。為了最小化M和Y之間的誤差，考慮Y的多樣化，采用多變量回歸模型，通過(guò)式（3）計(jì)算一個(gè)最優(yōu)的M，其中λ為歸一化參數(shù)，I為正則矩陣。

由于每個(gè)子塊的方向可能性為5種，因此對(duì)于2倍率SR，4個(gè)子塊的方向組合為54種，即可得到625種映射矩陣模型。

在所有映射矩陣模型得到后，SREO實(shí)際工作時(shí)，首先對(duì)于HR圖像中IP，計(jì)算LR圖像中對(duì)應(yīng)的待插值像素映射（interpolated pixel map，IPM）位置；然后采用式（2）計(jì)算基準(zhǔn)像素的mi和di，形成索引矩陣后，查表選取對(duì)應(yīng)的映射模型進(jìn)行映射；最終得到SR后的HR像素。以此類推，若為4倍SR，則映射矩陣數(shù)量為516。

2.2 問(wèn)題分析

根據(jù)以上算法介紹，可以看出SREO有如下問(wèn)題：

（1）圖像質(zhì)量角度。SREO僅采用5種紋理方向，遠(yuǎn)不能夠滿足自然界的紋理復(fù)雜情況，2倍率SR下SREO的625組映射矩陣，較經(jīng)典的稀疏矩陣的字典算法，也不能使映射矩陣達(dá)到最優(yōu)。因此SREO的圖像質(zhì)量較稀疏矩陣的字典算法有較大差距。

（2）實(shí)現(xiàn)成本角度。2倍率SR下，SREO的625組映射矩陣需要大量的存儲(chǔ)空間，且每個(gè)像素浮點(diǎn)型矩陣映射的運(yùn)算復(fù)雜度偏高。因此較插值算法，SREO實(shí)現(xiàn)的成本大幅度增加。

（3）規(guī)格角度。SREO針對(duì)2倍的低倍率SR進(jìn)行設(shè)計(jì)，對(duì)于中倍率情況，不僅較少的紋理方向會(huì)引起重建HR圖像質(zhì)量的下降，而且成指數(shù)級(jí)增長(zhǎng)映射矩陣的數(shù)量也會(huì)使SREO的實(shí)現(xiàn)成本不可接受。

針對(duì)以上問(wèn)題，本文提出了DDFI算法。

3 DDFI算法分析

3.1 金字塔字典映射模型

如圖2所示，根據(jù)在HR圖像中的IP確定其在LR圖像中的IPM，并以由IPM確定的Pix22、Pix23、Pix32和Pix33構(gòu)成的2×2區(qū)域?yàn)椴逯祬^(qū)域。將該2×2區(qū)域分別拓展為4×4和6×6區(qū)域，形成3層金字塔模型。先根據(jù)式（4）計(jì)算得到3層的4個(gè)方向的梯度權(quán)重，式中i和j分別為像素點(diǎn)的行列坐標(biāo)，k為層數(shù)。再根據(jù)式（5）計(jì)算得到4組梯度比。

Fig.2 Pyramid model圖2 金字塔模型

根據(jù)層數(shù)約束后，0層、1層和2層的4組梯度比分別有16、81和256種組合，3層共有331 776種組合。

HR圖像中的IP紋理方向dirt計(jì)算公式如式（6），dirt為某一方向的梯度權(quán)值，IPi為當(dāng)前IP，IPi-n和IPi+n為當(dāng)前IP按該方向的前后兩個(gè)IP，m和n為參數(shù)。最終dir對(duì)應(yīng)的角度為該IP的紋理角度。

為了確定LR圖像中IPM的331 776種組合和HR圖像中IP的360個(gè)角度的對(duì)應(yīng)關(guān)系，在固定SR倍率下，首先仿真約100萬(wàn)幅HR圖像和之對(duì)應(yīng)的LR圖像，得到IPM每個(gè)組合對(duì)應(yīng)的所有IP角度；然后對(duì)于1種IPM組合，在其對(duì)應(yīng)的所有IP角度中，選取出現(xiàn)概率最大的角度，作為該IPM組合的對(duì)應(yīng)角度；以此類推，得到所有IPM組合的對(duì)應(yīng)角度。最終，在實(shí)際工作之前形成一個(gè)331 776組合和360°的方向映射表。

3.2 IP的插值方向

實(shí)際工作中，首先確定IP的插值方向，具體如下：

步驟1插值區(qū)域主方向的確定

由于自然圖像中大部分區(qū)域?yàn)槠教箙^(qū)域，為了降低運(yùn)算復(fù)雜度，在IPM的插值區(qū)域中，當(dāng)均小于預(yù)設(shè)閾值時(shí)，確定該區(qū)域無(wú)方向；否則根據(jù)式（5）計(jì)算4組梯度比，通過(guò)方向映射表得到當(dāng)前插值區(qū)域的主紋理方向。

對(duì)于無(wú)主紋理方向的插值區(qū)域，采用雙線性插值的方法得到IP，否則進(jìn)入步驟2。

步驟2IP方向矯正

對(duì)于存在紋理主方向區(qū)域，由于插值區(qū)域中所有IPM并非嚴(yán)格同一方向，而是基于一個(gè)方向的方向簇，因此對(duì)于任1個(gè)IPM，需要考慮其相鄰插值區(qū)域的主紋理方向。如圖3所示，插值區(qū)域分為8個(gè)子區(qū)域，每個(gè)子區(qū)域需參考1個(gè)直相鄰和1個(gè)斜相鄰區(qū)域的主紋理方向。設(shè)插值區(qū)域?yàn)檎叫?，且每個(gè)邊長(zhǎng)為100，可得對(duì)角長(zhǎng)度為141。設(shè)t為IPM到當(dāng)前插值區(qū)域斜對(duì)角線的垂直距離，a為IPM到當(dāng)前插值區(qū)域中線垂直距離。根據(jù)式（7）計(jì)算得到IPM的具體方向。

Fig.3 Direction correction圖3 方向糾偏

式（7）中，條件為相鄰區(qū)域主紋理方向?yàn)闊o(wú)方向或相鄰區(qū)域與當(dāng)前插值區(qū)域主紋理方向差異超過(guò)45°。dcur、dadj和ddia分別是當(dāng)前、直相鄰和斜相鄰的區(qū)域的主紋理方向。

3.3 擬合插值

對(duì)于高次運(yùn)算插值算法，如雙立方插值（bicubic interpolation，BC），雖然圖像質(zhì)量更優(yōu)，但是復(fù)雜度過(guò)高；當(dāng)中低倍率SR時(shí)，相對(duì)高次運(yùn)算插值算法、低次運(yùn)算插值算法效果相近且復(fù)雜度更低。對(duì)于存在紋理主方向區(qū)域，考慮到算法的實(shí)時(shí)性和低復(fù)雜度，為了擬合曲線，本文采用單邊擬合插值的方法。如圖4所示，先利用水平或垂直單邊擬合插值計(jì)算得到IPM方向上4個(gè)子像素點(diǎn)，再利用這4個(gè)子像素點(diǎn)，進(jìn)行第二次單邊擬合插值，最終得到IPM對(duì)應(yīng)的IP。單邊擬合插值公式如式（8）：

Fig.4 Interpolation schematic diagram圖4 插值示意圖

根據(jù)以上算法原理描述，可以看出本文算法較傳統(tǒng)插值算法，引入金字塔方向字典映射，在SR實(shí)際工作之前，獲得更準(zhǔn)確的方向查詢模型，在SR實(shí)際工作中可得到更準(zhǔn)確的紋理方向判定。較SREO，不但具有更多的紋理主方向，而且插值區(qū)域內(nèi)形成了多方向的方向簇，更符合自然圖像形成規(guī)律，而且運(yùn)算復(fù)雜度較低。

3.4 復(fù)雜度分析

以2倍SR為例，SREO對(duì)于每個(gè)IPM，索引矩陣的形成需要28個(gè)加法器，4個(gè)乘法器，1個(gè)求根處理和1個(gè)查表運(yùn)算；查表之后的矩陣映射運(yùn)算需要8個(gè)乘法器和4個(gè)加法器。平均求根和查表處理運(yùn)算后，SREO共需要約50個(gè)乘法和32個(gè)加法。本文算法對(duì)于每個(gè)IPM，當(dāng)具有紋理方向時(shí)，運(yùn)算復(fù)雜度最高，方向判定階段最高需要約120個(gè)加法器，15個(gè)乘法器和1個(gè)查表運(yùn)算；插值階段需要10個(gè)乘法器和5個(gè)加法器。平均查表處理運(yùn)算后，本文共需要約50個(gè)乘法和32個(gè)加法，共需要40個(gè)乘法器和125個(gè)加法器。可以看出，本文算法的最大計(jì)算量與SREO基本持平，但是本文算法對(duì)于所有計(jì)算，均限制最大8比特精度，遠(yuǎn)小于SREO的16比特精度，因此本文算法的計(jì)算復(fù)雜度較SREO更低。

4 實(shí)驗(yàn)結(jié)果與分析

本文根據(jù)圖像質(zhì)量和運(yùn)算復(fù)雜度對(duì)算法進(jìn)行測(cè)試，其中圖像質(zhì)量包括主客觀圖像質(zhì)量，客觀圖像質(zhì)量分為PSNR和SSIM；運(yùn)算復(fù)雜度以運(yùn)行時(shí)間（run time，UT）進(jìn)行衡量。對(duì)比算法選取典型插值算法BC（bilinear interpolation）、典型深度學(xué)習(xí)算法RAISR（rapid and accurate image super resolution）[18]和SREO。SR倍數(shù)分別設(shè)置為2倍和4倍，運(yùn)行平臺(tái)為Intel I5-6400 CPU@2.7 GHz-8 GB內(nèi)存-Win 7 64位系統(tǒng)。選取的測(cè)試序列為經(jīng)典測(cè)試序列和如圖5所示的實(shí)際采集到的測(cè)試序列。

實(shí)驗(yàn)結(jié)果如表2所示。較BC，由于有了紋理的方向判定，圖像質(zhì)量更有大幅度提高，平均PSNR和SSIM分別提高1.4 dB和0.047；運(yùn)算復(fù)雜度方面，本文算法雖然對(duì)紋理區(qū)域的方向判定消耗了大量的運(yùn)算，但是平坦區(qū)域不需要方向判定且插值方法簡(jiǎn)單，而且擬合插值算法較BC的3次運(yùn)算也節(jié)省了大量的運(yùn)算量，因此平均運(yùn)算復(fù)雜度為BC的58.3%。

較RAISR，本文算法雖然PSNR和SSIM提高較小，但是算法運(yùn)算量?jī)H約為RAISR的2%。即無(wú)論在硬件系統(tǒng)還是軟件工程，RAISR目前是無(wú)法做到實(shí)時(shí)計(jì)算顯示的。且由于RAISR需要海量數(shù)據(jù)的存儲(chǔ)，也是不可工業(yè)化實(shí)現(xiàn)的。

Fig.5 Captured images sequence圖5 采集到的圖像序列

Table 2 Performance comparison among BC，RAISR，SREO and the proposal表2 BC、RAISR、SREO和本文算法的性能比較結(jié)果

較SREO，由于本文算法采用更高準(zhǔn)確率的紋理方向判定與擬合插值算法，因此平均PSNR和SSIM分別提高0.73 dB和0.042；由于SREO的4倍SR較2倍SR矩陣浮點(diǎn)乘運(yùn)算呈指數(shù)級(jí)增加，因此4倍SR時(shí)，本文算法運(yùn)算量?jī)H約為SREO的56.8%。

主觀圖像質(zhì)量對(duì)比如圖6所示，較BC和SREO，可以看出本文算法在紋理細(xì)節(jié)區(qū)域處理得更好，對(duì)于圖中頭發(fā)區(qū)域，本文算法的細(xì)節(jié)紋理更加豐富銳利。

5 總結(jié)

針對(duì)SREO的問(wèn)題，為了進(jìn)一步提高重建圖像質(zhì)量，并且降低運(yùn)算復(fù)雜度，提出了DDFI算法。DDFI首先采用金字塔紋理字典得到插值區(qū)域紋理主方向，然后根據(jù)IPM的位置矯正得到IP的紋理方向，最后采用單邊擬合的方向插值算法計(jì)算IP值。實(shí)驗(yàn)結(jié)果顯示，與SREO相比，本文算法的PSNR和SSIM平均分別提高了0.73 dB和0.04，同時(shí)大幅度降低了運(yùn)算復(fù)雜度。

Fig.6 Subjective image quality comparison among BC，RAISR，SREO and the proposal圖6 BC、RAISR、SREO和本文算法的主觀圖像質(zhì)量對(duì)比