周 國 棟
(中南大學機電工程學院 湖南 長沙 410083) (湖南廣播電視大學網絡技術學院 湖南 長沙 410004)
在圖像存儲和處理過程中,不可避免地含有噪聲,去除噪聲、提高圖像質量是圖像處理流程中非常重要的預處理環(huán)節(jié)。但是噪聲和圖像邊緣都屬于高頻信號,去噪的同時很容易破壞邊緣信息??傋兎旨s束去噪方法利用圖像本身的梯度信息,產生各向異性的去噪效果,去噪的同時還能有效保護邊緣和細節(jié)[1]。因此,受到眾多研究者的關注。
但是,總變分去噪方法有一個缺陷,就是在圖像平滑區(qū)域會產生階梯效應,出現(xiàn)虛假邊緣,給后續(xù)的分析、識別等處理工作帶來干擾。針對總變分去噪的階梯效應問題,很多研究者提出了優(yōu)化改進方法,主要思路都是優(yōu)化正則項,具體辦法是增大領域的擴散范圍,以減少階梯效應。比如利用高階微分[2];重疊組[3];非局部梯度[4]、高斯曲率[5]、趨勢保真項[6]、方向鄰域[7]等方法。
以上方法的實質都是利用更大的領域信息來重構像素點,以減少局部平滑區(qū)域梯度趨近零而帶來的階梯效應。但增大領域范圍,可能會導致圖像邊緣像素不沿切線方向擴散,從而帶來模糊或者虛影。
從當前研究來看,試圖通過一種統(tǒng)一的方法來解決階梯效應和邊緣保護的問題,效果往往不理想。因此,本文轉換思路,采用一種混合正則項的方法來去噪,在梯度小的平滑區(qū)域,增大領域范圍,在梯度大的邊緣區(qū)域,縮小領域范圍。
噪聲模型的表達式可以寫成I=u+n。其中:u代表原始圖像;n代表附加噪聲;I代表污染后的圖像。實際應用中,噪聲多為高斯噪聲[8],因此本文只考慮高斯噪聲的去除。
綜合現(xiàn)有總變分去噪原理,本文提出新的去噪模型的表達式如下:
(1)
式中:第一項為正則項,用來約束重構圖像u的梯度和(總變分)最小,從而最接近自然圖像;第二項為保真項,用來約束重構圖像要接近原始圖像I,從而不至于失真;λ為拉格朗日算子,其歐拉-拉格朗日方程如式(2)所示。
λ(u-I)-div(φ(▽u))=0
(2)
式中:div、▽分別為散度和梯度。針對φ(·)函數(shù)的改進是研究最多的地方,目前主流有兩種,分別是1范數(shù)|▽u|和2范數(shù)‖▽u‖2。1范數(shù)的擴散范圍小,可以很好地保護圖像邊緣,但是在平滑區(qū)域會引起階梯效應;2范數(shù)的領域范圍較大,可以較好地抑制平滑區(qū)域的階梯效應,但是會帶來圖像邊緣的模糊。
φ(·)函數(shù)的兩種形式各有利弊,又互相沖突,本文試圖把兩者結合起來,在平滑的區(qū)域使用2范數(shù),在邊緣區(qū)域使用1范數(shù),就可以實現(xiàn)在去噪的同時,做到抑制階梯效應和保護圖像邊緣的效果。具體可以描述如下:
(3)
式中:T是閾值,當像素的梯度大于該值時,可以認定為邊緣,否則認定為平滑區(qū)域。接下來的任務是如何確定梯度閾值T。T的取值跟圖像的梯度均值有關,而噪聲屬于高頻信號,會增加整個圖像的梯度值,因此,準確估計出圖像的噪聲水平,并研究其與T的關系至關重要。
圖像可以被分解為低頻和高頻部分,低頻部分保留著圖像的大部分信息,而高頻部分則主要包含噪聲。因此,噪聲水平可以利用小波變換的高頻子帶系數(shù)來進行預測。Donoho等[9]給出的高斯噪聲估計方法如下:
(4)
式中:σesti表示預測的噪聲標準差;Y(i,j)表示小波變換的HH子帶系數(shù);Median()為Y(i,j)絕對值的中值,變換用的小波為D4編碼。
在使用Donoho方法預測噪聲水平時,發(fā)現(xiàn)一個現(xiàn)象:噪聲方差越大,預測值越精確,偏差越?。辉肼暦讲钤叫?,預測值偏差越大。實驗數(shù)據(jù)如表1所示。
表1 Donoho方法噪聲預測結果
其變化曲線如圖1所示。
圖1 噪聲標準差預測值與誤差關系曲線
預測誤差隨著噪聲的增強而逐漸減小,其原因是小波變換提取的HH子帶不只有噪聲,還包含了圖像本身的邊緣信息。噪聲小時,邊緣占比大,導致預測誤差大,隨著噪聲不斷增強,圖像邊緣占比越來越小,預測值也逐漸收斂到真實值。因此,可以改進Donoho算法,以更準確獲得污染較輕圖像的噪聲預測。
從圖1的曲線可以看出,誤差曲線形狀接近指數(shù)函數(shù),可以用指數(shù)函數(shù)進行擬合。因此,擬合方程可以描述為:
(5)
式中:σDon表示Donoho算法預測的噪聲值;σreal表示真實的噪聲值。通過對大量實驗數(shù)據(jù)的擬合,得到了擬合指數(shù)函數(shù)的參數(shù)值:
C=18.12k=-2.29
(6)
最后,根據(jù)式(4)-式(6),得到改進的Donoho預測算法,記作IDon。
(7)
為驗證IDon算法的適用性,選取前述實驗圖像以外的圖像進行實驗,選取Airport(512×512)、Sailboat(512×384)、Parrot(512×384)、Boat(256×256)等四種不同類別和尺寸圖像進行實驗。分別給這些圖像添加標準差為2、10和18 dB的噪聲,然后用Don和IDon算法分別進行預測,結果如表2所示。可以看出,IDon算法預測的噪聲誤差有很明顯的下降。
表2 IDon和Don預測誤差性能對比
為了綜合1范數(shù)和2范數(shù)的優(yōu)點,本文設計出新的總變分模型,也就是式(1)。接下來研究該公式的迭代算法。
目前解決總變分代價函數(shù)的最小值問題,主流方法是交替方向乘子(ADMM)法,為了讓式(1)具有更好的凸性,在迭代過程中能快速收斂,本文將式(1)進行適當改造,添加兩個輔助變量,分離出為φ函數(shù),得到式(8)。
(8)
迭代過程是交替進行的,先計算wi+1,再計算zi+1,最后計算ui+1。具體過程如下。
首先是wi+1,因為φ函數(shù)可能有兩個結果,因此wi+1有兩種可能情況。
當φ=2時,表達式為:
(9)
當φ=1時,表達式為:
(10)
同樣地,z的迭代公式如下:
(11)
最后是u的迭代計算:
(12)
為減小大型矩陣相乘的計算量,利用傅里葉變換對式(12)進行簡化,得到u的迭代公式如下:
(13)
式中:F-1為傅里葉反變換運算;G、Z、D、W分別表示g、z、▽、w的傅里葉變換矩陣。
其處理流程如下。
1) 初始化參數(shù),取α=β=1,xmax=107,迭代次數(shù)n=1,最小誤差ξ=10-4。
2) 預測噪聲標準差σimproved。
3) 計算閾值T。
5) 如果|▽u|≥T,按照式(10)計算wi+1;否則,按照式(9)計算wi+1。
6) 按照式(11)計算zi+1。
7) 按照式(13)計算ui+1。
9) 如果β 實驗分兩部分來完成,首先是閾值T的確定。方案是通過4幅有代表性的圖像分別進行閾值測試,得到不同噪聲水平下的最佳閾值序列,并進行擬合,最后由4條擬合曲線求均值,得到最終閾值。 第二部分實驗是驗證本文方法的去噪效果,實驗采用對比驗證法,通過與常見的去噪方法進行對比,包括傳統(tǒng)全變分方法(TV)、三維塊匹配去噪算法(BM3D)[10]、非局部平均去噪(NL-Means)[11]方法等,得到最終的數(shù)據(jù)。 以Lena為例,首先在圖像上添加標準差(σ)為20的高斯噪聲,然后使用標準差為0~100、間隔為1的閾值數(shù)列,使用本文的去噪方法分別進行去噪,并計算去噪后圖像的峰值信噪比(PNSR)。最終得到的閾值和峰值信噪比關系曲線如圖2所示。 圖2 閾值T和PNSR關系曲線 可以看出,當閾值在24.1時,PNSR獲得最大值,去噪效果最好。 按照以上方法,分別添加不同級別的噪聲(標準差分別為4、8、12、16、20、24、28、32),并獲取對應噪聲估計值和最佳閾值,繪制出4個圖像(Boat、Bridge、Lena、Peppers)的不同點位圖,觀察該圖像近似為直線,因此擬合出直線方程,公式為: T=kiσ+bi (14) 擬合后的參數(shù)如表3所示,各個擬合圖如圖3所示。 表3 擬合直線參數(shù)表 圖3 最優(yōu)閾值T和估計噪聲擬合直線 最后對ki和bi取算術平均值,得到最終結果: (15) T=1.157σimproved (16) 為了驗證本文方法在其他圖像上的去噪效果,使用另外的三幅圖像(Barbara、Baboon、Lake)進行對比驗證實驗。分別添加均值為0,標準差為10、20、30和40的高斯噪聲。再分別用傳統(tǒng)全變分方法(TV)、BM3D、非局部平均去噪(NL-Means)方法等進行對比驗證。實驗對應的數(shù)據(jù)結果如表4所示,Barbara的去噪效果對比如圖4所示,Lake的去噪效果對比如圖5所示。 表4 去噪實驗結果對照表 圖4 Barbara去噪效果對比 圖5 Lake去噪效果對比 從去噪效果看,傳統(tǒng)TV算法去噪后有較明顯的局部階梯效應,NLM算法沒有有效區(qū)分噪聲和圖像,因此去噪效果一般,BM3D從整體來看有較好的去噪效果,某些圖像的去噪表現(xiàn)上是最好的,但是BM3D在眼部細節(jié)上丟失了細節(jié),導致了模糊。本文方法整體上表現(xiàn)突出,在邊緣區(qū)域和平滑區(qū)域都處理較好,對階梯效應的抑制較好。 分析了全變分圖像去噪的原理、階梯效應的形成原因。針對現(xiàn)有方法不能有效解決邊緣保護與平滑區(qū)域階梯效應的矛盾問題,提出一種利用局部梯度閾值有效調和現(xiàn)有矛盾的混合式解決辦法。并提出噪聲標準差的預測改進算法,進而提出局部梯度閾值的確定方法,最后給出完整的去噪算法步驟。實驗證明:1) 本文方法與傳統(tǒng)TV去噪算法相比,性能有了較大提升,峰值信噪比平均提高了1.74 dB。2) 與同類優(yōu)秀的去噪算法相比,本文方法在保護邊緣和抑制階梯效應上,有較好的表現(xiàn)。4 實 驗
4.1 閾值的確定
4.2 去噪性能對比實驗
5 結 語