基于改進UNet網(wǎng)絡的PCB缺陷檢測方法

打開文本圖片集

摘 要：針對PCB表面小尺寸缺陷難以檢測的問題，提出了一種改進的UNet 語義分割網(wǎng)絡，實現(xiàn)PCB表面缺陷圖像的精確檢測。首先，將UNet的四層網(wǎng)絡層次修改為三層，可以減少整體的計算工作量、提升網(wǎng)絡模型收斂速度、縮短訓練時間；其次，在 UNet 網(wǎng)絡中融入CBAM（Convolutional Block Attention Module）模塊來提升圖像中缺陷目標的顯著度；然后，在編碼階段使用混合空洞卷積替換原有卷積塊，增大感受野，獲取更多的上下文信息。（剩余10617字）