基于視覺的鐵路貨車滾動軸承表面缺陷分類研究

侯文英，楊懿，王鐵輝，黃嘉成

(1.內(nèi)蒙古科技大學(xué)，內(nèi)蒙古 包頭 014010；2.聯(lián)邦制藥(內(nèi)蒙古)有限公司，內(nèi)蒙古 巴彥淖爾 015000)

鐵路貨車滾動軸承是關(guān)系到鐵路運輸安全的關(guān)鍵部件，由軸承故障引發(fā)的車輛事故是威脅行車安全的重要因素，因此鐵路部門對軸承的檢修非常重視。然而現(xiàn)階段貨車進(jìn)入車輛段進(jìn)行一般檢修時，主要通過人工目測、手感及經(jīng)驗判斷軸承表面缺陷種類，該檢測方法依賴于檢測人員的經(jīng)驗和責(zé)任心，勞動強度大，長時間重復(fù)作業(yè)易引起視覺疲勞，工作效率低，很容易造成漏檢。

針對以上問題，選用CCD代替人眼對軸承表面缺陷進(jìn)行圖像采集，然后經(jīng)過設(shè)計的圖像處理技術(shù)進(jìn)行圖像處理，達(dá)到對10種常見軸承缺陷的自動分類、識別。

1 圖像預(yù)處理

10種常見的軸承表面缺陷如圖1所示。圖像預(yù)處理的具體流程如圖2所示，首先，由CCD黑白攝像機對軸承表面進(jìn)行圖像采集，獲得缺陷的灰度圖像，并經(jīng)中值濾波去除圖像中噪聲的干擾。通過對多種缺陷灰度直方圖的多次提取，發(fā)現(xiàn)缺陷的灰度大多集中在一個較小的范圍之內(nèi)，二值化后不僅可以有效地使缺陷與背景得到明顯區(qū)分，而且也為下一步確定缺陷最小外接矩形打下基礎(chǔ)。

圖1 軸承缺陷類型

圖2 圖像預(yù)處理流程圖

然后，對二值化后的圖像依次進(jìn)行卷積濾波與開、閉運算的處理，完好保留缺陷的同時去除了缺陷周圍零散邊緣點的干擾，使得缺陷與背景的區(qū)分更加明顯；接著運用旋轉(zhuǎn)法確定出了缺陷主要信息的最小外接矩形。

最后，由于灰度圖像較二值圖像對于缺陷信息的表達(dá)更為真實，根據(jù)二值圖像中獲得最小外接矩形所需圖像旋轉(zhuǎn)的角度及相應(yīng)最小矩形的范圍，在中值濾波后的灰度圖像中裁剪出包含缺陷主要信息的那部分灰度圖像，進(jìn)行后續(xù)的特征提取，這樣避免了計算其他不相關(guān)區(qū)域的特征量，減少了計算量及非缺陷區(qū)域?qū)θ毕輩^(qū)域特征量提取的影響，提高了網(wǎng)絡(luò)識別速度和精度。

2 缺陷特征提取

經(jīng)預(yù)處理后裁剪出的包含缺陷的灰度圖像稱為特征圖像，考慮到缺陷圖像種類繁多、缺陷特征復(fù)雜的特點，在提取如圖像長度、寬度、灰度均值、灰度方差及基于灰度共生矩陣的3個紋理特征(角二階矩、墑、相關(guān)度)等7個傳統(tǒng)特征的基礎(chǔ)上，又增加了如下特征量作為缺陷分類的依據(jù)。

2.1 壓縮度[1]

壓縮度也稱為環(huán)狀度，即

(1)

式中：LA，SA分別表示特征圖像內(nèi)不規(guī)則缺陷部分的周長和面積；分析中采用特征圖像的周長l和面積s代替目標(biāo)區(qū)域。

2.2 線度[2]

采集到的缺陷中除了單純的直線或曲線，更多的是有一定寬度的缺陷，因此，引入線度來描述圖像的線性程度，即

(2)

2.3 距離極值比[3]

特征圖像的距離極值比是用來描述該圖像的緊湊性，即

(3)

式中：dmax和dmin分別表示特征圖像內(nèi)缺陷部分的質(zhì)心(ci,cj)到缺陷區(qū)域邊緣的最大距離和最小距離，采用特征圖像的質(zhì)心(cx,cy)和距離代替。

2.4 NMI特征[4]

NMI(Normalized Moment of Inertia)也稱為歸一化轉(zhuǎn)動慣量，即

(4)

式中：將二維數(shù)字化灰度圖像M×N看成是二維xOy平面上的M×N個像素點，每個像素點(x,y)的灰度值為f(x,y)。NMI特征具有良好的縮放不變性、旋轉(zhuǎn)不變性和平移不變性，且不同物體的NMI特征差異明顯，非常適合作為缺陷的識別特征。

2.5 不變矩特征[5]

(5)

式中：mpq為中心矩，其反映了預(yù)處理后得到的特征圖像中的灰度相對于灰度重心是如何分布的。為了得到矩的不變特征，定義歸一化的中心矩為

(6)

利用歸一化的中心矩，可以獲得利用μpq表示的7個具有平移、比例和旋轉(zhuǎn)不變性的矩不變量，即預(yù)處理后得到的特征圖像的不變矩。

3 缺陷分類器的設(shè)計

分類器的設(shè)計除了要求準(zhǔn)確分類之外，還要能檢測出未知缺陷并確定分類的可靠性，指明缺陷的嚴(yán)重程度。針對貨車滾動軸承表面缺陷種類多、特征差異大等特點，選擇了BP神經(jīng)網(wǎng)絡(luò)作為缺陷分類的學(xué)習(xí)方法。典型的BP神經(jīng)網(wǎng)絡(luò)是一種具有3層或3層以上結(jié)構(gòu)的無反饋的、層內(nèi)無互聯(lián)結(jié)構(gòu)的前向網(wǎng)絡(luò)，其中首尾兩層分別稱為輸入層和輸出層，中間各層稱為隱含層。

3.1 輸入層

輸入層主要用作緩沖存儲，將源數(shù)據(jù)從輸入層加載至神經(jīng)網(wǎng)絡(luò)。輸入層的節(jié)點數(shù)目取決于源數(shù)據(jù)的維數(shù)，每一維均對應(yīng)一個輸入節(jié)點。下文中源數(shù)據(jù)均是從包含大量缺陷信息的特征圖像內(nèi)提取，選擇特征圖像長度、寬度2個幾何特征，灰度均值與方差2個灰度特征，3個紋理特征，特征圖像壓縮度、線度、距離極值比、NMI特征以及7個不變矩，共18個特征量作為BP網(wǎng)絡(luò)的源數(shù)據(jù)用于樣本學(xué)習(xí)訓(xùn)練和缺陷分類，即所使用的神經(jīng)網(wǎng)絡(luò)輸入層神經(jīng)元為18個。

3.2 輸出層

人工神經(jīng)網(wǎng)絡(luò)的輸出層節(jié)點數(shù)由采用的人工神經(jīng)網(wǎng)絡(luò)模型和需要的輸出表示方式確定。貨車滾動軸承的缺陷一共有15種，雖然預(yù)識別的缺陷只有10種，但是為了將來可以將識別系統(tǒng)進(jìn)行擴(kuò)展，輸出層神經(jīng)元設(shè)置為4個，并且輸出要求二進(jìn)制數(shù)，則其組合可完全表達(dá)15種缺陷。此次規(guī)定麻點、輾皮、剝離、擦傷、燒附、腐蝕、磕碰傷、凹痕、壓痕和劃傷這10種缺陷類型的期望輸出分別為0000，0001，0010，0011，0100，0101，0110，0111，1000，1001。選擇Sigmoid形函數(shù)f(x)=1/(1+e-x)作為網(wǎng)絡(luò)輸出層的傳遞函數(shù)。

3.3 隱含層

隱節(jié)點的作用是從樣本中提取并存儲其內(nèi)在規(guī)律，每個隱節(jié)點有若干個權(quán)值，而每個權(quán)值都是增強網(wǎng)絡(luò)映射能力的參數(shù)。隱節(jié)點數(shù)量太少，網(wǎng)絡(luò)從樣本中獲取信息的能力就差，不足以概括和體現(xiàn)訓(xùn)練集中的樣本規(guī)律；隱節(jié)點數(shù)量過多，又可能把樣本中非規(guī)律性的內(nèi)容如噪聲等也學(xué)會記牢，反而降低了泛化能力，此外隱節(jié)點數(shù)太多還會增加訓(xùn)練時間。在此采用試湊法確定隱節(jié)點數(shù)，首先根據(jù)經(jīng)驗公式

(7)

式中：l為隱含層節(jié)點數(shù)；n為輸入層節(jié)點數(shù)；m為輸出層節(jié)點數(shù)；a為1～10之間的調(diào)節(jié)常數(shù)。由經(jīng)驗公式確定隱含層節(jié)點數(shù)調(diào)節(jié)范圍為6～14，表1為在樣本集和循環(huán)迭代次數(shù)相同的情況下，不同隱含層節(jié)點數(shù)的BP神經(jīng)網(wǎng)絡(luò)訓(xùn)練誤差，由此可知該神經(jīng)網(wǎng)絡(luò)隱含層節(jié)點數(shù)取10最為合適。選擇雙曲正切S形函數(shù)g(x)=[2/(1+e-2x)]-1作為網(wǎng)絡(luò)隱含層的傳遞函數(shù)。

表1 不同隱含層節(jié)點數(shù)下的BP神經(jīng)網(wǎng)絡(luò)訓(xùn)練誤差

3.4 系統(tǒng)改進(jìn)

針對傳統(tǒng)BP算法的缺點，系統(tǒng)在以下方面進(jìn)行了改進(jìn)。

(1)傳統(tǒng)BP神經(jīng)網(wǎng)絡(luò)在訓(xùn)練學(xué)習(xí)新樣本時有遺忘舊樣本的趨勢，為了解決這一問題，在設(shè)計輸入向量時，將500個特征向量分為50組，每組均遍及10種缺陷，如此讓每種缺陷每隔一定的時間間隔反復(fù)對網(wǎng)絡(luò)進(jìn)行訓(xùn)練，提高其記憶能力。

(2)網(wǎng)絡(luò)的輸入節(jié)點物理量各不相同，有的數(shù)值相差很大，各指標(biāo)訓(xùn)練樣本之間不具備可比性，無法進(jìn)行綜合評估。因此，以S形函數(shù)作為轉(zhuǎn)移函數(shù)，該函數(shù)的值域為[0,1]，為防止小數(shù)信息被大數(shù)信息淹沒現(xiàn)象的發(fā)生，可以采用歸一化方法，將各輸入樣本歸一化到[0,1]區(qū)間內(nèi)。但是規(guī)范后的每一個輸出值序列中至少有一值為1，一值為0，恰好是S形函數(shù)的極小值和極大值，要求連接權(quán)足夠大才能使網(wǎng)絡(luò)的輸出值與其匹配，從而需要相當(dāng)多的訓(xùn)練次數(shù)來不斷修正權(quán)值，導(dǎo)致訓(xùn)練速度緩慢。所以，在計算過程中把原始數(shù)據(jù)歸一化到[0.05,0.95]區(qū)間內(nèi)，以便讓網(wǎng)絡(luò)輸出有足夠的增長空間。因此，在計算過程中用如下歸一化公式預(yù)處理數(shù)據(jù)

(8)

4 系統(tǒng)分類應(yīng)用

試驗中選取鐵路貨車滾動軸承表面常見10種缺陷的圖像，對現(xiàn)場采集的缺陷圖像和缺陷標(biāo)準(zhǔn)圖像進(jìn)行集中標(biāo)定后，從中選出700張(每種缺陷各70張)典型的能代表各類缺陷的圖像，其中500張(每種缺陷各50張)作為缺陷樣本對網(wǎng)絡(luò)進(jìn)行訓(xùn)練，200張(每種缺陷各20張)用于網(wǎng)絡(luò)的測試，BP分類器的缺陷識別結(jié)果見表2。

表2 BP分類器的缺陷識別結(jié)果

從表2給出的識別結(jié)果可以看出，測試數(shù)據(jù)集包括了10種缺陷類型，缺陷總數(shù)為200個，共正確識別出了187個，整體識別率達(dá)到了94%。

5 結(jié)論

(1)將視覺技術(shù)應(yīng)用于鐵路貨車滾動軸承表面缺陷檢測，降低了工人的勞動強度、提高了工作效率、加強了檢測結(jié)果的準(zhǔn)確性。

(2)卷積濾波與開、閉運算相結(jié)合的圖像處理方法，既能有效去除缺陷周圍邊緣點的干擾，又能夠滿足實時處理的要求。

(3)提取傳統(tǒng)特征基礎(chǔ)上增加了壓縮度、線度、距離極值比、NMI特征和不變矩等特征量，增強了缺陷分類的依據(jù)。

(4)采用具有較強泛化能力的BP神經(jīng)網(wǎng)絡(luò)對缺陷進(jìn)行分類，加強了分類的準(zhǔn)確性，通過對輸入矩陣和歸一化方法的改進(jìn)，提高了神經(jīng)網(wǎng)絡(luò)的記憶能力及識別速度。

(5)通過試驗對缺陷分類系統(tǒng)識別結(jié)果進(jìn)行統(tǒng)計，識別率達(dá)到94%，確認(rèn)了該系統(tǒng)的可靠性。