FCM-YOLO:一種基于特征增強(qiáng)和多尺度融合的PCB缺陷檢測方法
控制與決策
頁數(shù): 9 2024-04-10
摘要: 針對PCB缺陷檢測任務(wù)中存在的目標(biāo)與背景相混淆、缺陷目標(biāo)較小不易識別等問題,提出一種基于特征增強(qiáng)和多尺度融合的PCB缺陷檢測方法FCM-YOLO.所提出方法以YOLOv5s為基礎(chǔ),首先在特征提取網(wǎng)絡(luò)中引入由空間到深度層和非跨步卷積層的組合,構(gòu)建特征重提取模塊,以減少信息丟失,保留小目標(biāo)特征信息;然后,在特征提取網(wǎng)絡(luò)的最深層引入上下文注意力模塊,通過學(xué)習(xí)上下文信息,使用可變形卷積...