基于改进UNet网络的PCB缺陷检测方法

打开文本图片集
摘 要:针对PCB表面小尺寸缺陷难以检测的问题,提出了一种改进的UNet 语义分割网络,实现PCB表面缺陷图像的精确检测。首先,将UNet的四层网络层次修改为三层,可以减少整体的计算工作量、提升网络模型收敛速度、缩短训练时间;其次,在 UNet 网络中融入CBAM(Convolutional Block Attention Module)模块来提升图像中缺陷目标的显著度;然后,在编码阶段使用混合空洞卷积替换原有卷积块,增大感受野,获取更多的上下文信息。(剩余10617字)