基于SPSO+SVM 的水稻葉部病害識別方法研究

陳 宇,路 陽*,蔡 娣,姜 峰,楊化龍,2

(1 黑龍江八一農(nóng)墾大學(xué)電氣與信息學(xué)院,大慶 163319;2 雞西市公安局,雞西 158100)

北方是我國水稻主要產(chǎn)區(qū),水稻種植過程中常常會出現(xiàn)各種病害,其中以稻瘟病、紋枯病、白葉枯病造成的影響最為嚴重。 這些病害造成水稻品質(zhì)下降、減產(chǎn),給種植戶帶來嚴重的經(jīng)濟損失。 為有效降低損失,需要對病害進行高效正確的識別。 傳統(tǒng)診斷方法主要依靠植物保護專家的經(jīng)驗,肉眼觀察,存在專家數(shù)量不足、主觀性強等缺點。

基于機器學(xué)習(xí)的水稻病害識別方法目前主要有兩類。 一類方法是基于圖像分割提取病斑特征,通常通過對水稻病斑圖像進行預(yù)處理、病斑分割和特征提取實現(xiàn)分類識別,比較典型的方法有增長法[1]、自適應(yīng)閾值分類法[2]、非監(jiān)督的顏色聚類識別法[3]、水平集分割法[4]等。 此類方法以水稻病斑的顏色、大小和形態(tài)差異為基準(zhǔn),對于光線、圖片質(zhì)量要求較高,在實用性等方面不強。 另一類方法是基于深度學(xué)習(xí)的水稻病斑分類識別。 通過深度學(xué)習(xí)提取水稻病斑的特征,排除了人工提取特征的步驟,對復(fù)雜環(huán)境下的識別適應(yīng)性更強。 但基于深度學(xué)習(xí)的病斑識別準(zhǔn)確度與初始訓(xùn)練樣本的數(shù)量和質(zhì)量成正比,且很難在短期內(nèi)構(gòu)建大樣本病害圖像庫。 支持向量機(Support vector machine,SVM)的理論和算法成熟,實用性強,其在語音識別、故障診斷、數(shù)據(jù)預(yù)測等方面取得了一定進展。 支持向量機能夠處理非線性可分數(shù)據(jù),具有受非支持向量影響小、泛化能力較好等優(yōu)點,廣泛應(yīng)用在農(nóng)作物病害識別等領(lǐng)域[5-8]。

本試驗以北方寒地采集的水稻葉部病害圖像為研究對象,采用方向梯度直方圖(Histograms of oriented gradient,HOG)提取水稻病害多種特征,并使用交換粒子群算法(Switching particle swarm optimization,SPSO)尋找支持向量機最優(yōu)參數(shù),最后在matlabR2018a 環(huán)境下構(gòu)建SPSO-SVM 模型并訓(xùn)練和測試樣本,以提高識別速率和準(zhǔn)確性,彌補傳統(tǒng)識別方法的不足。

1 材料與方法

選取3 種危害較大的水稻稻瘟病、紋枯病、白葉枯病作為研究對象。 水稻病害葉片均在自然光下使用手機拍攝,照片像素為1 312 px×1 104 px。 每種病害選擇100 幅圖像作為樣本,每幅圖像采用JPG 格式存儲。 采集的原始水稻病害圖像如圖1 所示。

圖1 3 種水稻葉部病害圖像Fig.1 Images of 3 kinds of rice leaf diseases

2 水稻葉部病害圖像預(yù)處理

水稻病害圖像采集的過程會受到各種因素影響,導(dǎo)致樣本圖像有噪聲、病斑邊緣部分模糊等問題。中值濾波是基于排序統(tǒng)計理論的一種能有效抑制噪聲的非線性信號處理技術(shù)[9-10]。 其原理是把數(shù)字圖像或數(shù)字序列中一點的值用該點的一個鄰域中各點值的中值代替,從而消除孤立的噪聲點。 二維中值濾波輸出為:

其中,f(x,y)、g(x,y)分別為原始圖像和處理后圖像,W為二維模板。 本試驗用5 ×5 的二維滑動模板,模板內(nèi)像素按照單調(diào)上升排序,濾波降噪后圖像如圖2 所示。

圖2 3 種水稻葉片病害分割的病斑Fig.2 Disease spot segmentation of 3 kinds of rice leaf diseases

3 水稻病斑特征提取

方向梯度直方圖(HOG),是用梯度的方向密度來描述圖像局部特征。 通過計算和統(tǒng)計圖像局部區(qū)域的梯度方向直方圖來構(gòu)成特征,具有幾何和光學(xué)形變的不變性[11-13]。 本試驗中提取HOG 特征步驟如下:

(1)對圖像進行g(shù)amma 校正,歸一化圖像。 由于圖像的顏色信息對于方向梯度直方圖特征沒有影響,因此對濾波后的灰度圖像進行平方根gamma 壓縮,歸一化特征向量。 這種壓縮處理有效降低了圖像局部的陰影和光照變化。

(2)求出圖像水平垂直梯度分量。 首先用一維離散微分模板[-1,0,1]及其轉(zhuǎn)置分別對gamma 校正過的水稻葉部圖像進行卷積運算,得到水平方向的梯度分量以及垂直方向的梯度分量。 通過像素點的垂直梯度和水平梯度,計算當(dāng)前像素點的梯度幅值和梯度方向,如公式(2)—(5)所示。

其中,Gx(x,y)、Gy(x,y)、H(x,y)分別表示當(dāng)前像素點的水平梯度、垂直梯度和像素值。G(x,y)、α(x,y)分別為當(dāng)前像素點的梯度幅值和梯度方向。

(3)建立梯度方向直方圖。 對尺寸為64 px×128 px 的水稻葉部病害圖像進行分割,分為8 ×16 個網(wǎng)格,每個網(wǎng)格的像素為8 px×8 px。 梯度方向范圍為[0,π],平均分成9 個區(qū)間梯度方向,20°為一個區(qū)間,加權(quán)投影每個像素梯度直方圖,建立每個網(wǎng)格對應(yīng)的9 維向量。

(4)合成塊,歸一化塊內(nèi)梯度直方圖。 每兩個2 ×2 網(wǎng)格合成一個塊,每個塊對應(yīng)36 維特征向量。 水稻葉部病害圖像的光照變化以及前后背景對比度差異,會導(dǎo)致梯度強度變化范圍巨大,需要消除負面影響,最后進行歸一化處理,如公式(6)所示。

其中,v是未經(jīng)歸一化的描述子向量,是v的2 范數(shù),ε是一個極小的常數(shù)。

(5)特征收集。 如圖3 所示,使用滑動窗口法對水稻葉部病害圖像進行掃描,以1 個網(wǎng)格為1 個掃描步長,8 ×16 個網(wǎng)格需要7 個水平方向掃描窗口和15 個垂直方向掃描窗口,塊之間會產(chǎn)生重疊,串聯(lián)所有歸一化后的塊,最后得到36 ×7 ×15 =3 780 維特征向量。

圖3 HOG 特征提取Fig.3 HOG feature extraction

4 基于支持向量機的水稻病害識別

水稻葉部病害特征復(fù)雜,特征維數(shù)大大多于樣本數(shù),樣本數(shù)據(jù)線性可分。 因此,本試驗采用基于核函數(shù)支持向量機(SVM)進行訓(xùn)練和分類識別。 支持向量機算法的基本思想是在特征空間中找到最優(yōu)的分離超平面,使訓(xùn)練集上的正負樣本間隔最大化。 假設(shè)訓(xùn)練樣本集T={(xi,yi),i=1,2,…,N},其中xi∈Rn,yi∈R,N是訓(xùn)練樣本的大小,超平面的描述如下:

其中ω為權(quán)向量,b為偏移向量,偏移向量決定了超平面與原點平面的距離。 引入松弛變量ξi和定義優(yōu)化問題如下:

其中,c是懲罰參數(shù),ξi和是非負松弛變量。 通過引入拉格朗日乘數(shù),將公式(8)轉(zhuǎn)化為二元形式,具體如下:

為了避免空間災(zāi)難問題,通常將核函數(shù)φ(xi)φ(xj)替換為k(xi,yi)的radial 核函數(shù),核函數(shù)的基本作用就是接受兩個低維空間里的向量,能夠計算出經(jīng)過某個變換后在高維空間里的向量內(nèi)積值。 本試驗采用radial 核函數(shù),即徑向基核函數(shù),公式為:

5 交換粒子群算法對支持向量機參數(shù)進行優(yōu)化

粒子群優(yōu)化算法(Particle swarm optimization,PSO)源于對鳥群捕食的研究,其基本思路為:通過群體中個體之間的相互協(xié)作和信息共享來尋求最優(yōu)解。 PSO 的優(yōu)勢在于簡單容易實現(xiàn)并且沒有過多參數(shù)的調(diào)節(jié)。 目前廣泛應(yīng)用于函數(shù)優(yōu)化等各種領(lǐng)域[14]。 粒子群算法通過一群初始化隨機粒子,通過迭代尋找最優(yōu)解,粒子通過公式(13)、(14)進行速度更新和位置更新:

式中:i=1,2,…,N,N是粒子總數(shù)。υi是粒子的速度。 rand():介于(0,1)之間的隨機數(shù)。xi:粒子當(dāng)前位置。c1,c2:學(xué)習(xí)因子,通常c1=c2=2。υi的最大值為υmax(大于 0),如果vi大于vmax,則vi=vmax。 以公式(13)、(14)為基礎(chǔ)形成粒子群算法標(biāo)準(zhǔn)公式,如公式(15)所示:

ω為慣性因子,其值較大時,全局尋優(yōu)能力強,局部尋優(yōu)能力弱;其值較小時,全局尋優(yōu)能力弱,局部尋優(yōu)能力強。 動態(tài)ω能獲得比固定值更好的尋優(yōu)結(jié)果。

PSO 算法具有在執(zhí)行優(yōu)化任務(wù)時易于實現(xiàn)和快速收斂等優(yōu)點,但存在早熟收斂、不一致收斂到全局最優(yōu)問題,為了克服這一局限性,本試驗采用改進算法:交換粒子群算法對支持向量機的參數(shù)進行優(yōu)化。

支持向量機中比較重要的參數(shù)有兩個:c和γ,c是懲罰因子,即對誤差的寬容度,可與任意核函數(shù)搭配,c系數(shù)越高,表示不能允許誤差出現(xiàn),但是容易過擬合;c系數(shù)越小越容易出現(xiàn)欠擬合。c過大或者過小都會導(dǎo)致SVM 泛化能力變差。γ是徑向基核函數(shù)自帶的一個參數(shù),隱含地決定了當(dāng)數(shù)據(jù)被映射到新的特征空間后的分布,當(dāng)γ值較小,支持向量較多,當(dāng)γ值較大,支持向量則較少。 支持向量的多少會大大影響支持向量機的訓(xùn)練和預(yù)測速度[15]。

在交換粒子群算法中,通過引入具有馬爾可夫交換參數(shù)的模式相關(guān)速度更新方程,以克服局部搜索和全局搜索之間的矛盾。 通常在搜索早期,群體中的粒子保持其獨立性和群體多樣性,這樣有助于擴大搜索范圍并避免過早收斂到局部最優(yōu)。 在搜索過程的后期,所有群可能會收斂到最佳粒子,以獲得更準(zhǔn)確的解決方案。 粒子的速度和位置根據(jù)公式(16)、(17)更新:

式中,ω(ξ(k))、c1(ξ(k))和c2(ξ(k))是慣性權(quán)重和加速度系數(shù),都依賴于馬爾可夫鏈。 設(shè)ξ(k)為馬爾可夫鏈,它在有限狀態(tài)空間S={1,2,…,N}中具有概率轉(zhuǎn)移矩陣即從i到j(luò)的轉(zhuǎn)移速率和可以通過當(dāng)前搜索信息進行調(diào)整,以平衡全局和局部搜索能力,如公式(18)所示:

其中,S為群大小,|L|為搜索空間中最長對角線的長度,D為目標(biāo)問題的維數(shù),xid為第i個粒子的第d個值為整個群中平均點ˉx的第d個值,可通過公式(19)計算:

利用交換PSO 優(yōu)化算法對SVM 兩個重要參數(shù)求解最優(yōu)值,優(yōu)化流程為:首先初始化粒子群位置與速度,設(shè)定核函數(shù)參數(shù)的初始參數(shù)數(shù)值,計算每個粒子的適應(yīng)度,根據(jù)適應(yīng)度更新當(dāng)前Pbest、Gbest,更新粒子的位置和速度,直到找到粒子最優(yōu)解Pbest 和種群最優(yōu)解Gbest,最后最優(yōu)解Gbest 即為需要的最優(yōu)參數(shù)。 交換PSO 偽代碼見表1,交換PSO 優(yōu)化SVM 參數(shù)流程如圖4 所示。

圖4 交換PSO 算法優(yōu)化SVM 參數(shù)流程Fig.4 Optimization of SVM parameters flow by switching PSO algorithm

表1 交換PSO 偽代碼Table 1 Switching PSO pseudo code

6 試驗與分析

對于3 種常見水稻葉部病害稻瘟病、紋枯病、白葉枯病,每種病害各取100 幅圖像進行病害識別,取其前70 幅總計210 幅圖像作為訓(xùn)練圖像集,剩余圖像作為測試圖像集。 對樣本的圖像進行預(yù)處理,使用HOG 進行特征提取。

定義交換 PSO 參數(shù):交換 PSO 粒子數(shù)設(shè)置為20個,最大迭代次數(shù)設(shè)置為50,學(xué)習(xí)因子c1設(shè)置為1.6,c2設(shè)置為1.5,慣性權(quán)重ω 設(shè)置為0.6,懲罰參數(shù)與核函數(shù)參數(shù)范圍為[0.1,1 000]。 迭代結(jié)果如圖5 所示。 構(gòu)建SVM 模型,使用HOG 提取的特征構(gòu)造訓(xùn)練集進行訓(xùn)練和識別,水稻葉部3 種病害的識別結(jié)果如表2 所示。

圖5 SPSO 優(yōu)化結(jié)果Fig.5 SPSO optimization results

表2 3 種水稻葉部病害識別結(jié)果Table 2 Identification results of 3 kinds of rice leaf diseases

從表2 可以看出,當(dāng)定義SVM 模型參數(shù)c和γ值為78 和0.3,3 種水稻葉部病害平均識別率為88.2%,訓(xùn)練時間為1 084 s。 利用PSO 算法優(yōu)化SVM 模型參數(shù)得到c和γ值為100 和0.7,此時參數(shù)c達到最優(yōu),3 種水稻葉部病害平均識別率為89.3%,訓(xùn)練時間減少323 s。 利用SPSO 算法優(yōu)化SVM 模型參數(shù)得到c和γ的最優(yōu)值為100 和0.4,不僅3 種水稻葉部病害平均識別率最高,達到93.2%,而且訓(xùn)練時間最少,僅為459 s。

7 結(jié)論

基于交換粒子群算法優(yōu)化后的支持向量機對3 種水稻葉部病害進行識別,取得了較好的識別效果。首先采用中值濾波對3 種水稻葉部病害樣本圖像進行預(yù)處理,并采用方向梯度直方圖構(gòu)建特征向量數(shù)據(jù)庫,然后構(gòu)建支持向量機,利用交換粒子群算法選取最優(yōu)支持向量機參數(shù),通過驗證,平均識別率為93.2%。 該算法與不進行優(yōu)化的支持向量機和僅用基本粒子群優(yōu)化的SVM 相比,訓(xùn)練速度更快,泛化能力更強,而且所需的訓(xùn)練樣本需求數(shù)量遠遠小于深度學(xué)習(xí)方法,可為水稻病害識別診斷提供新的思路。