一种基于改进ResNet34网络的PCB板表面缺陷分类方法技术

本发明专利技术公开了一种基于改进ResNet34网络的PCB板表面缺陷分类方法，具体实施内容为：本发明专利技术首先建立PCB板表面缺陷数据集，将数据集按9：1分成训练集、测试集；然后对原有的ResNet34网络进行改进，加入通道注意力机制和空间注意力机制，并将改进后的ResNet34网络取名为ResNetAttention；对ResNetAttention网络进行迁移学习，加载PASCAL VOC数据集上的预训练权重；对进行迁移学习后的ResNetAttention网络使用训练集进行正式训练，得到ResNetAttention网络各层参数大小；训练完毕，使用测试集测试ResNetAttention网络分类性能指标，并与原ResNet34网络进行对比。该方法提高了PCB缺陷分类准确率，解决了传统人工PCB缺陷检测工作效率低、劳动强度大的问题。劳动强度大的问题。劳动强度大的问题。

A PCB surface defect classification method based on improved resnet34 network

【技术实现步骤摘要】
一种基于改进ResNet34网络的PCB板表面缺陷分类方法


[0001]本专利技术涉及PCB板表面缺陷领域，具体涉及深度学习领域一种基于改进ResNet34网络的PCB板表面缺陷分类方法。

技术介绍

[0002]印刷电路板(PCB)，是集成电路(IC)中较为基础的部件，自上个世纪50年代以来，集成复杂度由小规模发展到如今的特大规模，PCB在民用产品(如智能手机、便携电脑)或军用精密产品(如无人机)上的使用都极为广泛。随着制造技术越来越复杂，业界对PCB产品的工艺质量控制提出了更高的要求。然而由于生产过程天气等恶劣因素的不确定性和制作机器故障以及人为因素所造成的误差等影响，导致在过程中难免会出现同一批次质量不同的现象，如果不对这些因素进行最小限度的控制和最大程度的检测，所造成的损失是难以估量的，并且会对企业造成更大的影响。
[0003]目前，PCB板表面缺陷分类仍以人工分类为主，这种方式易受检测人员主观性影响大，检测费时费力，无法满足产品的工业自动化生产需求。另一方面，随着深度学习和数字图像处理技术不断的发展，基于这两种技术实现PCB板表本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于改进ResNet34网络的PCB板表面缺陷分类方法，其特征在于，具体按照以下步骤实施：步骤1：建立PCB板表面缺陷数据集，将数据集按一定比例分成训练集、测试集；步骤2：对原有的ResNet34网络进行改进，加入通道注意力机制和空间注意力机制，并将改进后的ResNet34网络取名为ResNetAttention；步骤3：对ResNetAttention网络进行迁移学习，加载PASCAL VOC开源数据集上的部分预训练权重；步骤4：对进行迁移学习后的ResNetAttention网络使用步骤1中的训练集进行正式训练，更新ResNetAttention网络的各层参数；步骤5：训练完毕，利用训练好的ResNetAttention网络对PCB板表面缺陷进行分类。2.根据权利要求1所述的基于改进ResNet34网络的PCB板表面缺陷分类方法，其特征在于：所述步骤1中，收集PCB板表面缺陷图像，将图像人工分为断路、短路、缺口、毛刺、针孔、露铜6类；在分类好的缺陷数据集基础上，对每类图像进行平移、旋转操作进行扩充，并按9：1的比例将数据集划分为训练集、测试集。3.根据权利要求1所述的基于改进ResNet34网络的PCB板表面缺陷分类方法，其特征在于：所述步骤2中，在原有的ResNet网络第一个卷积层之后和最后一个全连接层之前加入通道注意力机制和空间注意力机制，通道注意力机制表示如下：假设输入特征图X的大小为H
×
W
×
C，H、W、C分别表示输入特征图X的高、宽、通道数，通道注意力机制就可以表示为一个1
×1×
C的三维向量S，三维向量S中的每一个值代表输入特征图X对应通道的权重，经过通道注意力机制后的输出特征图Y如下，输出特征图Y的大小也为H
×
W
×
C：Y
H
×
W
×
C
＝X
H
×
W
×
C
*S1×1×
C

【专利技术属性】
技术研发人员：卢男凯，陈爱军，袁治未，蔡嘉乐，
申请(专利权)人：中国计量大学，
类型：发明
国别省市：