ViT和注意力融合的類別不均衡PCB缺陷檢測方法

摘要： 針對實際環(huán)境下印刷電路板(PCB)缺陷樣本難以收集造成的數(shù)據(jù)長尾分布和檢測精度低以及ViT用于檢測時計算復雜度高等問題,提出多尺度ViT特征提取和注意力特征融合的端到端PCB缺陷檢測算法。首先結合ViT和部分卷積構建多尺度特征提取網(wǎng)絡,利用層級多頭注意力對不同尺度的特征圖執(zhí)行自適應的注意力操作,使網(wǎng)絡能夠更好地捕捉局部和全局信息進而增強其特征提取能力,部分卷積可以降低計算開銷。... （共13頁）