一种基于注意力机制增强的PCB缺陷检测方法

本发明属于PCB缺陷检测技术领域，公开一种基于注意力机制增强的PCB缺陷检测方法。训练注意力机制增强的YOLOv8算法模型，用于图片检测。在原YOLOv8的Neck网络中每个检测头单元与其相连的C2f模块间，增加KAN‑Mamba线性注意力模块；KAN‑Mamba线性注意力模块为基于MLLA注意力机制改进，将MLLA注意力机制中MLLA模块结合KAN网络改进为KMLA模块，将MLLA注意力机制中2个Bath Norm模块替换为2个Grop Norm模块。本发明在保证精度的前提下，减少模块隐藏层神经元的数量，提高模型参数的利用效率，增强模型的特征表达能力。