基于注意力機制與多尺度融合的PCB缺陷檢測

摘要： 針對印制電路板(PCB)缺陷區(qū)域受背景干擾過多以及缺陷目標(biāo)尺度較小導(dǎo)致缺陷檢測精度低的問題，提出了一種基于注意力機制與多尺度融合的PCB缺陷檢測方法。在YOLOv5模型的特征提取網(wǎng)絡(luò)中，引入一種三維注意力模塊，以增強缺陷目標(biāo)特征的顯著度，使模型更加注重目標(biāo)特征；為充分利用微小缺陷目標(biāo)的多尺度特征，在特征融合網(wǎng)絡(luò)中引入加權(quán)雙向特征金字塔網(wǎng)絡(luò)(Bi-directional Feat...