一种融合ResNet和注意力机制的PCB语义分割方法

本发明公开了一种融合ResNet和注意力机制的PCB语义分割方法，该方法包括下述步骤：对PCB图像进行缺陷分割区域裁剪和数据增强；对增强后的PCB缺陷图像数据集采用泛化边缘保持和结构保持图像平滑模型进行图像平滑滤波，并输入UNet模型编码器提取缺陷区域特征；将缺陷区域特征进行逐层下采样，得到不同尺度的初始特征图；将不同尺度的初始特征图按其对应层次输入CBAM模块，CBAM模块将经过CBAM注意力机制处理后的特征图分别与不同尺度的初始特征图相加；UNet模型解码器获取CBAM模块每一层的特征图，逐层恢复图像分辨率，得到图像边界，输出缺陷分割结果。本发明能提高检测PCB图像的分割精度。