一種改進的暗通道先驗圖像去霧算法研究

高強

摘要：傳統的暗通道先驗去霧算法通過軟摳圖方法來優化大氣透射率，得到較好的去霧結果，但該算法存在復雜度高，去霧后圖像邊緣信息不足夠豐富、對比度不高等問題。針對上述問題提出一種改進的梯度引導算法，該算法在引導濾波器中構造一個能量函數，并在能量函數中加入一個邊緣保持項，以此來進一步優化大氣透射率。實驗結果表明，使用改進后的算法運算速度快，增加去霧后圖像的清晰度和對比度。

關鍵詞：圖像去霧;暗通道先驗;梯度引導算法

中圖分類號：TP391? ? ? ? 文獻標識碼：A

文章編號：1009-3044（2020）26-0017-03

Abstract：The traditional defogging algorithm based on dark channel prior optimizes the atmospheric transmittance by soft matting method.The result is good.However，there are still some problems in the algorithm， such as high complexity，low contrast，low definition.An improved gradient guided algorithm is proposed in order to solve the problems，which constructs an energy function in the guided filter.There is a special item in the energy function that can further improve the atmospheric transmittance.The experimental results show that the improved algorithm can increase the definition and contrast

Key words：image dehazing algorithm; dark channel prior; gradient guided filtering

1引言

圖像是現代社會數字信息的重要載體，其結構簡單、表達信息豐富、容易被理解，可由多種不同類型的監測設備、通過各種方式而獲得。在戶外等自然場景下拍攝的圖像，難免會受到霧、霾等不確定性天氣因素的影響，造成獲取的圖像中可見特征信息大幅減少，低對比度、低清晰度、高顏色失真度，邊緣細節模糊，非常不利于圖像信息的處理、應用。因此，圖像去霧方法的研究有十分重要的應用價值[1-2]。

圖像去霧方法主要有兩種，一種是基于圖像增強的方法[3，4]，結合圖像的時域和頻域特征通過霧霾圖像增強算法來提高圖像的清晰度和對比度，但是這種方法并沒有分析霧霾天氣條件下成像降質的本質，不計算圖像場景深度，容易造成去霧后圖像顏色失真、低清晰度、低對比度等現象，并對濃霧霾區域去除效果不明顯[5-7]。另外一種是在分析成像降質的本質的基礎上，通過先驗來估算出大氣透射率[8，9]。He[10]等人結合暗通道先驗方法計算出透射率的大致估計值，然后使用Soft Matting優化透射率，代入成像模型中，得到去霧圖像。通過該算法得到的去霧圖像，與基于圖像增強的方法相比較，去霧結果較好，但是該算法計算量大、處理速度慢，去霧結果圖像不足夠清晰、邊緣信息不足夠豐富。為此，本文在霧霾天氣條件下成像模型的基礎上，使用改進的引導梯度算法代替Soft Matting方法對透射率進行精細優化，降低算法計算量，使去霧后圖像清晰度高、邊緣信息豐富、色彩自然。

2基于暗通道先驗的改進去霧算法

2.1暗通道先驗理論

物體表面接收自然光照射，從而形成反射光。反射光經過霧氣和空中微小顆粒時會發生散射現象，帶走一部分的反射光，但大部分的物體表面反射光反能被成像視覺所接收，同時自然光在經過霧氣和空中微小顆粒后也會發生散射現象，這部分散射光也會被成像視覺接收。根據此現象可以來定義散射的數學模型如下：

在公式（1）中，I（x）為視覺所成像的有霧圖像;J（x）為加去霧算法后的圖像;A為大氣光值;t（x）為原始圖像的大氣透射圖，表示到達視覺鏡頭的光;x表示圖像的坐標。去霧算法的基本原理就是在已知大氣光值A和大氣透射圖t（x）的情況下，根據（1）公式來求解J（x）。

暗通道先驗原理是指在戶外的圖像中，剔除掉天空的區域內，近乎每一個像素都至少會有一個通道有很低的、近乎為0的灰度值，因此可以對原始圖像的暗通道定義為：

通過公式（7）可知，若想求解出[t（x）]，還必須知道大氣光值A。我們采取這樣的辦法來求取A：第一步，求解原始圖像暗通道圖，并對此圖中的像素按照亮度大小來進行排序，記錄排序位于前0.1%的像素點;第二步，對記錄到的像素點，通過最大值濾波法計算出原始圖像的亮度最大的像素點，將該亮度記為大氣光值A。當求解出[t（x）]，就可以根據公式（1）求解出去霧后的圖像J（x）。

2.2 改進算法

由暗通道先驗理論可知，若想求取去霧效果較好的圖像J（x），被記為大氣透射率的[t（x）]尤為關鍵。通過暗通道先驗理論計算出的[t（x）]，在還原去霧后的圖像中，較為容易出現圖像明亮位置顏色偏移和光暈問題，造成還原去霧后圖像視覺模糊，無法保留很多的邊緣信息細節。針對暗通道先驗方法出現的問題，本文對其進行改進，使用改進梯度引導算法對大氣透射率[t（x）]進行精細計算，解決還原去霧后圖像無法保留很多邊緣細節的問題。

假設原始圖像的參考圖像為C（x），經過導向濾波器后的圖像為F（x），F（x）與C（x）有如下的線性關系：

在公式（13）中，[aq]、[bq]分是X正方形區域內的aq、bq的平均值。通過求取平均值的方法，使得原始圖像I的梯度參數與經過導向濾波器后的圖像F在梯度值上較為接近，因此在去霧后會更多地保留很多的邊緣信息細節。

3實驗結果和分析

本實驗通過使用Matlab 2016a在Window10系統環境下來驗證改進算法的有效性。實驗結果如圖1所示。從圖1中（f）和（g）的圖像相比較，改進算法去霧結果圖像清晰，在細節上可觀察出更多的景物邊緣信息，且邊緣信息在景深大致一樣的區域內會顯得較為平滑;去霧后圖像，會顯得清新自然，色彩柔和，不會出現暗通道先驗去霧結果圖像的顏色過飽和與去霧效果不足的現象。

本文引入代碼在Matlab平臺的運行時間和PSNR峰值信噪比，這兩個客觀指標來評價兩種不同算法在去霧結果的優劣。運行時間越短，說明算法的實時性更高，有利于算法的實際應用。PSNR峰值信噪比越大，說明去霧后的圖像與原始無霧圖像更加接近。針對圖1中（d）和（f）去霧結果圖像，這兩個客觀指標的結果如表1所示。

從兩種不同算法的去霧結果綜合來看，改進去霧算法去霧效果優于暗通道先驗去霧效果，去霧后成像結果更加接近原始圖像，結構清晰，對比度高，邊緣細節信息保留多，圖像色彩清新自然真實，有更好的去霧效果，且實時性好。

3小結

基于暗通道先驗理論的去霧算法，在還原去霧后的圖像結果時，透射率精度不高，容易造成圖像結果視覺模糊，無法保留很多的邊緣信息細節，去霧結果不理想。本文針對暗通道先驗去霧算法出現的問題，使用改進后的梯度引導算法對大氣光值進行精細計算，并在MATLAB平臺中進行仿真實驗，對實驗結果進行分析，本文提出的改進算法有較好的去霧效果，結果圖像結構清晰，色彩自然，消除塊效應，保留了較多的邊緣細節，解決還原去霧后圖像無法保留很多邊緣細節的問題。同時，改進算法還提高了運行速度。

參考文獻：

[1] 楊燕，王志偉.基于均值不等關系優化的自適應圖像去霧算法[J].電子與信息學報，2020，42（3）：755-763.

[2] 孫士偉，劉金虎，馬文君，等.基于線性模型的自適應優化去霧算法[J].應用光學，2020，41（1）：114-119.

[3] 齊永鋒，李占華.一種基于多尺度卷積神經網絡和分類統計的圖像去霧霾方法[J].紅外技術，2020，42（2）：190-197.

[4] 徐巖，孫美雙.基于多特征融合的卷積神經網絡圖像去霧算法[J].激光與光電子學進展，2018，55（3）：031012.

[5] He L Y，Zhao J Z，Zheng N N，et al.Haze removal using the difference- structure-preservation prior[J].IEEE Transactions on Image Processing，2017，26（3）：1063-1075.

[6] Zhu S，Cao D H，Wu Y B，et al.Edge-aware dynamic programming-based cost aggregation for robust stereo matching[J].Journal of Electronic Imaging，2015，24（4）：043016.

[7] 楊燕，李一菲，岳輝.一種自適應線性透射率估計去霧算法[J].應用光學，2019，40（3）：447-453.

[9] 王雪梅，鞠銘燁.基于暗通道先驗的快速圖像去霧[J].科學技術與工程，2016，16（20）：66-72.

[10] He K M，Sun J，Tang X O.Single image haze removal using dark channel prior[J].IEEE Transactions on Pattern Analysis and Machine Intelligence，2011，33（12）：2341-2353.

【通聯編輯：唐一東】