一种基于改进YOLOv8算法的轻量化PCB缺陷检测方法

本发明涉及PCB检测技术领域，且公开了一种基于改进YOLOv8算法的轻量化PCB缺陷检测方法，包括以下步骤：步骤一、获取公开的PCB缺陷数据集，将数据集划分为训练集、验证集和测试集；步骤二、建立基于改进YOLOv8算法的PCB缺陷检测模型；步骤三、使用改进后的PCB缺陷检测模型对训练集样本进行训练；步骤四、使用训练好的最优模型对测试集数据进行检测，然后对模型复杂度和检测结果评估。本发明针对PCB表面缺陷检测，通过改进Neck网络、改进SPPF特征聚合模块、优化损失函数得到改进后的YOLOv8模型，大幅减少了网络层数、网络参数量以及模型计算量，加强了特征聚合能力，加速模型收敛，提升了模型对PCB表面缺陷的检测精度和速度。