一种基于改进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板表面缺陷分类成为一种更好的选择。
[0004]本专利技术针对现有技术的不足，提出了一种基于改进ResNet34网络的PCB板表面缺陷分类方法，首先建立PCB板表面缺陷数据集，将数据集按9：1分成训练集、测试集；然后对原有的ResNet34网络进行改进，加入通道注意力机制和空间注意力机制，并将改进后的ResNet34网络取名为ResNetAttention；对ResNetAttention网络进行迁移学习，加载PASCAL VOC开源数据集上的部分训练权重；对进行迁移学习后的ResNetAttention网络使用训练集进行正式训练；训练完毕，使用测试集测试ResNetAttention网络分类性能指标，并与原ResNet34网络进行对比。该方法提高了PCB缺陷分类准确率，解决了传统人工PCB缺陷检测工作效率低、劳动强度大的问题。
[0005]本专利技术所采用的技术方案是，一种基于改进ResNet34网络的PCB板表面缺陷分类方法，具体按照以下步骤实施：
[0006]步骤1：建立PCB板表面缺陷数据集，将数据集按一定比例分成训练集、测试集；
[0007]步骤2：对原有的ResNet34网络进行改进，加入通道注意力机制和空间注意力机制，并将改进后的ResNet34网络取名为ResNetAttention；
[0008]步骤3：对ResNetAttention网络进行迁移学习，加载PASCAL VOC开源数据集上的部分训练权重；
[0009]步骤4：对进行迁移学习后的ResNetAttention网络使用步骤1中的训练集进行正式训练，更新ResNetAttention网络的各层参数；
[0010]步骤5：训练完毕，使用步骤1中的测试集测试ResNetAttention网络分类性能指标，并与原ResNet34网络进行对比。
[0011]至此，实现了基于改进ResNet34网络的PCB板表面缺陷分类方法。
[0012]本专利技术的有益效果是：
[0013]1.在原有ResNet34网络的第一个卷积层之后和最后一个全连接层之前，添加了通道注意力机制和空间注意力机制，相比原ResNet34网络，对PCB板表面缺陷的分类准确率有所提高。
[0014]2.在网络正式训练之前进行迁移学习，预加载模型参数，相比不进行迁移学习直接训练，网络分类准确率显著提高，同时加快了模型的收敛速度。
附图说明
[0015]图1是本专利技术方法的步骤流程图；
[0016]图2是PCB板表面缺陷训练集样本数图；
[0017]图3是PCB板表面缺陷测试集样本数图；
[0018]图4是通道注意力机制示意图；
[0019]图5是空间注意力机制示意图；
[0020]图6是ResNetAttention网络训练损失曲线图；
[0021]图7是ResNetAttention网络训练准确率曲线图；
[0022]图8是训练完毕的ResNetAttention网络混淆矩阵图；
[0023]图9是ResNetAttention网络与原ResNet34网络F1score对比曲线图；
具体实施方式
[0024]以下结合附图对本专利技术作进一步说明。
[0025]如图1所示，本专利技术的步骤是：
[0026]步骤1：建立PCB板表面缺陷数据集，将数据集按一定比例分成训练集、测试集
[0027]对PCB板表面缺陷图像进行缺陷分类，分为断路、短路、缺口、毛刺、针孔、露铜6类；在分类好的缺陷数据集基础上，对每类图像进行平移、旋转操作进行扩充，扩充后的数据集总数为2950，对扩充后的数据集按9：1的比例划分为训练集、测试集，训练集各PCB缺陷类别样本数如图2所示，测试集各PCB缺陷类别样本数如图3所示。
[0028]步骤2：对原有的ResNet34网络进行改进，加入通道注意力机制和空间注意力机制，并将改进后的ResNet34网络取名为ResNetAttention
[0029]为了提高网络分类准确率，在ResNet34网络中引入注意力机制。注意力机制是机器学习模型嵌入的一种特殊结构，用以来自动学习和计算输入数据对输出数据的贡献大小。注意力机制运用于分类网络则表现为网络会自动重点学习输入图像的某部分特征，注意力机制分为通道注意力机制和空间注意力机制。通道注意力机制如图4所示，假设输入特征图X的大小为H
×
W
×
C，H、W、C分别表示输入特征图X的高、宽、通道数，通道注意力机制就可以表示为一个1
×1×
C的三维向量S，三维向量S中的每一个值代表输入特征图X对应通道的权重，经过通道注意力机制后的输出特征图Y如下，输出特征图Y的大小也为H
×
W
×
C：
[0030]Y
H
×
W
×
C
＝X
H
×
W
×
C
*S1×1×
C
ꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀ
(1)
[0031]空间注意力机制如图5如下：假设输入特征图X的大小为H
×
W
×
C，H、W、C分别表示输入特征图X的高、宽、通道数，空间注意力机制可以表示为一个H
×
W
×
1的三维向量G，三维向量G中的每一个值代表输入特征图X对应每一行特征的权重，经过空间注意力机制后输出特征图Y如下，输出特征图Y的大小也为H
×
W
×
C：
[0032]Y
H
×
W
×
C
＝X
H
×
W
×
C
*G
H
×...

【技术保护点】

【技术特征摘要】
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

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