改進神經(jīng)網(wǎng)絡的船舶紅外圖像邊緣檢測方法

劉志東

(1.四川大學，四川成都 610065；2.四川城市職業(yè)學院，四川成都 610110)

0 引言

紅外圖像采集系統(tǒng)具有采集圖像速度快、不受天氣影響等優(yōu)點，被應用在船舶檢測領域。通過紅外圖像采集系統(tǒng)可準確采集海上船舶紅外圖像，便于實施海上船舶管理。局部極值點匯聚一起形成圖像邊緣[1]，因船舶紅外圖像邊緣中存在關鍵的特征數(shù)據(jù)，因此對船舶紅外圖像邊緣檢測至關重要[2]。圖像處理中通過邊緣檢測方法提取圖像邊緣中背景和目標灰度差異部分，有利于目標識別[3–4]。目前，相關學者已經(jīng)開始對船舶紅外圖像邊緣檢測方法展開研究。王睿男[5]研究形態(tài)學的邊緣檢測方法，采用多方面元素提取船舶紅外圖像邊緣特征，經(jīng)多特征的加權融合計算后，完成船舶紅外圖像邊緣檢測。因該方法加權求解過程過于復雜，導致船舶紅外圖像邊緣檢測速度比較慢。何謙等[6]研究DexiNed 改進的邊緣檢測方法，通過優(yōu)化損失函數(shù)改進DexiNed 方法，調整網(wǎng)絡模型，對該模型實施訓練后輸出船舶紅外圖像邊緣數(shù)據(jù)，實現(xiàn)船舶紅外圖像邊緣檢測。雖然該方法在少量的樣本船舶紅外圖像邊緣檢測方面取得一定應用效果，但對于海量樣本的船舶紅外圖像邊緣檢測中應用效果不太理想，邊緣特征提取過慢，強紅外圖像邊緣檢測效率低，適應性低。

改進神經(jīng)網(wǎng)絡具有收斂性快、尋優(yōu)能力強、學習能力強等優(yōu)點，被廣泛應用在圖像識別領域，并取得較好應用效果。針對以上方法在船舶紅外圖像邊緣檢測中存在的局限性，研究改進神經(jīng)網(wǎng)絡的船舶紅外圖像邊緣檢測方法，提高船舶紅外圖像邊緣檢測能力。

1 船舶紅外圖像邊緣檢測方法

1.1 船舶紅外圖像去噪處理

因船舶紅外圖像在采集過程中容易受到外界因素的影響，導致采集的船舶紅外圖像中存在大量噪聲數(shù)據(jù)，因此采用基于塊匹配的主成分分析(principal component analysis，PCA)圖像去噪方法對船舶紅外圖像實施去噪處理，為船舶紅外圖像邊緣檢測奠定基礎。將船舶紅外圖像中需要去噪的像素點分為1 個變量塊作為訓練樣本，設置船舶紅外圖像中用Y表示矩陣，行列數(shù)量分別用a，b描述，P C A 變換求解流程為：

步驟1求解船舶紅外圖像Y矩陣的各行均值，其求解公式如下：

其中：像素點分別用i，j描述。

中心矩陣計算公式如下：

步驟2求解船舶紅外圖像Y矩陣的協(xié)方差矩陣，其求解公式為：

DY特征值對角矩陣表達式如下：

式中，特征值用 ν描述。

特征向量矩陣表達式為：特征向量用 ?描述，依據(jù)大小順序對 ν ， ?進行排序。

步驟3投影矩陣的生成，投影矩陣表達式為：

PCA 反變換方法：設置變換矩陣用P=?T描述，此時原始矩陣經(jīng)過降維后得出投影矩陣。PCA 含噪聲矩陣Y a通過最小均方誤差估計算法對其實施降噪，區(qū)分船舶紅外圖像中圖像數(shù)據(jù)與噪聲數(shù)據(jù)。用表示Y a第l行，該算法表達式如下：

式中：均值權重系數(shù)分別用E[·]， ωl描述；無噪聲估計值用描述。

σ表示噪聲標準差，方差用var()描述，權重系數(shù)求解公式：

其中：含噪聲像素值用s描述；無噪聲像素值用o描述。

通過式(7)可去除船舶紅外圖像矩陣各行的噪聲后，采用PCA 反變換后，可獲取去噪后船舶紅外圖像。

1.2 改進神經(jīng)網(wǎng)絡的圖像邊緣檢測

1.2.1 改進BP神經(jīng)網(wǎng)絡

BP神經(jīng)網(wǎng)絡算法流程如圖1所示。BP神經(jīng)網(wǎng)絡算法流程為：將神經(jīng)元實施初始化，計算隱含層、輸出層每個節(jié)點的輸出后，給定輸入向量與目標輸出，并計算目標值和實際輸出的偏差，求解反向誤差進行取值學習，對學習結果實施判斷，滿足終止條件，停止迭代，反之需要重新計算隱含層、輸出層每個節(jié)點的輸出。

圖1 BP神經(jīng)網(wǎng)絡算法流程Fig.1 BP neural network algorithm flow

通過附加動量法-自適應學習速率對BP 神經(jīng)網(wǎng)絡實施改進，調整BP 神經(jīng)網(wǎng)絡權值，提高網(wǎng)絡訓練魯棒性。附加動量法權值調整算法主要是通過改變學習速率，增強網(wǎng)絡性能，用 ?M表示網(wǎng)絡權值增量，附加動力因子權值調整表達式如下：

式中：動量因子用nc描述；網(wǎng)絡權值用M描述；訓練次數(shù)用k描述；誤差函數(shù)用F描述；學習速率用 γ描述。

在網(wǎng)絡訓練時，需要對附加動力因子設置判斷條件，才能正確使用權值調整公式，其判斷條件公式如下：

式中，第k代誤差平方和用F(k)描述。

學習速率過大過小均會影響神經(jīng)網(wǎng)絡學習訓練效率，因此需要自適應調節(jié)學習速率，學習速率自適應調整表達式如下：

1.2.2 圖像邊緣檢測

改進神經(jīng)網(wǎng)絡的船舶紅外圖像邊緣檢測基本流程為：將降噪處理后船舶紅外圖像轉換為二值圖像后作為改進BP神經(jīng)網(wǎng)絡的輸入，經(jīng)過網(wǎng)絡學習訓練，完成船舶紅外圖像檢測。輸入層確定流程如下：

步驟1輸入預處理后船舶紅外圖像α， 用α(i,j,g),g=1,2,3表 示圖像像素點 (i,j)的彩色分量；

步驟2通過彩色梯度算子求解出船舶紅外圖像中(i1,j1) ，(i2,j2)像素點的歐式距離，2個像素點歐式距離表達式為：

經(jīng)梯度算子處理后將船舶紅外圖像α 轉換為灰度圖像。

步驟3采用梯度算子將灰度圖像 α1轉換成二值圖像 α2。

步驟4輸入樣本為轉換后二值圖像，再實施下一步網(wǎng)絡訓練。

隱含層確定：依據(jù)相關資料可知，隱含層數(shù)量求解公式如下：

其中：整數(shù)用 ε 描述 ε =[0?10]；輸出和輸入神經(jīng)元數(shù)量分別用 β ， χ描述。此處構建由3層隱含層組成的神經(jīng)網(wǎng)絡。

輸出層確定：神經(jīng)網(wǎng)絡輸入層、輸出層中分別包括9個神經(jīng)元、1個神經(jīng)元。當輸出值比設置閾值0.1低時，表示檢測點不是船舶紅外圖像邊緣點；當輸出值比設置閾值0.9高時，表示檢測點為船舶紅外圖像邊緣點。

經(jīng)過上述過程，實現(xiàn)船舶紅外圖像檢測。

2 實驗分析

在MatLab 7.0軟件中測試本文方法的應用效果，將某地區(qū)海事管理部門實際采集船舶紅外圖像數(shù)據(jù)庫作為試驗對象，該數(shù)據(jù)庫中包括不同距離采集的船舶紅外圖像，試驗數(shù)據(jù)庫見表1。

表1 試驗數(shù)據(jù)庫Tab.1 Test database

設置分別用W，B表示 ι ?τ的噪聲圖像與干凈圖像，均方差誤差(MSE)表達式如下：

峰值信噪比(PSNR)作為評價指標，其計算公式如下：

其中，最大像素值用MAXB描 述。當PS NR>40表示圖像質量極好， 2 0

采用基于塊匹配的PCA 算法對試驗數(shù)據(jù)庫中船舶紅外圖像進行降噪處理后，取試驗數(shù)據(jù)的平均值，統(tǒng)計出本文方法對于不同距離采集船舶紅外圖像降噪后PSNR，結果見表2。分析可知，采用基于塊匹配的PCA 算法對于不同距離采集船舶紅外圖像實施降噪后PSNR 值均有所提升，本文方法降噪后PSNR 值全部高于40 dB，說明本文方法降噪效果較好，同時適用于不同噪聲標準差下的船舶紅外噪聲圖像去噪。

表2 不同距離采集船舶紅外圖像降噪PSNR(dB)Tab.2 PSNR of infrared image noise reduction of ships collected at different distances(dB)

在試驗數(shù)據(jù)庫中隨機選取1張近距離采集的船舶紅外圖像，采用本文方法對其紅外圖像實施邊緣檢測，邊緣檢測結果見圖2。分析圖2(a)可知，受霧天的影響采集原始船舶紅外圖像中存在大量噪聲，同時圖像清晰度較低；由圖2(b)可知，采用基于塊匹配的PCA 算法對近距離采集原始船舶紅外圖像實施降噪處理，經(jīng)PCA 反轉換有效去除原始圖像中多余噪聲，保留較好船舶紅外圖像邊緣數(shù)據(jù)，經(jīng)去噪后圖像清晰度得到顯著提高，視覺效果優(yōu)良；由圖2(c)可知，采用梯度算子將降噪后船舶紅外圖像轉換為二值圖像，保留完整的船舶紅外圖像中數(shù)據(jù)，為改進神經(jīng)網(wǎng)絡的檢測奠定基礎；由圖2(d)可知，將二值圖像作為改進神經(jīng)網(wǎng)絡輸入，經(jīng)過學習訓練輸出船舶紅外圖像邊緣檢測結果，采用本文方法可準確檢測出船舶紅外圖像邊緣，邊緣檢測結果的清晰度高且連貫，達到船舶紅外圖像邊緣檢測標準，有利于船舶分類管理。

圖2 本文方法的船舶紅外圖像邊緣檢測結果Fig.2 Edgedetection results of ship infrared image based on this method

3 結語

因船舶紅外圖像中含有大量噪聲數(shù)據(jù)，加大了船舶紅外圖像邊緣檢測難度，本文研究基于改進神經(jīng)網(wǎng)絡的船舶紅外圖像邊緣檢測方法，提高船舶紅外圖像邊緣檢測速度，便于對船舶管理。

試驗結果表明，本文方法可有效提取船舶紅外圖像真實邊緣，其應用效果較好。