深度學(xué)習(xí)電解電容識(shí)別與極性檢測(cè)
電解電容作為印刷電路板的重要極性元器件之一,起到濾波和耦合的作用。目前許多工廠仍采用人工目檢的方式來(lái)檢測(cè)電解電容的極性方向,然而這種方式效率低,成本高,并且隨著勞動(dòng)時(shí)間的增長(zhǎng),檢測(cè)人員會(huì)感到視覺疲勞導(dǎo)致其檢測(cè)的正確率降低。
近年來(lái),隨著計(jì)算機(jī)視覺的發(fā)展,基于卷積神經(jīng)網(wǎng)絡(luò)的檢測(cè)
方法被應(yīng)用到 PCB 的各項(xiàng)檢測(cè)任務(wù)中。Lin Y L等人提出了一種基于 YOLO 算法的 PCB 組件的電容檢測(cè)方法。Susa J等人提出了一種利用 YOLOv3 檢測(cè)電路板上電容的方法,該方法總體精度為93.33 % 。LiJ等人提出一種改進(jìn) YOLOv3 的算法,實(shí)現(xiàn)了電解電容的檢測(cè),精度達(dá)到 93.07 % 。然而,上述方法均是對(duì)電容的檢測(cè)識(shí)別,并未涉及到極性檢測(cè)。目前,對(duì)于電容極性判斷的方法分為傳統(tǒng)視覺識(shí)別方法和機(jī)器學(xué)習(xí)融合算法。
檢測(cè)方案設(shè)計(jì)
本文電容識(shí)別與極性檢測(cè)方案如圖 1 所示,算法總體設(shè)計(jì)共分為 3 個(gè)部分。
首先,提出一種改進(jìn)的 YOLOv5算法對(duì)電容進(jìn)行識(shí)別定位,同時(shí)提取電容感興趣區(qū)域;然后,利用語(yǔ)義分割模型對(duì)電容極性區(qū)域以及基圓區(qū)域進(jìn)行分割,如圖 2 所示;最后,對(duì)分割結(jié)果進(jìn)行形態(tài)學(xué)處理得到電容的極性方向。
改進(jìn)的 YOLOv5算法YOLOv5 網(wǎng)絡(luò)結(jié)構(gòu)主要包括 4 個(gè)部分:輸入層、 主干層、頸部和預(yù)測(cè)頭部。本文對(duì) YOLOv5 算法的改進(jìn)分為 2 個(gè)方面,首先在主干網(wǎng)絡(luò)的 Backbone中融合 SwinTransformer 模塊[以提高模型的特征提取能力;其次在主干網(wǎng)絡(luò)Neck網(wǎng)絡(luò)中,使用雙向特征金字塔網(wǎng)絡(luò)(來(lái)進(jìn)行特征融合。改進(jìn)后的網(wǎng)絡(luò)模型如圖 3 所示。