本發(fā)明涉及圖像去噪領域,尤其是涉及了一種基于深度卷積神經網絡去除泊松噪聲的方法。
背景技術:
圖像去噪常用于視頻監(jiān)控、醫(yī)學、天文圖像等領域,還原噪聲圖像并保留圖像關鍵信息,即去除影響對圖像源信息進行理解分析的因素,獲得視覺更清晰的效果。具體地安全領域內可以在模糊視頻中是幀中目標輪廓清晰以助于辨別特定人或物,而醫(yī)學上,由于醫(yī)學成像系統的復雜性常產生噪聲導致醫(yī)學圖像質量下降,進而影響醫(yī)學分析診斷系統分析的精確性,故去除泊松噪聲可以使得醫(yī)學分析診斷能夠得到精確的圖像利于醫(yī)學后續(xù)工作。故盡可能的減少噪聲、提高圖像的質量,選用適當的方法盡可能地去除噪聲干擾是一個非常重要的圖像預處理步驟,換句話說圖像去噪算法研究是一切圖像處理的前提,具有相當重要的意義。
本發(fā)明提出了一種基于深度卷積神經網絡去除泊松噪聲的方法,采取監(jiān)督的方法,并使用由深度神經網絡所表現的強大表示能力,在沒有明確依賴一個模型的情況下,學習去除泊松噪聲。通過構建一個深度神經網絡DeNet,接收噪聲灰度圖像作為輸入,網絡每一層均用3×3的卷積核對上一層輸出進行卷積,然后提取最后一個通道與輸入圖像組合預測原圖的清晰圖像,產生對原始清晰圖像的估計,最終輸出所得清晰圖像。本發(fā)明突破了對數據模型的依賴,易于通過訓練適應某種數據類型,此外,高度可并行化可在GPU上進行快速運算,使得能夠更快速得到更精確的圖像,推動了醫(yī)學和天文等領域后續(xù)工作的研究和發(fā)展。
技術實現要素:
針對現有方法多需要更有效的對比推動研究的問題,本發(fā)明的目的在于提供一種基于深度卷積神經網絡去除泊松噪聲的方法,通過構建一個深度神經網絡,采取監(jiān)督的方法學習去除泊松噪聲,易于通過訓練適應某種數據類型,此外,高度可并行化可在GPU上進行快速運算,使得能夠更快速得到更精確的圖像,推動了醫(yī)學和天文等領域后續(xù)工作的研究和發(fā)展。
為解決上述問題,本發(fā)明提供一種基于深度卷積神經網絡去除泊松噪聲的方法,其主要內容包括:
(一)網絡架構;
(二)訓練網絡;
(三)監(jiān)督框架;
(四)去除泊松噪聲。
其中,所述的網絡架構,是一個深度神經網絡,用于將被泊松噪聲污染的圖像恢復成清晰圖像,即去除泊松噪聲,表示為DeNet,當網絡估計噪聲圖像與清晰圖像之間的差異時是以超分辨率為目的,并且與殘差網絡相似的是其權重梯度也是要通過后面的層并且直接從損失函數傳播到每一層。
進一步地,所述的DeNet,DeNet網絡接收噪聲灰度圖像作為輸入,產生對原始清晰圖像的估計,在每一層上,利用64個尺寸為3×3的卷積核在上一層輸出上以步長為1做卷積,第63個輸出通道用于計算后續(xù)步驟,而最后一個通道被提取出來直接與輸入圖像組合來預測清晰輸出,這些提取層可以被視為負噪聲成分因為它們的和抵銷了噪聲,網絡包括20個卷積層,其中第18層使用非線性修正線性單元(ReLU),而最后兩層完全保持線性。
其中,所述的訓練網絡,在采用數據流圖用于數值計算的開源軟件庫中執(zhí)行網絡,對5000幅圖像做240K次迭代,采用了64個尺寸為128×128的圖像塊,圖像被轉換為YCbCr,Y通道在經過峰值縮放并偏移后,用作輸入灰度圖像,為擴張數據,在訓練過程中,隨機對訓練圖像進行裁剪和翻轉垂直軸得到新的圖像塊,此外,噪聲實現也是隨機的。
進一步地,所述的訓練過程,使用自適應矩估計(ADAM)優(yōu)化器完成訓練,分別用不同的峰值單獨訓練網絡,為避免卷積在圖像塊的邊界產生偽圖像,在訓練期間使用l2損失在圖像塊的中心部分裁減了外部的21個像素。在測試時,將使用對稱映射的21個像素在其通過網絡傳遞之前用于填充圖像,然后裁剪回原來的大小輸出最終結果。
進一步地,所述的自適應矩估計(ADAM),利用梯度的一階矩估計和二階矩估計動態(tài)調整每個參數的學習率,學習率為α=10-4,β1=0.9,β2=0.999和∈=10-8,優(yōu)點主要在于經過偏置校正后,每一次迭代學習率都有個確定范圍,使得參數比較平穩(wěn)。
其中,所述的監(jiān)督框架,是為泊松去噪而構建,使網絡具備固有靈活性可對特定數據進行微調,此框架使用語義類作為先驗,并構建類感知去噪器,即限制為特定語義類,其中類感知是通過訓練而得,并非如一般方法是通過設計或用戶手動提供而得,因此它可以自動擴展到任何類型或數量的類,例如,選擇人臉去噪以獲得個人清晰照片集合,或者,一個潛在地訓練的同時另一個深度網絡對噪聲圖像進行自動分類。
進一步地,所述的泊松噪聲,既不是累加的也不是固定的,因為其強度取決于圖像強度,當信噪比(SNR)在每個像素中是時,降低圖像中的強度會產生更強的噪聲,因此,利用Y的最大值(其峰值)定義圖像中的噪聲功率,假設強度值在整個動態(tài)范圍內均勻分布時,此方法具有優(yōu)越的性能,并且適用于大多數自然圖像。
其中,所述的去除泊松噪聲,使用語義類作為先驗,并構建類感知去噪器,通過深度卷積神經網絡DeNet,以噪聲灰度圖像作為輸入,每一層均用3×3的卷積核對上一層輸出進行卷積,然后提取最后一個通道與輸入圖像組合預測原圖的清晰圖像,產生對原始清晰圖像的估計,最終輸出所得清晰圖像。
附圖說明
圖1是本發(fā)明一種基于深度卷積神經網絡去除泊松噪聲的方法的系統流程圖。
圖2是本發(fā)明一種基于深度卷積神經網絡去除泊松噪聲的方法的DeNet網絡架構圖。
圖3是本發(fā)明一種基于深度卷積神經網絡去除泊松噪聲的方法的去噪流程圖。
具體實施方式
需要說明的是,在不沖突的情況下,本申請中的實施例及實施例中的特征可以相互結合,下面結合附圖和具體實施例對本發(fā)明作進一步詳細說明。
圖1是本發(fā)明一種基于深度卷積神經網絡去除泊松噪聲的方法的系統流程圖。主要包括網絡架構、訓練網絡、監(jiān)督框架、去除泊松噪聲。
其中,所述的網絡架構,是一個深度神經網絡,用于將被泊松噪聲污染的圖像恢復成清晰圖像,即去除泊松噪聲,表示為DeNet,當網絡估計噪聲圖像與清晰圖像之間的差異時是以超分辨率為目的,并且與殘差網絡相似的是其權重梯度也是要通過后面的層并且直接從損失函數傳播到每一層。
DeNet網絡接收噪聲灰度圖像作為輸入,產生對原始清晰圖像的估計,在每一層上,利用64個尺寸為3×3的卷積核在上一層輸出上以步長為1做卷積,第63個輸出通道用于計算后續(xù)步驟,而最后一個通道被提取出來直接與輸入圖像組合來預測清晰輸出,這些提取層可以被視為負噪聲成分因為它們的和抵銷了噪聲,網絡包括20個卷積層,其中第18層使用非線性修正線性單元(ReLU),而最后兩層完全保持線性。
其中,所述的訓練網絡,在采用數據流圖用于數值計算的開源軟件庫中執(zhí)行網絡,對5000幅圖像做240K次迭代,采用了64個尺寸為128×128的圖像塊,圖像被轉換為YCbCr,Y通道在經過峰值縮放并偏移后,用作輸入灰度圖像,為擴張數據,在訓練過程中,隨機對訓練圖像進行裁剪和翻轉垂直軸得到新的圖像塊,此外,噪聲實現也是隨機的。訓練期間,使用自適應矩估計(ADAM)優(yōu)化器完成訓練,分別用不同的峰值單獨訓練網絡,為避免卷積在圖像塊的邊界產生偽圖像,在訓練期間使用l2損失在圖像塊的中心部分裁減了外部的21個像素。在測試時,將使用對稱映射的21個像素在其通過網絡傳遞之前用于填充圖像,然后裁剪回原來的大小輸出最終結果。
其中,所述的監(jiān)督框架,是為泊松去噪而構建,使網絡具備固有靈活性可對特定數據進行微調,此框架使用語義類作為先驗,并構建類感知去噪器,即限制為特定語義類,其中類感知是通過訓練而得,并非如一般方法是通過設計或用戶手動提供而得,因此它可以自動擴展到任何類型或數量的類,例如,選擇人臉去噪以獲得個人清晰照片集合,或者,一個潛在地訓練的同時另一個深度網絡對噪聲圖像進行自動分類。
注意,泊松噪聲既不是累加的也不是固定的,因為其強度取決于圖像強度,當信噪比(SNR)在每個像素中是時,降低圖像中的強度會產生更強的噪聲,因此,利用Y的最大值(其峰值)定義圖像中的噪聲功率,假設強度值在整個動態(tài)范圍內均勻分布時,此方法具有優(yōu)越的性能,并且適用于大多數自然圖像。
其中,所述的去除泊松噪聲,使用語義類作為先驗,并構建類感知去噪器,通過深度卷積神經網絡DeNet,以噪聲灰度圖像作為輸入,每一層均用3×3的卷積核對上一層輸出進行卷積,然后提取最后一個通道與輸入圖像組合預測原圖的清晰圖像,產生對原始清晰圖像的估計,最終輸出所得清晰圖像。
圖2是本發(fā)明一種基于深度卷積神經網絡去除泊松噪聲的方法的DeNet網絡架構圖。DeNet網絡接收噪聲灰度圖像作為輸入,產生對原始清晰圖像的估計,在每一層上,利用64個尺寸為3×3的卷積核在上一層輸出上以步長為1做卷積,第63個輸出通道用于計算后續(xù)步驟,而最后一個通道被提取出來直接與輸入圖像組合來預測清晰輸出,這些提取層可以被視為負噪聲成分因為它們的和抵銷了噪聲,網絡包括20個卷積層,其中第18層使用非線性修正線性單元(ReLU),而最后兩層完全保持線性。
圖3是本發(fā)明一種基于深度卷積神經網絡去除泊松噪聲的方法的去噪流程圖。以噪聲灰度圖像作為輸入,通過深度卷積神經網絡DeNet,每一層均用3×3的卷積核對上一層輸出進行卷積,然后提取最后一個通道與輸入圖像組合預測原圖的清晰圖像,產生對原始清晰圖像的估計,最終輸出所得清晰圖像。
對于本領域技術人員,本發(fā)明不限制于上述實施例的細節(jié),在不背離本發(fā)明的精神和范圍的情況下,能夠以其他具體形式實現本發(fā)明。此外,本領域的技術人員可以對本發(fā)明進行各種改動和變型而不脫離本發(fā)明的精神和范圍,這些改進和變型也應視為本發(fā)明的保護范圍。因此,所附權利要求意欲解釋為包括優(yōu)選實施例以及落入本發(fā)明范圍的所有變更和修改。