公开(公告)号：CN117351288A

公开(公告)日：2024-01-05

申请号：CN202311430793.6

申请日：2023-10-31

Applicant: 湖北工业大学

Inventor： 周先军 ,  刘航 ,  石一可 ,  郭旋 ,  张浩

IPC: G06V10/764 ,  G06V10/774 ,  G06V10/40 ,  G06V10/762 ,  G06V10/80

Abstract: 本发明提出了一种基于改进YOLO v5算法的PCB缺陷检测方法，包括以下步骤：S1、采集待检测PCB缺陷图像数据构成样本图像数据集；S2、对所述样本图像数据集进行预处理，得到第一图像数据集；所述第一图像数据集包括训练集和测试集；S3、使用EfficientViT特征网络搭建PCB缺陷检测网络模型；S4、利用所述训练集对PCB缺陷检测网络模型进行训练，得到训练好的PCB缺陷检测网络模型；S5、使用训练好的PCB缺陷检测网络模型测试集进行检测，得到PCB图像的缺陷检测结果。本申请使用EfficientViT特征网络对训练集的图像进行特征提取，提高PCB缺陷检测网络对缺陷的检测能力。