王永虎,管慶吉,李勁松,齊 妙
(東北師范大學(xué) 計算機(jī)科學(xué)與信息技術(shù)學(xué)院,長春 130117)
在霧、霾等惡劣天氣下,由于大氣粒子散射作用導(dǎo)致戶外拍攝的圖像質(zhì)量較差,從而對智能交通系統(tǒng)和戶外視頻監(jiān)控等產(chǎn)生較大影響.當(dāng)人們需要對在霧霾天氣下所拍攝的視頻或圖像進(jìn)行目標(biāo)提取、跟蹤或識別時,霧霾的存在將導(dǎo)致圖像色彩失真、模糊不清,圖像信息不易提取.因此,圖像的去霧增強(qiáng)處理具有重要意義.
目前,主要利用一些先驗(yàn)或假設(shè)進(jìn)行去霧,不僅去霧效果好且時間復(fù)雜度小,實(shí)用性強(qiáng).直方圖均衡化[1]是一種簡單的圖像對比度增強(qiáng)方法,該方法通過將原始有霧圖像的直方圖進(jìn)行均勻拉伸,使其直方圖分布更均勻,增加了原始圖像像素灰度值的動態(tài)范圍,進(jìn)而達(dá)到增強(qiáng)原始圖像的整體對比度效果.該算法相對簡單,對景深變化不大的圖像復(fù)原效果較好,但對于景深變化較大的圖像處理效果不理想.Tan[2]通過統(tǒng)計發(fā)現(xiàn)無霧圖像的對比度相對于有霧圖像更高,因此通過最大化局部對比度進(jìn)行去霧,再使用Markov隨機(jī)模型進(jìn)一步歸一化結(jié)果,該方法可一定程度恢復(fù)霧天圖像的局部紋理和結(jié)構(gòu),但恢復(fù)圖像顏色常過于飽和.Fattal[3]在大氣散射物理模型的基礎(chǔ)上,假設(shè)圖像局部的反射率為常向量,場景目標(biāo)表面反射率和大氣光透射率在局部是獨(dú)立不相關(guān)的,使用獨(dú)立成分分析估計場景輻射率,該方法對于部分圖像可得到較好的去霧效果.但由于該方法是基于局部的統(tǒng)計獨(dú)立假設(shè),對于霧霾濃度較大的圖像去霧效果較差.何凱明[4]通過對戶外大量無霧圖像的觀察統(tǒng)計,提出了利用圖像暗通道先驗(yàn)知識和大氣物理散射模型進(jìn)行圖像去霧,該方法去霧效果客觀真實(shí),但對于目標(biāo)場景顏色和大氣光相似的情形得不到正確結(jié)果,同時該方法的時間代價對于實(shí)時系統(tǒng)應(yīng)用是不可接受的.Tarel等[5]對有霧圖像進(jìn)行預(yù)處理和中值濾波等操作,估計出大氣散射光進(jìn)而實(shí)現(xiàn)單幅圖像去霧,該方法可達(dá)到實(shí)時任務(wù)的要求,但中值濾波后得到了平滑的大氣散射光,無法保持景深跳變的邊緣信息,因此在一些較小的邊緣區(qū)域去霧效果不理想.Nishino等[6]提出了一種基于Bayes概率模型的去霧方法,該方法能有效去除霧的影響,但其需要建立統(tǒng)計先驗(yàn)求解概率模型.
本文基于Tarel方法進(jìn)行改進(jìn),提出一種新的去霧方法.首先利用雙邊濾波器和自適應(yīng)中值濾波器求得更符合真實(shí)有霧場景下的大氣散射光圖像;然后在YCbCr空間的Y通道上進(jìn)行四叉樹搜索求得更準(zhǔn)確的大氣光;最后,根據(jù)霧天退化模型得到復(fù)原的圖像.通過與幾種經(jīng)典的圖像去霧方法對比,實(shí)驗(yàn)結(jié)果表明本文算法具有更好的去霧效果.
在計算機(jī)視覺圖形圖像處理領(lǐng)域,針對單幅有霧圖像復(fù)原,主要采用有霧天氣條件下圖像質(zhì)量下降的物理退化模型[4,7-8]:
其中:I(x)表示觀察到的有霧圖像;J(x)表示正常場景下的清晰圖像;A表示大氣光強(qiáng)度;t(x)表示介質(zhì)透射率.假設(shè)大氣層各種介質(zhì)是同性同質(zhì)時,透射率t(x)可表示為
其中:β為大氣散射系數(shù);d(x)為景深,即景物到成像系統(tǒng)間的距離.由式(2)可計算出透射率t(x)隨著場景深度d(x)的增加以指數(shù)形式遞減.J(x)t(x)稱為直接衰減項(xiàng),表示景物光線經(jīng)過大氣介質(zhì)透射衰減后到達(dá)成像系統(tǒng)的部分;V(x)=A(1-t(x))表示大氣散射光,指大氣中光經(jīng)過各種散射而進(jìn)入到成像系統(tǒng)的部分.圖像去霧的目的就是從式(1)的已知I(x)中復(fù)原得到J(x),t(x)和d(x),是一個欠約束問題,通常需要添加不同的先驗(yàn)信息或約束條件才能求解.
其中Ω為中值濾波窗口的大小.中值濾波器是一種非線性濾波器,常用于去除圖像或其他信號中的噪聲.中值濾波本身沒有保角平滑的特性,因此在對W(x)進(jìn)行中值濾波過程中,最小通道圖的部分邊緣信息丟失,導(dǎo)致圖像的景深信息不能真實(shí)體現(xiàn),從而導(dǎo)致原圖像中的霧霾不能有效地去除.針對上述問題,本文在兩方面做出了改進(jìn):
1)對最小通道圖進(jìn)行雙邊濾波處理,得到初始的大氣散射光估計;
2)進(jìn)行兩次自適應(yīng)的中值濾波處理,得到邊緣清晰、紋理細(xì)節(jié)模糊的大氣散射光.
雙邊濾波器[11]是一種能很好保持圖像邊緣信息同時平滑圖像局部信息的非迭代濾波器,它由兩個核函數(shù)構(gòu)成,一個函數(shù)是由幾何空間距離決定濾波器系數(shù),另一個由像素差值決定濾波器系數(shù).因此,雙邊濾波器在處理相鄰各像素值的灰度值和彩色信息時,不僅考慮了幾何上的鄰近關(guān)系,也考慮了亮度上的相似性,處理過的圖像在濾除噪聲的同時還能很好地保持圖像的邊緣信息.
對于最小通道圖像W上的某點(diǎn)x,Ω(x)表示x周圍鄰域,W(x)和W(y)分別表示像素點(diǎn)x及其鄰域像素值,對x點(diǎn)處進(jìn)行雙邊濾波后得到的圖像B為
對大氣散射光進(jìn)行初始估計后再細(xì)化.自適應(yīng)濾波器可根據(jù)設(shè)定的條件自動改變其觀察窗口的大小,本文提出利用自適應(yīng)的中值濾波代替中值濾波[12].自適應(yīng)中值濾波器濾波過程中,在景深變化較平緩的區(qū)域,判斷濾窗中心的像素值是否與周圍像素值相似,此時相當(dāng)于中值濾波.在圖像景深變化較大的區(qū)域,當(dāng)濾窗中心的像素值和周圍像素值差別較大時,不改變該像素值大小.自適應(yīng)中值濾波算法如下.
1)選定某個濾波窗口Ω(x),Zmax,Zmin,Zmed分別為該濾窗內(nèi)像素灰度值最大、最小和中值,Zxy為該濾波窗口內(nèi)點(diǎn)(x,y)處像素灰度值,Ωmax為該濾波窗口允許的最大值.
2)設(shè)P1=Zmed-Zmin,P2=Zmed-Zmax.如果求得P1>0且P2<0,則轉(zhuǎn)3);否則按奇數(shù)增加濾窗窗口大小.滿足條件Ω(x)<Ωmax時,重復(fù)執(zhí)行步驟2);否則將Zxy作為輸出值.
3)令Q1=Zxy-Zmin,Q2=Zxy-Zmax.如果Q1>0且Q2<0,則將Zxy作為輸出值;否則輸出Zmed.
本文算法中,首先對有霧降質(zhì)圖像的最小通道圖W(x)進(jìn)行了一次雙邊濾波,從而得到邊緣清晰、紋理細(xì)節(jié)平滑的大氣散射光初始估計Wbf(x),然后利用自適應(yīng)的中值濾波器對Wbf(x)進(jìn)行濾波,得到Wada(x),由于相對紋理較豐富的近景區(qū)域可能沒有霧,因此將Wbf(x)和Wada(x)差的絕對值作為R(x).為了更進(jìn)一步強(qiáng)化圖像輪廓,對上述得到的結(jié)果再次進(jìn)行自適應(yīng)中值濾波得到S(x),然后使R(x)-S(x)得到C(x):
由于很多有霧的圖像沒有對比圖像,即不可能完全恢復(fù)出完全無霧的圖像,直接將有霧圖像所有霧都去除會導(dǎo)致圖像顏色失真,因此為了控制能見度的恢復(fù)程度,加入控制因子p∈[0,1].
為了更準(zhǔn)確地估計大氣光,基于圖像中天空區(qū)域的像素值方差較小,Kim等[13]提出了基于四叉樹搜索方法,首先將原圖像的每個RGB通道平均劃分為4部分,然后定義每部分得分為區(qū)域像素的平均值減去標(biāo)準(zhǔn)差值,選取得分最高的區(qū)域作為要迭代處理的部分,繼續(xù)劃分為4個更小的塊,再次計算得分,迭代進(jìn)行,直到選取的區(qū)域小于預(yù)設(shè)范圍時終止.最終選取區(qū)域中的最亮點(diǎn)作為整體大氣光值,該方法選取的像素值更接近天空區(qū)域,得到的大氣光A也更準(zhǔn)確.本文提出在YCbCr空間中運(yùn)用四叉樹搜索方法.在YCbCr空間中,Y通道表示亮度信息,因此只需對Y通道進(jìn)行四叉樹搜索,即可得到接近天空區(qū)域亮度最大值的大氣光A.
當(dāng)已知大氣散射光V(x)和大氣光A時,先根據(jù)V(x)=A(1-t(x))得到透射率:
再將A和t(x)代入式(1),可推導(dǎo)出場景輻射率J(x),即清晰的復(fù)原圖像:
在式(6)中,參數(shù)Φ∈(0,1]用于控制圖像能見度的恢復(fù)程度,通常設(shè)為90%~95%.Φ的設(shè)定可避免復(fù)原圖像出現(xiàn)過飽和或色彩偏暗現(xiàn)象.式(7)中,當(dāng)t(x)→0時,直接衰減項(xiàng)趨近于0,此時復(fù)原的圖像可能包含噪聲,所以對透射率t(x)設(shè)定一個下限to,本文實(shí)驗(yàn)中取to=0.1.
本文在Tarel方法的基礎(chǔ)上進(jìn)行改進(jìn),先對最小通道圖進(jìn)行雙邊濾波和自適應(yīng)的中值濾波,其中雙邊濾波中,窗口大小Ω=5×5,核函數(shù)參數(shù)σs=3,σr=0.1;再進(jìn)行兩次不同尺度的自適應(yīng)中值濾波,最大窗口大小均設(shè)為11×11.在式(5)和式(6)中,為了能使復(fù)原圖像不失真,需要保留一小部分霧霾,引入了參數(shù)p和Φ,其中p=0.9,而Φ則根據(jù)不同的有霧圖像設(shè)定,Φ∈[0.75,0.95].
圖1 本文算法去霧效果Fig.1 Effect of proposed dehazing method
實(shí)驗(yàn)數(shù)據(jù)庫選用4幅圖像,分別為canon3,canon7,house和ny17,這些圖像都具有不同的景深信息,同時所帶的霧霾程度也不同.圖1為canon3和canon7的去霧結(jié)果.圖1(A)為原始有霧圖像,canon3中近景較清晰,而遠(yuǎn)景處的樓房明顯被霧覆蓋,canon7中景物都在遠(yuǎn)處,但遠(yuǎn)景處霧霾較嚴(yán)重;圖1(B)為大氣散射光圖像,大氣散射圖為灰度圖像,圖像中除了在深度發(fā)生突變的邊緣處有跳躍外,整體區(qū)域都是平滑的;圖1(C)為透射率圖像,圖像邊緣清晰、紋理模糊;圖1(D)為復(fù)原圖像,原圖中大部分的霧霾被去除,且顏色沒有失真.其中canon3中可清楚地分辨出馬路上的車輛顏色,canon7中原來被霧覆蓋的田地也可清晰地分辨.
圖2是house圖像不同方法的去霧結(jié)果對比.圖2(A)為原始有霧圖像;圖2(B)為文獻(xiàn)[3]去霧方法,house去霧后圖像整體對比度相對提高,去除了中間大部分區(qū)域的霧,局部的樹葉間霧的去除也較好,但右上角窗戶下和草叢中的霧去除較差;圖2(C)為文獻(xiàn)[5]方法的去霧結(jié)果,圖像中紅墻顏色較接近于真實(shí)顏色,圖像大部分區(qū)域的霧霾被去除,但樹葉間的霧未被除掉;圖2(D)為使用本文方法的去霧結(jié)果.經(jīng)對比發(fā)現(xiàn),本文方法整體去霧效果明顯,特別是樹葉間的細(xì)節(jié)部分,基本沒有霧霾存在,且顏色接近真實(shí)場景顏色.
圖2 “House”去霧對比結(jié)果Fig.2 Images of dehazed“house”
圖3是ny17圖像不同方法的去霧結(jié)果對比.圖3(A)為原始有霧圖像,遠(yuǎn)景處的高樓已經(jīng)被霧覆蓋;圖3(B)為文獻(xiàn)[2]方法去霧結(jié)果,圖像的顏色過于飽和,已經(jīng)失真;圖3(C)為文獻(xiàn)[5]方法去霧結(jié)果,去霧效果較明顯,遠(yuǎn)景處的霧也去除的較干凈,但天空顏色較暗;圖3(D)為本文算法去霧結(jié)果,整體去霧效果明顯,不僅近景紋理保持的很好,且顏色沒有失真,整幅圖像更真實(shí).
圖3 “ny17”去霧對比結(jié)果Fig.3 Images of dehazed“ny17”
綜上所述,本文提出了一種基于濾波的單幅圖像去霧算法.首先根據(jù)有霧圖像中大氣散射光的特性,利用雙邊濾波保持邊緣平滑,估計出初始大氣散射光;然后利用自適應(yīng)中值濾波器進(jìn)一步求得邊緣清晰、紋理平滑、景深信息精確的大氣散射光;最后根據(jù)大氣物理模型求得復(fù)原圖像.本文所得復(fù)原圖像更接近晴朗天氣條件下的真實(shí)圖像.通過與已有幾種圖像去霧算法對比,實(shí)驗(yàn)結(jié)果表明本文算法復(fù)原的圖像更真實(shí)、自然.
[1]Kim T K,Park J K,Kang B S.Contrast Enhancement System Using Spatially Adaptive Histogram Equalization with Temporal Filtering[J].IEEE Transactions on Consumer Electronics,1998,44(1):82-87.
[2]Tan R.Visibility in Bad Weather from a Single Image[C]//IEEE Conference on Computer Vision and Pattern Recognition.Piscataway:IEEE Press,2008:1-8.
[3]Fattal R.Single Image Dehazing [C]//Special Interest Group for Computer GRAPHICS Asia.New York:ACM Press,2008:1-9.
[4]HE Kaiming.Single Image Haze Removal Using Dark Channel Prior[J].IEEE Transactions on Pattern Analysis and Machine Intelligence,2011,33(12):2341-2353.
[5]Tarel J P,Hautiere N.Fast Visibility Restoration from a Single Color or Gray Level Image[C]//IEEE 12th Internation and Conference on Computer Vision.Piscataway:IEEE Press,2009:2201-2208.
[6]Nishino K,Kratz L,Lombardi S.Bayesian Defogging[J].Internation Journal of Computer Vision,2012,98(3):263-278.
[7]Nayar S K,Narasimhan S G.Vision in Bad Weather[C]//IEEE International Conference on Computer Vision.Piscataway:IEEE Press,1999:820-827.
[8]Narasimhan S G,Nayar S K.Vision and the Atmosphere[J].International Journal of Computer Vision,2002,48(3):233-254.
[9]孫抗,汪渤,周志強(qiáng),等.基于雙邊濾波的實(shí)時圖像去霧技術(shù)研究 [J].北京理工大學(xué)學(xué)報,2011,31(7):810-813.(SUN Kang,WANG Bo,ZHOU Zhiqiang,et al.Real Time Image Haze Removal Using Bilateral Filter[J].Transactions of Beijing Institute of Technology,2011,31(7):810-813.)
[10]甘佳佳,肖春霞.結(jié)合精確大氣散射圖計算的圖像快速去霧 [J].中國圖象圖形學(xué)報,2013,18(5):583-590.(GAN Jiajia,XIAO Chunxia.Fast Image Dehazing Based on Accurate Scattering Map[J].Journal of Image and Graphics,2013,18(5):583-590.)
[11]Tomasi C,Manduchi R.Bilateral Filtering for Gray and Color Images[C]//Proceedings of the 6th International Conference on Computer Vision.Piscataway:IEEE Press,1998:839-846.
[12]Hwang H,Haddad R A.Adaptive Median Filters:New Algorithms and Results[J].IEEE Transactions on Image Processing,1995,4(4):499-502.
[13]Kim J H,Tang W D,Sim J Y,et al.Optimized Contrast Enhancement for Real-Time Image and Video Dehazing[J].Journal of Visual Communication and Image Representation,2013,24(3):410-425.