一种基于深度神经网络的数码印花织物缺陷检测方法技术

本发明专利技术公开了一种基于深度神经网络的数码印花织物缺陷检测方法，具体按照以下步骤实施：步骤1、采集分辨率为416

【技术实现步骤摘要】
一种基于深度神经网络的数码印花织物缺陷检测方法


[0001]本专利技术属于纺织品疵点检测方法
，涉及一种基于深度神经网络的数码印花织物缺陷检测方法。

技术介绍

[0002]纺织品是人们生活中必不可少的产品，印花工艺是提升纺织品附加值的关键工序。数码印花作为一种新型的印花技术，将印花图像输入至计算机，利用计算机分色处理后，由专用RIP软件转换为数字点阵信息，控制孔径精度达微米级的喷头在织物上进行固定方向的喷印，形成预期高精度印花图案。因此，数码印花会由于喷孔堵塞、电机步进偏差、喷墨气压不稳定、喷头出墨、调试不均匀、设备压布不平整等故障，使印花产品出现PASS道、漏墨、喷墨不均以及布匹褶皱等缺陷，从而使产品的销售价格在其原价格的基础上降低45％～65％。因此，为确保数码印花织物产品的质量，缺陷检测是纺织品生产中质量控制的核心部分。
[0003]在一系列基于印花织物检测算法的基础上，数码印花织物缺陷检测方法主要分为两种，一种是基于传统图像处理的方法、一种基于深度神经网络的目标检测方法。传统的印花织物缺陷检测方法存在检测速度低、准确性差的缺点，并且其检测对象主要集中于白坯布、净色布、电子布以及纹理单一的色织物，针对图案丰富、色彩鲜明的印花产品，目前没有较为成熟的检测方法；基于此背景，基于深度神经网络的目标检测方法在数码印花织物缺陷检测领域中广泛应用。

技术实现思路

[0004]本专利技术的目的是提供一种基于深度神经网络的数码印花织物缺陷检测方法，能够实现数码印花缺陷的实时准确检测。
[0005]本专利技术所采用的技术方案是，一种基于深度神经网络的数码印花织物缺陷检测方法，具体按照以下步骤实施：
[0006]步骤1、采集分辨率为416
×
416大小的RGB彩色数码印花织物缺陷图像；建立神经网络；
[0007]步骤2、利用步骤1获得的彩色数码印花织物缺陷图像，进行目标信息的提取和标定，建立数码印花织物缺陷样本数据集，得到训练集、验证集和测试集；
[0008]步骤3、利用步骤1建立的神经网络和ImageNet数据集的训练集建立损失函数，利用损失函数训练神经网络，得到预训练模型，利用步骤2得到的训练集和验证集对预训练模型进行调整和验证；
[0009]步骤4、利用步骤2得到的测试集对模型进行评价。
[0010]本专利技术的特点还在于：
[0011]步骤1中采集分辨率为416
×
416大小的RGB彩色数码印花织物缺陷图像具体按照以下实施：利用扫描仪获取数码印花织物缺陷图像，运用局部均值法调整图像分辨率为416
×
416。包含PASS道、漏墨、布匹褶皱以及喷墨不均4类缺陷，每类缺陷800张，共3200张样本图像，并将图像统一命名为######.jpg格式。
[0012]步骤1中的建立神经网络具体按照以下实施：神经网络的构成具体如下：
[0013]1)建立CSPDarkNet53特征提取网络的子模块，CSPDarkNet53特征提取网络的子模块是在原DarkNet53子模块的基础上加上Cross Stage Partial，使其在保持网络结构轻量化的同时，增强其语义信息表达能力，子模块可对输入特征图进行2倍下采样，连续堆叠5个子模块，分别得到对输入图像进行8倍、16倍、32倍下采样的特征图作为特征融合网络的输入信息；
[0014]2)采用空间金字塔池化对经1)提取的特征图进行特征融合，使其输出特征图的像素点对于输入图像拥有较大的感受野，使其每个像素点都能够实现对于目标的检测，同时增加了每个像素点的语义信息，提高模型对于不同尺度目标的检测能力；
[0015]3)路径聚合网络旨在通过自底向上的路径增强顶层特征图的位置信息，缩短了低层与顶层特征之间的信息路径，通过自顶向下和自底向上的双向特征融合路径，使顶层的强语义信息与底层的强位置信息充分融合，提高了数码印花缺陷检测算法对于不同尺寸目标的检测能力；
[0016]4)利用分层预测的思想实现对于大、中、小目标的分层检测，使得顶层更关注于大目标的检测，底层更关注于小目标的检测。
[0017]步骤2具体按照以下实施：使用LabelImg标注工具对步骤1获得的彩色数码印花织物缺陷图像的印花缺陷信息进行标记，标记内容主要有输入缺陷标签，同时手动框选缺陷生成以图像左上角位置为基准的坐标信息，以及图像尺寸、存储路径的信息，并将标记信息生成的######.xml格式文件与######.jpg格式文件对应；从每类缺陷中随机选取600张作为训练样本，50张作为验证样本，150张作为测试样本。
[0018]步骤3具体按照以下实施：模型训练采用Python语言，Pytorch深度学习框架、第三方函数库Anaconda3.4.1完成。首先在ImageNet数据集上预训练网络模型，保存预训练的模型参数并使用我们的数据集对模型进行调整，作为数码印花缺陷检测算法的权重参数。反向传播过程中采用动量随机梯度下降算法更新网络的权值和偏置参数，经过53个epoch迭代得到最优的数码印花缺陷检测模型。
[0019]步骤3中建立损失函数具体按照以下实施：
[0020]1)置信度损失函数：
[0021]置信度损失函数缓解了单阶段目标检测中正负样本比例严重失衡和小目标缺陷难以检测的问题，权衡了难分、易分样本与正负样本(即织物有无缺陷)数量、大小织物缺陷(织物缺陷尺寸占输入图像尺寸的比率)目标数量的关系，公式(1)为置信度损失函数；
[0022]L
obj
＝-αylog(y
′
)(1-y
′
)2(1-β)-(1-α)(1-y)log(1-y
′
)(y
′
)2ꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀ
(1)
[0023]上式中，y为锚包含目标的真实值、y
′
为其对应的预测值、参数α用来调整正负样本的比例、β为真实框面积占输入图像面积的比例；
[0024]2)类别损失函数：
[0025]类别损失函数增加了小目标缺陷的权重，权衡了大目标和小目标在损失函数所占的比例，使其更关注于小目标的检测，公式(2)为类别损失函数；
[0026]L
cls
＝-ylog(y
′
)(1-β)-(1-y)log(1-y
′
)
ꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀꢀ
(2)
[0027]上式中，y为锚包含目标类别的真实值、y
′
为其对应的预测值、β为真实框面积占输入图像面积的比例；
[0028]3)边界框回归损失函数：
[0029]边界框回归损失函数同时考虑了预测框和目标框重叠区域、中心点距离、长宽比的关系并权衡了大小缺陷目标对边界框的回归影响，提高了小目标边界框的回归能力，公式(3)为边界框回归损失函数；
[0030][0031]上式中，p表示预本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于深度神经网络的数码印花织物缺陷检测方法，其特征在于，具体按照以下步骤实施：步骤1、采集分辨率为416
×
416大小的RGB彩色数码印花织物缺陷图像；建立神经网络；步骤2、利用步骤1获得的彩色数码印花织物缺陷图像，进行目标信息的提取和标定，建立数码印花织物缺陷样本数据集，得到训练集、验证集和测试集；步骤3、利用步骤1建立的神经网络和ImageNet数据集的训练集建立损失函数，利用损失函数训练神经网络，得到预训练模型，利用步骤2得到的训练集和验证集对预训练模型进行调整和验证；步骤4、利用步骤2得到的测试集对模型进行评价。2.根据权利要求1所述的一种基于深度神经网络的数码印花织物缺陷检测方法，其特征在于，所述步骤1中集分辨率为416
×
416大小的RGB彩色数码印花织物缺陷图像具体按照以下实施：利用扫描仪获取数码印花织物缺陷图像，运用局部均值法调整图像分辨率为416
×
416；包含PASS道、漏墨、布匹褶皱以及喷墨不均4类缺陷，每类缺陷800张，共3200张样本图像，并将图像统一命名为######.jpg格式。3.根据权利要求1所述的一种基于深度神经网络的数码印花织物缺陷检测方法，其特征在于，所述步骤1中的建立神经网络体按照以下实施：神经网络的构成具体如下：1)建立CSPDarkNet53特征提取网络的子模块，CSPDarkNet53特征提取网络的子模块是在原DarkNet53子模块的基础上加上Cross Stage Partial，使其在保持网络结构轻量化的同时，增强其语义信息表达能力，子模块可对输入特征图进行2倍下采样，连续堆叠5个子模块，分别得到对输入图像进行8倍、16倍、32倍下采样的特征图作为特征融合网络的输入信息；2)采用空间金字塔池化对经1)提取的特征图进行特征融合，使其输出特征图的像素点对于输入图像拥有较大的感受野，使其每个像素点都能够实现对于目标的检测，同时增加了每个像素点的语义信息，提高模型对于不同尺度目标的检测能力；3)路径聚合网络旨在通过自底向上的路径增强顶层特征图的位置信息，缩短了低层与顶层特征之间的信息路径，通过自顶向下和自底向上的双向特征融合路径，使顶层的强语义信息与底层的强位置信息充分融合，提高了数码印花缺陷检测算法对于不同尺寸目标的检测能力；4)利用分层预测的思想实现对于大、中、小目标的分层检测，使得顶层更关注于大目标的检测，底层更关注于小目标的检测。4.根据权利要求1所述的一种基于深度神经网络的数码印花织物缺陷检测方法，其特征在于，所述步骤2具体按照以下实施：使用LabelImg标注工具对步骤1获得的彩色数码印花织物缺陷图像的印花缺陷信息进行标记，标记内容主要有输入缺陷标签，同时手动框选缺陷生成以图像左上角位置为基准的坐标信息，以及图像尺寸、存储路径的信息，并将标记信息生成的######.xml格式文件与######.jpg格式文件对应；从每类缺陷中随机选取600张作为训练样本，50张作为验证样本，150张作为测试样本。5.根据权利要求1所述的一种基于深度神经网络的数码印花织物缺陷检测方法，其特征在于，所述步骤3具体按照以下实施：模型训练采用Python语言...

【专利技术属性】
技术研发人员：苏泽斌，武静威，李鹏飞，景军锋，张缓缓，
申请(专利权)人：西安工程大学，
类型：发明
国别省市：