基于改進YOLOv5的船體焊縫缺陷自動檢測方法

0 引言

目前中大型船舶的主要建造形式是將船舶各部分通過焊接的方法進行組合。建造一艘船舶至少 25%~40%的時間消耗在船舶焊接中。船舶的焊接性能是船舶密閉性的保證，因此對船舶焊接處的檢測至關(guān)重要。

張人杰等利用M-SMOTE 算法對焊縫樣本進行精細化處理，提取焊縫的不同特征，經(jīng) AP 聚類后輸入神經(jīng)網(wǎng)絡(luò)完成對船體焊縫的缺陷檢測。高翌飛等分析 FPSO 彎管特點，結(jié)合相控陣超聲，制作出相控陣掃描器，能夠契合 FPSO 彎管，實現(xiàn)對 FPSO 彎管在狹小空間的焊縫檢測。上述方法雖然能夠完成對船體焊縫缺陷的檢測，但也存在著一定的缺點。如對于船體焊縫圖像的采集要求嚴格、檢測時間長，對于一些微小的缺陷存在檢測準確率過低甚至無法檢測的問題對于正常的焊縫缺陷存在檢測準確率不高的問題等。

改進的 YOLOv5 具有輕量化、易部署、速度快準確度高等優(yōu)點，因此研究基于改進 YOLOv5 的船體焊縫缺陷自動檢測，可更加準確且快速完成船體焊縫檢測。

1 船體焊縫缺陷自動檢測

1.1 船體焊縫圖像采集及處理

在多種光照條件下，對不同產(chǎn)品焊接處進行圖像采集形成數(shù)據(jù)集。由于船體焊縫圖像的質(zhì)量會直接影響到船體焊縫檢測的結(jié)果，因此需要對船體焊縫圖像進行處理。本文利用灰度變化方法對采集到的船體焊接圖像進行處理，即可以避免圖像中的特征消失，也可以提高正常焊縫與存在缺陷的焊縫間的對比度使船體焊縫缺陷檢測更加容易。

對船體焊縫圖像進行正弦灰度計算完成圖像的處理，正弦灰度計算公式如下:

式中: g（x,y)為輸出像素灰度值;f(x,y) 為輸入像素灰度值; min(f)、max(f) 分別為輸入像素灰度值的最小值、最大值;r為輸入相對強度;(x,y) 為圖像像素的坐標點。

1.2 基于 YOLOv5 網(wǎng)絡(luò)的船體焊縫檢測

在YOLOv5 網(wǎng)絡(luò)中，主要組成部分為 Input、Back-bone、Neck 以及 Head。將灰度變換后的焊縫圖像通過Input 輸人到 YOLOv5 網(wǎng)絡(luò)中，Backbone 的作用為提取焊縫圖像特征，包含 Focus、SPP 以及 CSP。Focus 主要完成焊縫圖像的壓縮，使焊縫特征的提取速度加快;SPP 通過池化功能增加網(wǎng)絡(luò)感受野，使網(wǎng)絡(luò)可以處理多尺度問題;CSP 用于提取焊縫圖像特征，其中包含基本卷積 CBS 模塊。Neck 的主要作用是將 Backbone中提取的焊縫特征進行融合。Head 是檢測網(wǎng)絡(luò)也作為YOLOv5 網(wǎng)絡(luò)的輸出層，通過損失函數(shù)去除多余目標框根據(jù)圖形特征進行檢測得到最佳船體焊縫缺陷檢測結(jié)果。YOLOv5 網(wǎng)絡(luò)的船體焊縫檢測結(jié)構(gòu)如圖 1所示。

1.3 YOLOv5 的主干網(wǎng)絡(luò)改進

通過圖1可知，用于船體焊縫檢測的 YOLOv5 結(jié)構(gòu)中存在諸多 CBS 模塊，由于在 CBS 模塊中存在卷積層，且大量的卷積層會增加網(wǎng)絡(luò)的負載，給計算帶來壓力。因此，對YOLOv5 網(wǎng)絡(luò)中 Backbone 部分的 CSP1殘差模塊進行改進，使用 GhostNet 進行替換，改進后的 CSP1 如圖 2 所示。

采用 GhostNet 替換卷積層，GhostNet 內(nèi)部采用元余特征生成器與卷積層相結(jié)合，此種組合方式可 以有效降低網(wǎng)絡(luò)資源使用和計算時間。與一般卷積層相比，GhostNet 卷積層需要兩步完成卷積，一般卷積公式為:

其計算量為:

GhostNet 卷積公式為:

GhostNet 卷積計算量為 :

式中: b為偏置;Y為卷積輸出的焊縫特征圖，該焊縫特征圖高為H'，寬為w'，且Y∈RH'xw'xN;X為卷積輸入的焊縫圖像，該焊縫圖像高為h，通道數(shù)為c，寬為w，且X∈ Rhxwxc;Y'為 Gost 經(jīng)過一般卷積后輸出的焊縫特征圖，該焊縫特征圖大小為 H'xW'，數(shù)量為n個; kxk表示卷積核大小;f∈RcxNxkxk為存在 N個卷積核; Y'i為第i個在Y'中的焊縫特征圖，Yij為Y'被卷積核變換函數(shù)φij;轉(zhuǎn)換后生成的第j個 Ghost焊縫特征圖像，s為 Ghost焊縫特征圖像總數(shù)。

GhostNet 卷積如圖 3 所示。

1.4 損失函數(shù)

應(yīng)用改進 YOLOv5 網(wǎng)絡(luò)的損失函數(shù)采用的是CIoU，基于 DIOU 損失函數(shù)，并考慮了兩框的長寬比后形成的。其中 DIOU 損失函數(shù)的計算公式為:

式中: bgt為目標框; b'為預(yù)測框; o為兩框的最小外接矩形對角線長度; IoU為交并比;p2(b',bgt)為兩框中心點歐氏距離。

CIOU 的計算公式為:

其中：

式中: a為權(quán)衡參數(shù);v為衡量長寬比一致參數(shù); wgt為目標框?qū)挾? hgt為目標框高度; w'為預(yù)測框?qū)挾?:h' 為預(yù)測框高度。

2 仿真實驗

2.1 實驗對象

選擇某貨輪進行船體焊縫缺陷檢測，其參數(shù)如表 1 所示。

2.2 結(jié)果與分析

為驗證本文正弦灰度的計算效果，選擇船中某處具有缺陷的焊縫采集圖像并利用正弦灰度變換和伽馬灰度變換 2 種方式對圖像進行處理，并對比二者的焊縫缺陷檢測結(jié)果，如圖 4 所示?？芍?，當(dāng)采用伽馬灰度變換時圖像較為模糊，基本無法有效突出焊縫缺陷特征點，導(dǎo)致焊縫缺陷檢測結(jié)果不精確，預(yù)測框較大;而采用正弦灰度變換則能夠準確地突出缺陷所在位置的特征點，獲取精準的焊縫缺陷檢測結(jié)果。

為保證網(wǎng)絡(luò)的檢測準確性，需要對 YOLOv5 網(wǎng)絡(luò)進行足夠的訓(xùn)練。準備各種焊縫缺陷圖像各 500 幅進行訓(xùn)練，訓(xùn)練結(jié)果如圖 5 所示。可知，網(wǎng)絡(luò)的損失值隨著訓(xùn)練次數(shù)的增加逐漸減小，損失函數(shù)在前 100 次的訓(xùn)練中收斂速度極快，并在 200 次訓(xùn)練后損失逐漸達到平穩(wěn)狀態(tài)，經(jīng)過 500 次訓(xùn)練后平均準確度達到98%，損失函數(shù)為 0.15。

為證明本文方法的實際應(yīng)用效果，隨機采集船舶5 處焊縫進行缺陷檢測，檢測結(jié)果如表 2 所示?？芍瑑H在船長室中的焊縫處發(fā)現(xiàn)缺陷，關(guān)鍵的貨艙和輪機室未發(fā)現(xiàn)焊縫缺陷，說明該船舶的焊接大部分處于合格狀態(tài)，也證明了本文方法能夠準確檢測出不易發(fā)現(xiàn)的焊縫缺陷。

3 結(jié)語

本文方法對焊縫圖像進行的灰度變換，可使圖像更加清晰，訓(xùn)練的網(wǎng)絡(luò)準確率高損失值低，并且能夠快速準確發(fā)現(xiàn)船舶中的隱藏焊縫缺陷。

文章來源：新機器視覺

審核編輯：湯梓紅