基于深度迁移学习的烟支缺陷检测方法技术

本发明专利技术涉及基于深度迁移学习的烟支缺陷检测方法，属于烟支缺陷检测、图像处理技术领域。本发明专利技术通过CCD工业相机获取数据集，通过对图像裁剪、压缩、图像标准化等操作生成新的数据集；将数据集按7:3成训练集和测试集；获取VGG19网络模型，改进网络全连接层，将模型全连接层以前的所有网络作为瓶颈层，冻结瓶颈层的权重值并加入新的全连接层从而开始迁移学习；添加二分类Soft Max层进行检测分类、计算准确率、训练保存网络模型；获取并处理待检测烟支图片，调用已训练的网络模型进行预测分类。本发明专利技术所提供的基于深度迁移学习的烟支缺陷检测方法，对烟支缺陷检测具有更高的准确性。对烟支缺陷检测具有更高的准确性。对烟支缺陷检测具有更高的准确性。

【技术实现步骤摘要】
基于深度迁移学习的烟支缺陷检测方法


[0001]本专利技术涉及基于深度迁移学习的烟支缺陷检测方法，属于烟支缺陷检测、图像处理


技术介绍

[0002]我国烟支生产量十分巨大，工业生产对烟支进行包装时，会存在破损、褶皱、黄斑等问题。传统方法处理这类问题是通过工作人员依靠肉眼检查是否有缺陷产品，该方法受到工人专注度制约，导致检测精度低、效率低，烟支缺陷问题以然存在，且所需人工成本高。随着图像处理技术的发展，不少学者提出了新的研究方法进行烟支缺陷检测，2018年肖峙宇采用边缘算子、滤波、分割等方法结合Matlab处理灰度值等过程检测烟支缺陷，最终的检测结果达到84.7％的准确率；2018年杨钰煊采用各种边缘检测方法并结合阈值分割进行烟支瑕疵检测，得到当参数值为0.35的阈值分割图像检测效果最佳；2019年周明等人应用CCD线阵相机和视觉算法结合检测烟支缺陷，最终分类准确率达到90.4％。这些检测方法虽然都取得了不错的检测率，但过程太复杂，无法保证运算的快速性，检测率也仍有一定的上升空间。随着图像处理技术的发展，利用深度学习检测烟支缺陷是一个新的研究方向。

技术实现思路

[0003]本专利技术提供了基于深度迁移学习的烟支缺陷检测方法，采用冻结瓶颈层权重值，只训练全连接层参数值，以用于解决小样本烟支数据集导致的过拟合问题，提高了小样本烟支数据检测精度。
[0004]本专利技术的技术方案是：基于深度迁移学习的烟支缺陷检测方法，首先获取VGG19网络模型，改进网络模型全连接层，将网络模型全连接层以前的所有网络作为瓶颈层，冻结瓶颈层的权重值并加入新的全连接层从而开始迁移学习；添加二分类Soft Max层进行检测分类、计算准确率、训练保存网络模型；获取并处理待检测烟支图片，调用已训练的网络模型进行预测分类。
[0005]作为本专利技术的进一步方案，所述方法的具体步骤如下：
[0006]步骤1、获取正常和缺陷的烟支图像数据集，采用SSD工业相机获取烟支的平展图像数据集，图像的尺寸为4096x800；
[0007]步骤2、对获取的烟支数据图像进行预处理；
[0008]步骤3、将已经预处理的烟支数据集按7：3划分为训练集和测试集；
[0009]步骤4、加载VGG19网络模型，改进其全连接层，将原来的三个全连接层1x1x4096、1x1x4096、1x1x1000改为1x1x4096、1x1x4096、1x1x2，并随机失活部分神经元，网络底层接入Soft Max层进行烟支缺陷分类；
[0010]步骤5、VGG19网络各层中的权重值是通过100万张数据集训练得到，将所有卷积层和池化层的每个神经元的权重值，从训练好的VGG19网络中迁移到一个全新的网络的过程称为迁移学习；此步骤是将VGG19网络全连接层以前的卷积层和池化层都划为瓶颈层，冻结
瓶颈层的权重值，迁移冻结的权重值到改进的VGG19网络中，并等待加载训练集和测试集开始训练改进VGG19网络中三个全连接层的权重值，形成自己的预训练模型；
[0011]步骤6、定义一个优化器Adam，，用来更新学习梯度，并申明网络模型的学习率为0.001，同时设置计算网络模型损失公式和网络模型准确率计算公式；网络模型采用交叉熵损失函数：
[0012][0013]其中：是预测输出值，y是真实输出值；
[0014]步骤7、针对烟支缺陷分类问题，在改进的全连接层后面使用一个二分类的Soft Max分类器，从而对烟支是否存在缺陷进行分类，Soft Max分类器采用Logistic回归模型公式计算，公式如下：
[0015][0016]其中，g(x)＝w0+w1x(1)+w2x(2)+
…
w
n
x(n),这里的w0,w1,
…
,w
n
是神经网络中输入Soft Max层的各个神经元权重值的参数w；向量x＝[x(1),x(2),....x(k)]是由k个独立变量组成，x(k)代表k种样本类别，P(y＝1|x)代表标签值为1的x这种样本类别发生的可能性；
[0017]步骤8、加载划分好的758张训练集和344张测试集进行网络模型的迭代训练，训练过程中计算检测的正确率，绘制出正确率曲线图，并保存训练完成的模型数据；其中，可以设置迭代次数为10次和30次，计算出10个epoch和30个epoch的训练集正确率，以及10个epoch和30个epoch的测试集正确率；
[0018]步骤9、将待检测的数据集进行步骤2一样的预处理，随机抽取待检测烟支数据中一个batch_size的烟支图像，调用步骤8中已训练完成的模型对待检测烟支进行预测是否存在缺陷；如果烟支存在缺陷，剔除缺陷烟支，完成工业烟支缺陷检测。
[0019]作为本专利技术的进一步方案，所述步骤2中，对获取的烟支数据图像进行预处理包括如下：
[0020](1)对数据集图像进行中心主体裁剪，裁剪的图像尺寸为2000x800，裁去非烟支信息；
[0021](2)对数据集图像进行压缩处理，压缩尺寸为224x224，得到适合神经网络输入图像的大小；最终使得图像的大小满足VGG19网络的输入尺寸，便于网络对图像进行处理；
[0022](3)对数据集图像进行水平方向翻折，扩充数据集的随意性；
[0023](4)对数据集图像进行标准化处理，执行均值、标准差参数均为0.5的标准化操作，将图像像素归一化在[
‑
1,1]的范围内。
[0024]作为本专利技术的进一步方案，所述步骤7中，当回归模型Logistic应用到分类问题中就产生了Soft Max分类器，Soft Max分类器的输入为样本特征，输出为样本属于各个类别的概率，训练样本集由k个被标签的样本构成：
[0025]T＝{(x(1),y(1)),(x(2),y(2)),
…
,(x(k),y(k))}
[0026]其中x(k)是k种样本类别，y(k)是分类标签，单个样本是第k种类别的概率为：
[0027]P[y(k)＝j|x(k)](j＝1,2,
…
,k)
[0028]其中j代表不同样本类别。
[0029]本专利技术的有益效果是：
[0030]1、本专利技术采用改进的VGG19网络进行迁移学习检测烟支缺陷，无需各种复杂操作过程提取烟支缺陷特征，利用卷积神经网络自动学习烟支缺陷特征，并对烟支是否存在缺陷进行快速精准的分类；
[0031]2、本专利技术采用冻结瓶颈层权重值，只训练全连接层参数值，解决了小样本烟支数据集导致的过拟合问题，提高了小样本烟支数据检测精度；
[0032]3、本专利技术深度学习神经网络模型来检测烟支缺陷，与以前的机器学习相比，本专利技术的检测率更高，检测速度更快。
附图说明
[0033]图1为本专利技术中的流程图；
[0034]图2为本专利技术中VGG19网络模型的结构示意图；
[0035]图3为本专利技术中10本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.基于深度迁移学习的烟支缺陷检测方法，其特征在于：首先获取VGG19网络模型，改进网络模型全连接层，将网络模型全连接层以前的所有网络作为瓶颈层，冻结瓶颈层的权重值并加入新的全连接层从而开始迁移学习；添加二分类Soft Max层进行检测分类、计算准确率、训练保存网络模型；获取并处理待检测烟支图片，调用已训练的网络模型进行预测分类。2.根据权利要求1所述的基于深度迁移学习的烟支缺陷检测方法，其特征在于：所述方法的具体步骤如下：步骤1、获取正常和缺陷的烟支图像数据集，采用SSD工业相机获取烟支的平展图像数据集；步骤2、对获取的烟支数据图像进行预处理；步骤3、将已经预处理的烟支数据集划分为训练集和测试集；步骤4、加载VGG19网络模型，改进其全连接层，将原来的三个全连接层1x1x4096、1x1x4096、1x1x1000改为1x1x4096、1x1x4096、1x1x2，并随机失活部分神经元，网络底层接入Soft Max层进行烟支缺陷分类；步骤5、将VGG19网络全连接层以前的卷积层和池化层都划为瓶颈层，冻结瓶颈层的权重值，迁移冻结的权重值到改进的VGG19网络中，并等待加载训练集和测试集开始训练改进VGG19网络中三个全连接层的权重值；步骤6、定义一个优化器Adam，并申明网络模型的学习率为0.001，同时设置计算网络模型损失公式和网络模型准确率计算公式；网络模型采用交叉熵损失函数：其中：是预测输出值，y是真实输出值；步骤7、针对烟支缺陷分类问题，在改进的全连接层后面使用一个二分类的Soft Max分类器，从而对烟支是否存在缺陷进行分类，Soft Max分类器采用Logistic回归模型公式计算，公式如下：其中，g(x)＝w0+w1x(1)+w2x(2)+
…
w
n
x(n),这里的w0,w1,

【专利技术属性】
技术研发人员：王端生，管一弘，杨雄飞，崔云月，罗亚桃，胡鹤鸣，
申请(专利权)人：昆明理工大学，
类型：发明
国别省市：