一種基于小波分析的圖像去雨方法及系統(tǒng)的制作方法
【專利摘要】本發(fā)明屬于圖像處理技術領域,尤其涉及一種基于小波分析的圖像去雨方法及系統(tǒng)。所述基于小波分析的圖像去雨方法包括:步驟a:根據(jù)小波分析對視頻幀圖像進行圖層分解,并分析各個分解圖層的圖像信息;步驟b:計算融合系數(shù)矩陣,根據(jù)所述融合系數(shù)矩陣對不同圖像信息的分解圖層分別進行小波融合,并根據(jù)融合結果進行圖像重構得到去雨圖像;步驟c:利用各向異性擴散濾波對所述去雨圖像進行濾波處理。本發(fā)明的實施可以避免受到動態(tài)特性的干擾,更加準確有效的去除雨滴,提高了去雨算法的使用范圍,在雨勢很大的情況下也能有良好的去雨效果;并通過各向異性擴散濾波對去雨圖像進行濾波處理,使圖像更加清晰自然,并使得去雨算法更加具有實用性。
【專利說明】
-種基于小波分析的圖像去雨方法及系統(tǒng)
技術領域
[0001] 本發(fā)明屬于圖像處理技術領域,尤其設及一種基于小波分析的圖像去雨方法及系 統(tǒng)。
【背景技術】
[0002] 雨對圖像成像有很大的影響,會造成圖像成像模糊和信息覆蓋,其直接結果是視 頻圖像的清晰度下降,視頻圖像的數(shù)字化處理也會受此影響而性能下降。對受雨滴污染的 視頻圖像進行修復處理有利于圖像的進一步處理,包括基于圖像的目標檢測、識別、追蹤、 分割和監(jiān)控等技術的性能提高。而且視頻基于小波分析的圖像去雨技術在現(xiàn)代軍事、交通 W及安全監(jiān)控等領域都有廣泛的應用前景。
[0003] 有關視頻圖像中雨滴特性的研究已受到國際學術界的廣泛關注,去雨算法的研究 也從2003年Starik等(Starik S,Werman Μ. Simulation of rainin videos[C]Proceeding of Texture Workshop, ICCV. Ni ce, France: 2003,2:406-409)提出的中值法開始得到了迅 速的發(fā)展,處理的方法已經(jīng)不再局限于最初簡單的中值計算,偏度計算、Κ均值聚類、卡爾曼 濾波、字典學習和稀疏編碼、引導濾波、帖間亮度差、HSV空間、光流法、運動分割等很多方法 也逐漸開始應用在視頻圖像中雨滴檢測與去除的算法中,雨滴去除的效果也逐漸被提高。 Garg等最先提出利用雨滴帶來的帖間亮度差進行雨滴初檢,然后利用雨滴的直線性和方向 一致的特點進一步篩選,最后根據(jù)前后帖的像素亮度去除雨滴影響,可W較好地滿足雨滴 不覆蓋連續(xù)帖圖像情況下的雨滴檢測與去除;Zhang等將雨滴給像素帶來的色彩影響考慮 在內(nèi),從而提高雨滴檢測的準確性,改善了基于亮度變化的去雨算法在彩色圖像上的應用 效果;Liu等將雨滴的亮度影響和色彩影響同時應用在算法中,用兩帖檢測雨滴并去除; TripatM等先研究雨滴像素亮度變化的概率統(tǒng)計特性,然后利用雨滴像素亮度變化的對稱 性實現(xiàn)雨滴檢測,僅基于時域和另外考慮空間位置的影響時效果不完全相同;Kang等首先 利用雙邊濾波將雨圖分成高頻部分和低頻部分,并對高頻部分進一步處理得到非雨成分, 結合低頻部分得到去雨圖;Huang等首先利用上下文約束進行圖像分割,并利用上下文感知 進行單幅基于小波分析的圖像去雨,并在此基礎上提出了改進算法,文中首先用到了超完 備的字典對高頻部分進行處理。
[0004] 特別是最近幾年,視頻圖像去雨技術已成為新的研究熱點。如何在保證高魯棒性 的前提下提高去雨的準確率和實時性,是目前視頻圖像去雨領域的焦點。目前存在的算法 中,應用于靜態(tài)場景視頻雨滴檢測與去除的算法有較為成熟的研究成果,但是應用在動態(tài) 場景中的視頻上時,算法考慮的是視頻中出現(xiàn)運動物體所帶來的干擾,對于與雨滴特性區(qū) 別度不高的運動物體無法達到理想的檢測效果。此外,實時處理在多項技術應用的自動導 航系統(tǒng)、安全監(jiān)控系統(tǒng)等場合中有很大的應用需求。運些應用場合中往往需要及時得到處 理結果,反饋給用戶,視頻處理的滯后有可能導致用戶做出錯誤的判斷。因此視頻中雨滴檢 測與去除不僅需要提高精度,也需要提高處理速度,而且需要找到二者之間最佳平衡點。但 是當前算法還無法兼顧各種場景的處理速度和精度,實現(xiàn)去雨算法的實時性是當前研究面 對的一個重要課題。
[0005] 綜上所述,現(xiàn)有的圖像去雨技術存在的缺點在于:現(xiàn)有的圖像去雨算法基本是基 于像素亮度和雨滴形態(tài)特征進行去雨,去雨效果不是很理想;同時,現(xiàn)有的圖像去雨技術對 于動態(tài)場景的去雨效果不是很理想,算法復雜度和算法實時性也不能很好地兼顧。
【發(fā)明內(nèi)容】
[0006] 本發(fā)明提供了一種基于小波分析的圖像去雨方法及系統(tǒng),旨在解決現(xiàn)有的圖像去 雨技術對于動態(tài)場景的去雨效果不是很理想,且算法復雜度和算法實時性不能兼顧的技術 問題。
[0007] 本發(fā)明是運樣實現(xiàn)的,一種基于小波分析的圖像去雨方法,包括:
[0008] 步驟a:根據(jù)小波分析對視頻帖圖像進行圖層分解,并分析各個分解圖層的圖像信 息;
[0009] 步驟b:計算融合系數(shù)矩陣,根據(jù)所述融合系數(shù)矩陣對不同圖像信息的分解圖層分 別進行小波融合,并根據(jù)融合結果進行圖像重構得到去雨圖像;
[0010] 步驟C:利用各向異性擴散濾波對所述去雨圖像進行濾波處理。
[0011] 本發(fā)明實施例采取的技術方案還包括:在所述步驟a中,所述根據(jù)小波分析對視頻 帖圖像進行圖層分解具體為:基于小波分析的Malla算法將所述視頻帖圖像分解為十層, Malla算法的分解公式為:
[0016] 在上述公式中,Η和G分別是尺度函數(shù)Φ (X)和小波函數(shù)Φ(χ)的系數(shù)矩陣,Cl、 卻、巧和片分別對應圖像Ci-l的低頻部分、垂直方向的高頻部分、豎直方向的高頻部分W 及對角線方向的高頻部分。
[0017] 本發(fā)明實施例采取的技術方案還包括:在所述步驟a中,所述分析各個分解圖層的 圖像信息具體包括:所述圖層包括包含雨滴噪聲的圖層、包含背景與顏色信息的圖層W及 包含圖像紋理與物體邊緣信息的圖層,所述包含雨滴噪聲的圖層為第二到第四高頻系數(shù)圖 層,所述包含圖像背景與顏色信息的圖層為第五到第十高頻系數(shù)圖層,所述包含圖像紋理 與物體邊緣信息的圖層為第一層圖層。
[0018] 本發(fā)明實施例采取的技術方案還包括:在所述步驟b中,所述計算融合系數(shù)矩陣的 計算方式為:根據(jù)雨滴的亮度特性定義雨滴污染程度系數(shù),根據(jù)雨滴污染程度系數(shù)計算融 合系數(shù)矩陣;所述計算融合系數(shù)矩陣具體為:令雨滴污染程度系數(shù)S = G X E,其中,G為局部 梯度,E為局部能量,局部梯度和局部能量兩個參數(shù)相乘得到一個新的變量S,S值越大污染 越嚴重;對S矩陣進行歸一化處理得到S',系數(shù)矩陣和S'矩陣用于對圖像重構算法進行加權 優(yōu)化得到融合系數(shù)矩陣。
[0019] 本發(fā)明實施例采取的技術方案還包括:所述局部梯度G定義為:
[0020]
[0021 ]在上述公式中,Δ xf (i,j)和Δ yf (i,j)分別為點(i,j)的水平和垂直方向,Μ和N分 別為區(qū)域的邊長;
[0022] 所述像素的局部能量E表示為:
[0023]
[0024] 本發(fā)明實施例采取的另一技術方案為:一種基于小波分析的圖像去雨系統(tǒng),包括 小波分解模塊、矩陣計算模塊、小波融合模塊和圖像濾波模塊;所述小波分解模塊用于根據(jù) 小波分析對視頻帖圖像進行圖層分解,并分析各個分解圖層的圖像信息;所述矩陣計算模 塊用于計算融合系數(shù)矩陣;所述小波融合模塊用于根據(jù)所述融合系數(shù)矩陣對不同圖像信息 的分解圖層分別進行小波融合,并根據(jù)融合結果進行圖像重構得到去雨圖像;所述圖像濾 波模塊用于利用各向異性擴散濾波對所述去雨圖像進行濾波處理。
[0025] 本發(fā)明實施例采取的技術方案還包括:所述小波分解模塊根據(jù)小波分析對視頻帖 圖像進行小波分解具體為:基于小波分析的Malla算法將所述視頻帖圖像分解為十層, Malla算法的分解公式為:
[0030] 在上述公式中,Η和G分別是尺度函數(shù)Φ (X)和小波函數(shù)Φ(χ)的系數(shù)矩陣,Cl、 巧、巧和巧分別對應圖像Ci-l的低頻部分、垂直方向的高頻部分、豎直方向的高頻部分W 及對角線方向的高頻部分。
[0031] 本發(fā)明實施例采取的技術方案還包括:所述小波分解模塊分析各個分解圖層的圖 像信息具體包括:所述圖層包括包含雨滴噪聲的圖層、包含背景與顏色信息的圖層W及包 含圖像紋理與物體邊緣信息的圖層,所述包含雨滴噪聲的圖層為第二到第四高頻系數(shù)圖 層,所述包含圖像背景與顏色信息的圖層為第五到第十高頻系數(shù)圖層,所述包含圖像紋理 與物體邊緣信息的圖層為第一層圖層。
[0032] 本發(fā)明實施例采取的技術方案還包括:所述矩陣計算模塊計算融合系數(shù)矩陣的計 算方式為:根據(jù)雨滴的亮度特性定義雨滴污染程度系數(shù),根據(jù)雨滴污染程度系數(shù)計算融合 系數(shù)矩陣;所述計算融合系數(shù)矩陣具體為:令雨滴污染程度系數(shù)S = G X E,其中,G為局部梯 度,E為局部能量,局部梯度和局部能量兩個參數(shù)相乘得到一個新的變量S,S值越大污染越 嚴重;對S矩陣進行歸一化處理得到S',系數(shù)矩陣和S'矩陣用于對圖像重構算法進行加權優(yōu) 化得到融合系數(shù)矩陣。
[0033] 本發(fā)明實施例采取的技術方案還包括:所述局部梯度G定義為:
[0034]
[0035] 在上述公式中,Δ xf (i,j)和Δ yf (i,j)分別為點(i,j)的水平和垂直方向,Μ和N分 別為區(qū)域的邊長;
[0036] 所述像素的局部能量E表示為:
[0037]
[0038] 相對于現(xiàn)有技術,本發(fā)明產(chǎn)生的有益效果在于:本發(fā)明實施例的基于小波分析的 圖像去雨方法及系統(tǒng)采用小波多級分解對圖像進行圖層分解,分析各個圖層的圖像信息, 并根據(jù)雨滴影響的程度定義小波融合的規(guī)則,根據(jù)小波融合的規(guī)則對不同圖像信息的圖層 分別進行小波融合進行雨滴去除,從而可W避免受到動態(tài)特性的干擾,更加準確有效的去 除雨滴,提高了去雨算法的使用范圍,在雨勢很大的情況下也能有良好的去雨效果;并通過 各向異性擴散濾波對去雨圖像進行濾波處理,使圖像更加清晰自然,并使得去雨算法更加 具有實用性。
【附圖說明】
[0039] 圖1是本發(fā)明實施例的基于小波分析的圖像去雨方法的流程圖;
[0040] 圖2是小波分解示意圖;其中,圖2(a)為原始圖像,圖2(b)是圖像的低頻信息,圖2 (C)至圖2(1)是圖像的第十到第一層的高頻結構;
[0041 ]圖3是小波分解重構流程圖;
[0042] 圖4是本發(fā)明實施例的基于小波分析的圖像去雨系統(tǒng)的結構示意圖。
【具體實施方式】
[0043] 為了使本發(fā)明的目的、技術方案及優(yōu)點更加清楚明白,W下結合附圖及實施例,對 本發(fā)明進行進一步詳細說明。應當理解,此處所描述的具體實施例僅僅用W解釋本發(fā)明,并 不用于限定本發(fā)明。
[0044] 請參閱圖1,是本發(fā)明實施例的基于小波分析的圖像去雨方法的流程圖。本發(fā)明實 施例的基于小波分析的圖像去雨方法包括W下步驟:
[0045] 步驟S100:輸入視頻帖圖像;
[0046] 步驟S200:根據(jù)小波分析對圖像進行小波多層分解,并分析各個圖層的圖像信息, 分別得到包含背景與顏色信息的圖層、包含雨滴噪聲的圖層W及包含圖像紋理與物體邊緣 信息的圖層;
[0047] 在步驟S200中,小波分析具有很好的時間頻率定位特性,能夠把信號分解成多個 具有不同子頻段,頻率特性和方向特性的子頻段,所W小波分析又被稱作數(shù)學顯微鏡。受圖 像分解和重構算法的啟發(fā),Mallat提出了基于小波分析的Malla算法,即圖像多分辨分解和 重構金字塔算法。重構算法是分解算法的逆過程,經(jīng)過逆濾波就能恢復出原始的信號序列。 空間的投影f(x,y)可W用來表示二維圖像信號:
[(Κ)加 ]
[0051]如果尺度函數(shù)Φ(Χ)和小波函數(shù)Φ(Χ)的系數(shù)矩陣分別是Η和G,Malla算法得分解公 式為:
[0化6]在式(4)中,Ci、巧、啤和碑分別對應圖像Ci-i的低頻部分、垂直方向的高頻部 分、豎直方向的高頻部分W及對角線方向的高頻部分。Malla算法的重構算法可W表示為:
[0化7]
[005引在式(5)中,H*、G*分別為H、G的逆矩陣。
[0059] 所述圖層包括包含背景與顏色信息的圖層、包含雨滴噪聲的圖層W及包含圖像紋 理與物體邊緣信息的圖層。小波分析能夠分別對圖像進行線性、高通和低通濾波。在原始圖 像上進行行和列的低通濾波,可W得到第一層的低頻系數(shù)近似分量Ci,它包括圖像背景與 顏色信息。在原始圖像上進行行和列的高通濾波,可W得到水平高頻細節(jié)系數(shù)&f,垂直高 頻細節(jié)系數(shù)均'和對角線高頻細節(jié)系數(shù)它們包括不同方向的圖像紋理與物體邊緣信息。 上述分界操作重復在Cl低頻分量上進行操作,可W得到相對應的第二層的各頻率分量C2、 Df、巧和化^。如果上述分界操作重復在第m-1層進行的話,能夠得到和 城。
[0060] 雨滴噪聲的頻率很高,而圖像紋理與物體邊緣的噪聲比雨滴還要高,圖像背景與 顏色信息的頻率很低。所W,由受雨滴污染圖像的基于小波分析多層分解,可W發(fā)現(xiàn)雨滴噪 聲應該被包含在低圖層的高頻系數(shù)部分,更大的分解層數(shù)通常被用來確保去雨后圖像的細 節(jié)信息。在本發(fā)明實施例中,對圖像進行小波多層分解的分解層數(shù)為十層。具體如圖2所示, 是小波分解示意圖;其中,圖2(a)為原始圖像,圖2(b)是圖像的低頻信息,圖2(c)至圖2(1) 是圖像的第十到第一層的高頻結構。由分析可知,大部分的雨滴噪聲集中在第二到第四高 頻系數(shù)圖層上,而第五到第十高頻系數(shù)圖層則包含絕大部分圖像背景與顏色信息,第一層 包含圖像紋理與物體邊緣信息。
[0061] 步驟S300:根據(jù)雨滴的亮度特性定義雨滴污染程度系數(shù),根據(jù)雨滴污染程度系數(shù) 計算融合系數(shù)矩陣;
[0062] 在步驟S300中,由于被雨滴覆蓋的像素的灰度等級比背景灰度大,會產(chǎn)生邊緣效 應,于是,局部梯度能夠用來測量灰度的變化,局部梯度定義為:
[0063]
[0064] 式(10)中,Δ xf (i,j)和Δ yf (i,j)分別為點(i,j)的水平和垂直方向,Μ和N分別為 區(qū)域的邊長。由于雨滴亮度是基本不變的,雨滴像素具有更高和更穩(wěn)定的能量,像素的局部 能量可W表示為:
[00 化]
[0066] 令雨滴污染程度系數(shù)S = GXE,即局部梯度和局部能量兩個參數(shù)相乘得到一個新 的變量S,S值越大污染越嚴重。對S矩陣進行歸一化處理得到S',系數(shù)矩陣和S'矩陣用于對 圖像重構算法進行加權優(yōu)化得到融合系數(shù)矩陣.
[0067] 步驟S400:根據(jù)融合系數(shù)矩陣對包含背景與顏色信息的圖層、包含雨滴噪聲的圖 層W及包含圖像紋理與物體邊緣信息的圖層分別進行小波融合,并通過融合結果進行圖像 重構,得到去雨圖像;
[0068] 在步驟S400中,根據(jù)融合系數(shù)矩陣對包含背景與顏色信息、雨滴噪聲和圖像紋理 與物體邊緣信息的圖層分別進行小波融合,最后將Ξ部分融合結果進行圖像重構得到去雨 圖像,運樣可W使圖像的顏色和細節(jié)不失真。在本發(fā)明實施例中,為了去除雨滴,對于雨滴 噪聲所在的圖層,融合系數(shù)矩陣的權值應該小于1;而對于沒有被雨滴污染的圖層,融合系 數(shù)矩陣的權值設成大于1;具體如圖3所示,是小波分解重構流程圖。
[0069] 在進行不同圖像信息的圖層分別進行小波融合之后,再對連續(xù)的9個視頻帖圖像 進行圖像融合,融合后的去雨圖像代替中間第五帖圖像,W達到去雨效果。
[0070] 步驟S500:利用各向異性擴散濾波對去雨圖像進行濾波處理,得到優(yōu)化的去雨圖 像;
[0071] 在步驟S500中,各向異性擴散濾波方法可W實現(xiàn)圖像的非線性濾波操作,使得在 消噪的同時,能夠很好的保持圖像邊緣。一維熱傳導方程,也稱一維擴散方程,其為各向異 性擴散方程的基礎。熱能是由分子的不規(guī)則運動產(chǎn)生的。在熱能流動中有兩種基本過程:傳 導和對流。傳導由相鄰分子的碰撞產(chǎn)生,一個分子的振動動能被傳送到其最近的分子。運種 傳導導致了熱能的傳播,即便分子本身的位置沒有什么移動,熱能也傳播了。此外,如果振 動的分子從一個區(qū)域運動到另一個區(qū)域,它會帶走其熱能。運種類型的熱能運動稱為對流。 W此為基礎,基于各向異性擴散方程的濾波方法是近年來出現(xiàn)的一種新的圖像濾波技術, 即求解初始值為輸入圖像的非線性熱擴散方程。在擴散方程中,通過引入圖像特征,設計合 適的擴散系數(shù)來控制擴散方程的擴散行為,使得在平滑圖像的同時能夠保留甚至增強圖像 的特征信息。本發(fā)明在得到小波融合去雨圖像之后,通過使用各向異性擴散濾波處理去雨 圖像,優(yōu)化了圖像的視覺效果,使圖像更加清晰自然,并使得去雨算法更加具有實用性。
[0072] 步驟S600:輸出去雨圖像。
[0073] 請參閱圖4,是本發(fā)明實施例的基于小波分析的圖像去雨系統(tǒng)的結構示意圖。本發(fā) 明實施例的基于小波分析的圖像去雨系統(tǒng)包括圖像輸入模塊、小波分解模塊、矩陣計算模 塊、小波融合模塊、圖像濾波模塊和圖像輸出模塊;具體地:
[0074] 圖像輸入模塊用于輸入視頻帖圖像;
[0075] 小波分解模塊用于根據(jù)小波分析對圖像進行小波多層分解,并分析各個圖層的圖 像信息,分別得到包含背景與顏色信息的圖層、包含雨滴噪聲的圖層W及包含圖像紋理與 物體邊緣信息的圖層;其中,小波分析具有很好的時間頻率定位特性,能夠把信號分解成多 個具有不同子頻段,頻率特性和方向特性的子頻段,所W小波分析又被稱作數(shù)學顯微鏡。受 圖像分解和重構算法的啟發(fā),Mallat提出了基于小波分析的Malla算法,即圖像多分辨分解 和重構金字塔算法。重構算法是分解算法的逆過程,經(jīng)過逆濾波就能恢復出原始的信號序 列。^;^?間的投影f(x,y)可W用來表示二維圖像信號:
[0079]如果尺度函數(shù)Φ(χ)和小波函數(shù)Φ(χ)的系數(shù)矩陣分別是Η和G,Malla算法得分解公 式為:
[0084] 在式(4)中,Ci、〇i、/才和巧分別對應圖像Ci-1的低頻部分、垂直方向的高頻部 分、豎直方向的高頻部分W及對角線方向的高頻部分。Malla算法的重構算法可W表示為:
[0085]
[0086] 在式(5)中,H*、G*分別為H、G的逆矩陣。
[0087] 所述圖層包括包含背景與顏色信息的圖層、包含雨滴噪聲的圖層W及包含圖像紋 理與物體邊緣信息的圖層。小波分析能夠分別對圖像進行線性、高通和低通濾波。在原始圖 像上進行行和列的低通濾波,可W得到第一層的低頻系數(shù)近似分量Ci,它包括圖像背景與 顏色信息。在原始圖像上進行行和列的高通濾波,可W得到水平高頻細節(jié)系數(shù)Of,垂直高 頻細節(jié)系數(shù)和對角線高頻細節(jié)系數(shù)踩",它們包括不同方向的圖像紋理與物體邊緣信 息。上述分界操作重復在Cl低頻分量上進行操作,可W得到相對應的第二層的各頻率分量 〇2、化"、巧和踩。如果上述分界操作重復在第m-1層進行的話,能夠得到Cm、Of、成和 執(zhí)
[0088] 雨滴噪聲的頻率很高,而圖像紋理與物體邊緣的噪聲比雨滴還要高,圖像背景與 顏色信息的頻率很低。所W,由受雨滴污染圖像的基于小波分析多層分解,可W發(fā)現(xiàn)雨滴噪 聲應該被包含在低圖層的高頻系數(shù)部分,更大的分解層數(shù)通常被用來確保去雨后圖像的細 節(jié)信息。在本發(fā)明實施例中,對圖像進行小波多層分解的分解層數(shù)為十層。具體如圖2所示, 是小波分解示意圖;其中,圖2(a)為原始圖像,圖2(b)是圖像的低頻信息,圖2(c)至圖2(1) 是圖像的第十到第一層的高頻結構。由分析可知,大部分的雨滴噪聲集中在第二到第四高 頻系數(shù)圖層上,而第五到第十高頻系數(shù)圖層則包含絕大部分圖像背景與顏色信息,第一層 包含圖像紋理與物體邊緣信息。
[0089] 矩陣計算模塊用于根據(jù)雨滴的亮度特性定義雨滴污染程度系數(shù),根據(jù)雨滴污染程 度系數(shù)計算融合系數(shù)矩陣;其中,由于被雨滴覆蓋的像素的灰度等級比背景灰度大,會產(chǎn)生 邊緣效應,于是,局部梯度能夠用來測量灰度的變化,局部梯度定義為:
[0090]
[0091] 式(10)中,Δ xf (i,j)和Δ yf (i,j)分別為點(i,j)的水平和垂直方向,Μ和N分別為 區(qū)域的邊長。由于雨滴亮度是基本不變的,雨滴像素具有更高和更穩(wěn)定的能量,像素的局部 能量可W表示為:
[0092]
[0093] 令雨滴污染程度系數(shù)S = GXE,即局部梯度和局部能量兩個參數(shù)相乘得到一個新 的變量S,S值越大污染越嚴重。對S矩陣進行歸一化處理得到S',系數(shù)矩陣和S'矩陣用于對 圖像重構算法進行加權優(yōu)化得到融合系數(shù)矩陣.
[0094] 小波融合模塊用于根據(jù)融合系數(shù)矩陣對包含背景與顏色信息的圖層、包含雨滴噪 聲的圖層W及包含圖像紋理與物體邊緣信息的圖層分別進行小波融合,并通過融合結果進 行圖像重構,得到去雨圖像;其中,根據(jù)融合系數(shù)矩陣對包含背景與顏色信息、雨滴噪聲和 圖像紋理與物體邊緣信息的圖層分別進行小波融合,最后將Ξ部分融合結果進行圖像重構 得到去雨圖像,運樣可W使圖像的顏色和細節(jié)不失真。在本發(fā)明實施例中,為了去除雨滴, 對于雨滴噪聲所在的圖層,融合系數(shù)矩陣的權值應該小于1;而對于沒有被雨滴污染的圖 層,融合系數(shù)矩陣的權值設成大于1;具體如圖3所示,是小波分解重構流程圖。在進行不同 圖像信息的圖層分別進行小波融合之后,再對連續(xù)的9個視頻帖圖像進行圖像融合,融合后 的去雨圖像代替中間第五帖圖像,W達到去雨效果。
[0095] 圖像濾波模塊用于利用各向異性擴散濾波對去雨圖像進行濾波處理,得到優(yōu)化的 去雨圖像;其中,各向異性擴散濾波方法可W實現(xiàn)圖像的非線性濾波操作,使得在消噪的同 時,能夠很好的保持圖像邊緣。一維熱傳導方程,也稱一維擴散方程,其為各向異性擴散方 程的基礎。熱能是由分子的不規(guī)則運動產(chǎn)生的。在熱能流動中有兩種基本過程:傳導和對 流。傳導由相鄰分子的碰撞產(chǎn)生,一個分子的振動動能被傳送到其最近的分子。運種傳導導 致了熱能的傳播,即便分子本身的位置沒有什么移動,熱能也傳播了。此外,如果振動的分 子從一個區(qū)域運動到另一個區(qū)域,它會帶走其熱能。運種類型的熱能運動稱為對流。W此為 基礎,基于各向異性擴散方程的濾波方法是近年來出現(xiàn)的一種新的圖像濾波技術,即求解 初始值為輸入圖像的非線性熱擴散方程。在擴散方程中,通過引入圖像特征,設計合適的擴 散系數(shù)來控制擴散方程的擴散行為,使得在平滑圖像的同時能夠保留甚至增強圖像的特征 信息。本發(fā)明在得到小波融合去雨圖像之后,通過使用各向異性擴散濾波處理去雨圖像,優(yōu) 化了圖像的視覺效果,使圖像更加清晰自然,并使得去雨算法更加具有實用性。
[0096] 圖像輸出模塊用于輸出去雨圖像。
[0097] 本發(fā)明實施例的基于小波分析的圖像去雨方法及系統(tǒng)采用小波多級分解對圖像 進行圖層分解,分析各個圖層的圖像信息,并根據(jù)雨滴影響的程度定義小波融合的規(guī)則,根 據(jù)小波融合的規(guī)則對不同圖像信息的圖層分別進行小波融合進行雨滴去除,從而可W避免 受到動態(tài)特性的干擾,更加準確有效的去除雨滴,提高了去雨算法的使用范圍,在雨勢很大 的情況下也能有良好的去雨效果;并通過各向異性擴散濾波對去雨圖像進行濾波處理,使 圖像更加清晰自然,并使得去雨算法更加具有實用性。
[0098] W上所述僅為本發(fā)明的較佳實施例而已,并不用W限制本發(fā)明,凡在本發(fā)明的精 神和原則之內(nèi)所作的任何修改、等同替換和改進等,均應包含在本發(fā)明的保護范圍之內(nèi)。
【主權項】
1. 一種基于小波分析的圖像去雨方法,其特征在于,包括: 步驟a:根據(jù)小波分析對視頻幀圖像進行圖層分解,并分析各個分解圖層的圖像信息; 步驟b:計算融合系數(shù)矩陣,根據(jù)所述融合系數(shù)矩陣對不同圖像信息的分解圖層分別進 行小波融合,并根據(jù)融合結果進行圖像重構得到去雨圖像; 步驟c:利用各向異性擴散濾波對所述去雨圖像進行濾波處理。2. 根據(jù)權利要求1所述的基于小波分析的圖像去雨方法,其特征在于,在所述步驟a中, 所述根據(jù)小波分析對視頻幀圖像進行圖層分解具體為:基于小波分析的Malla算法將所述 視頻幀圖像分解為十層,Malla算法的分解公式為: Ci = HcHrCi-l _ = U,G,.C,' 在上述公式中,H和G分別是尺度函數(shù)Φ(χ)和小波函數(shù)Φ(χ)的系數(shù)矩陣,CnD、和 Df分別對應圖像Ch的低頻部分、垂直方向的高頻部分、豎直方向的高頻部分以及對角線方 向的高頻部分。3. 根據(jù)權利要求2所述的基于小波分析的圖像去雨方法,其特征在于,在所述步驟a中, 所述分析各個分解圖層的圖像信息具體包括:所述圖層包括包含雨滴噪聲的圖層、包含背 景與顏色信息的圖層以及包含圖像紋理與物體邊緣信息的圖層,所述包含雨滴噪聲的圖層 為第二到第四高頻系數(shù)圖層,所述包含圖像背景與顏色信息的圖層為第五到第十高頻系數(shù) 圖層,所述包含圖像紋理與物體邊緣信息的圖層為第一層圖層。4. 根據(jù)權利要求1所述的基于小波分析的圖像去雨方法,其特征在于,在所述步驟b中, 所述計算融合系數(shù)矩陣的計算方式為:根據(jù)雨滴的亮度特性定義雨滴污染程度系數(shù),根據(jù) 雨滴污染程度系數(shù)計算融合系數(shù)矩陣;所述計算融合系數(shù)矩陣具體為:令雨滴污染程度系 數(shù)S = GXE,其中,G為局部梯度,E為局部能量,局部梯度和局部能量兩個參數(shù)相乘得到一個 新的變量S,S值越大污染越嚴重;對S矩陣進行歸一化處理得到S',系數(shù)矩陣和S'矩陣用于 對圖像重構算法進行加權優(yōu)化得到融合系數(shù)矩陣。5. 根據(jù)權利要求4所述的基于小波分析的圖像去雨方法,其特征在于,所述局部梯度G 定義為:在上述公式中,A xf ( i,j )和Δ yf ( i,j )分別為點(i,j )的水平和垂直方向,Μ和N分別為 區(qū)域的邊長; 所述像素的局部能量Ε表示為:6. -種基于小波分析的圖像去雨系統(tǒng),其特征在于,包括小波分解模塊、矩陣計算模 塊、小波融合模塊和圖像濾波模塊;所述小波分解模塊用于根據(jù)小波分析對視頻幀圖像進 行圖層分解,并分析各個分解圖層的圖像信息;所述矩陣計算模塊用于計算融合系數(shù)矩陣; 所述小波融合模塊用于根據(jù)所述融合系數(shù)矩陣對不同圖像信息的分解圖層分別進行小波 融合,并根據(jù)融合結果進行圖像重構得到去雨圖像;所述圖像濾波模塊用于利用各向異性 擴散濾波對所述去雨圖像進行濾波處理。7. 根據(jù)權利要求6所述的基于小波分析的圖像去雨系統(tǒng),其特征在于,所述小波分解模 塊根據(jù)小波分析對視頻幀圖像進行小波分解具體為:基于小波分析的Malla算法將所述視 頻幀圖像分解為十層,Malla算法的分解公式為: Ci = HcHrCi-l d; = ir.Grc,, D- = GG, C;, 在上述公式中,H和G分別是尺度函數(shù)Φ (X)和小波函數(shù)Φ(χ)的系數(shù)矩陣,Ci、對、Df和 W分別對應圖像Ch的低頻部分、垂直方向的高頻部分、豎直方向的高頻部分以及對角線方 向的高頻部分。8. 根據(jù)權利要求7所述的基于小波分析的圖像去雨系統(tǒng),其特征在于,所述小波分解模 塊分析各個分解圖層的圖像信息具體包括:所述圖層包括包含雨滴噪聲的圖層、包含背景 與顏色信息的圖層以及包含圖像紋理與物體邊緣信息的圖層,所述包含雨滴噪聲的圖層為 第二到第四高頻系數(shù)圖層,所述包含圖像背景與顏色信息的圖層為第五到第十高頻系數(shù)圖 層,所述包含圖像紋理與物體邊緣信息的圖層為第一層圖層。9. 根據(jù)權利要求6所述的基于小波分析的圖像去雨系統(tǒng),其特征在于,所述矩陣計算模 塊計算融合系數(shù)矩陣的計算方式為:根據(jù)雨滴的亮度特性定義雨滴污染程度系數(shù),根據(jù)雨 滴污染程度系數(shù)計算融合系數(shù)矩陣;所述計算融合系數(shù)矩陣具體為:令雨滴污染程度系數(shù)S = GXE,其中,G為局部梯度,E為局部能量,局部梯度和局部能量兩個參數(shù)相乘得到一個新 的變量S,S值越大污染越嚴重;對S矩陣進行歸一化處理得到S',系數(shù)矩陣和S'矩陣用于對 圖像重構算法進行加權優(yōu)化得到融合系數(shù)矩陣。10. 根據(jù)權利要求9所述的基于小波分析的圖像去雨系統(tǒng),其特征在于,所述局部梯度G 定義為:在上述公式中,A xf ( i,j )和Δ yf ( i,j )分別為點(i,j )的水平和垂直方向,Μ和N分別為 區(qū)域的邊長; 所述像素的局部能量Ε表示為:
【文檔編號】G06T5/00GK106067163SQ201610349107
【公開日】2016年11月2日
【申請日】2016年5月24日 公開號201610349107.6, CN 106067163 A, CN 106067163A, CN 201610349107, CN-A-106067163, CN106067163 A, CN106067163A, CN201610349107, CN201610349107.6
【發(fā)明人】朱青松, 袁杰, 王磊
【申請人】中國科學院深圳先進技術研究院