一种基于轻量化网络的工业产品表面缺陷检测方法技术

本发明专利技术公开了一种基于轻量化网络的工业产品表面缺陷检测方法，所述方法包括以下步骤：确定深度学习网络模型的结构；将训练样本输入网络模型进行训练，其中，在使用网络对样本进行检测之前，需要对网络的参数进行训练，使每一个参数达到合适的数值；保存训练获得的参数文件；将参数赋值给网络并对测试样本进行检测；根据检测结果对样本图像中的缺陷进行标注。故因此，本发明专利技术的有益效果在于使用更少的参数的同时，达到了更高的检测准确率。达到了更高的检测准确率。达到了更高的检测准确率。

【技术实现步骤摘要】
一种基于轻量化网络的工业产品表面缺陷检测方法


[0001]本专利技术属于产品表面缺陷检测的研究
，特别涉及一种基于轻量化网络的工业产品表面缺陷检测方法。
技术背景
[0002]表面缺陷检测是在工业检测中产品质量保障的重要手段，缺陷检测可以找出有表面缺陷的产品，从而避免这些有质量问题的产品在之后的生产建设环节中造成隐患，减少经济损失。不同的产品，缺陷的大小形状也不相同，而针对这些不同的缺陷，往往需要用到不同的方法，如木材的表面缺陷通常以裂痕为主，形状比较固定形，但通常比较细小，容易漏检，而钢材的表面缺陷则多种多样，用传统图像处理方法可能难以完全检测出来。
[0003]表面缺陷检测作为保障产品质量的重要手段，被用于各种各样的产品，比如，被用于纺织品、轮胎、钢条、芯片。随着科研的进展，表面缺陷检测主要运用的技术也不断地变化，大致可以分为三个阶段。
[0004]首先是人工检测，这是最原始的表面缺陷检测方式，显然这种方式极度依赖于检测人员的经验，并且准确率也不高，同时，这样的方法也消耗了大量的人力成本。
[0005]接下来是以图像处理和机器学习为主的传统方法。针对图像处理，以模板匹配(Template Matching)为例，该方法被广泛应用于工业的表面缺陷检测，归一化交叉相关(Normalized Cross Correlation,NCC)是经常被用于实现模板匹配的方法，它可以检测模板和待检测图像的非相似性。而机器学习相关的方法一般基于特征提取的方法，如结合了特征提取和SVM的针对热轧的检测方法，以及通过提取集合特征和小波特征，针对印制电路板的检测方法。这些基于图像处理和机器学习的方法，被广泛应用于工业检测，但这些方法往往十分复杂，且难以优化。
[0006]最后是目前业内主要研究的基于深度学习的检测方法。AlexNet的提出使得深度学习得到了广泛的关注，而在深度学习领域卷积神经网络对目标检测和语义分割都有着巨大的贡献，其中有代表性的网络模型有Faster R
‑
CNN、RetinaNet、SSD、FPN、YOLOv3等。大量基于这些深度学习网络的模型，进行适当地优化之后，都获得了不错的检测效果。
[0007]在实际的工业生产环境中，基于图像处理和机器学习的传统方法仍然是主流，但这些方法普遍有一些局限性。首先，这些方法通常会严重的受图像的噪声和图像中光照的影响，其次，特征提取需要根据经验去设计合理的方法，这是十分复杂的，会耗费一定人力成本，最后，传统的方法没有很好的通用性，一个方法设计好之后，很难适用别的类型的产品，因此针对不同种类的产品，不同种类的缺陷，需要设计不同的方法。以上面提到的NCC为例，目前已有的基于NCC的模板匹配方法，都是位移敏感的，且对于不同类型的产品，需要有不同的模板，没有很好地通用性。而大部分基于图像处理和机器学习的方法，往往十分复杂，且难以优化。
[0008]而反观基于深度学习的方法，往往拥有较高的通用性，且对于图像噪声的处理也更好。但现有的深度学习网络结构通常十分复杂，检测的耗时往往较大，而工业检测对实时
性又恰好有较高的要求。为解决这一问题，比较常见的思路是使用一个轻量级的网络，来减少每个样本的检测时间，然而减小网络层数的代价就是，检测的准确率往往不是太好。

技术实现思路

[0009]鉴于现有技术的缺陷，本专利技术旨在于提供一种基于轻量化网络的工业产品表面缺陷检测方法，为了平衡检测的实时性与检测准确率这两点，本专利技术基于非对称卷积、深度可分离卷积、SE注意力模块，YOLOv3的检测方法，以及其他的网络结构优化。该网络最后的参数数量约为YOLOv3
‑
Tiny的一半，符合轻量级的要求。而在实验阶段，通过在DAGM2007数据集上进行检测，检测率和召回率均超过98％，而基于YOLOv3
‑
Tiny，在该数据集上的召回率仅为81.8％，故本专利技术的网络结构在使用更少的参数的同时，达到了更高的检测准确率。最后，对于检测速度，在一张1080Ti显卡下，达到了每个样本小于10ms的检测时间，满足了实时性。
[0010]为了实现上述目的，本专利技术采用的技术方案如下：
[0011]一种基于轻量化网络的工业产品表面缺陷检测方法，所述方法包括以下步骤：
[0012]S1确定深度学习网络模型的结构；
[0013]S2将训练样本输入网络模型进行训练，其中，在使用网络对样本进行检测之前，需要对网络的参数进行训练，使每一个参数达到合适的数值；
[0014]S3保存训练获得的参数文件；
[0015]S4将参数赋值给网络并对测试样本进行检测；
[0016]S5根据检测结果对样本图像中的缺陷进行标注。
[0017]需要说明的是，还包括在步骤1与步骤2之间的优化训练样本的标记步骤，其中，在将训练样本输入进网络模型前，根据不同的网络模型，需要对训练样本的标记进行一定的预处理。
[0018]需要说明的是，所述预处理是求出一个矩形，将椭圆标记包括在内，具体的求法是，分别找到椭圆内所有点横纵坐标的最大值和最小值，以此分别求得矩形标记框的位置和大小。
[0019]需要说明的是，所述步骤S3包括在每一轮训练的过程中，会将训练样本的一部分，作为训练阶段的测试样本。
[0020]需要说明的是，所述步骤4中包括将训练获得的参数，赋值给网络模型，将测试样本作为网络模型的输入，网络模型会经过运算后，得到一个输出，该输出为测试样本图像中所有缺陷的位置及大小的预测。
[0021]需要说明的是，所述网络模型的结构共有23层，在0
‑
7层的网络前端，采用了非对称卷积进行特征提取；其中，将输入分别用3*3、1*3、3*1的卷积核进行卷积，再分别进行批标准化，最后将得到的结果相加，再使用LeakyReLU进行激活，作为最终的结果。
[0022]需要说明的是，在8
‑
12层的网络中部，采用的是深度可分离卷积降低参数数量，即对于普通3*3的卷积，所需要的参数为输入通道数*输出通道数*3*3，其中，参数为输入通道数*3*3+输入通道数*输出通道数*1*1，当输出通道数较大时，深度可分离卷积所需要的参数数量约为普通3*3卷积的3*3分之一，即9分之一。
[0023]需要说明的是，在13层的网络中部设有SE注意力模块，用于加强网络对特征图的
每一个通道的相关性的学习。
[0024]需要说明的是，在0
‑
12层的网络主干部分的下采样采用了Stride为2的3*3卷积。
[0025]本专利技术的有益效果在于，该网络的参数数量仅为YOLOv3
‑
Tiny的一半，且该网络在DAGM2007数据集上，缺陷检测准确率远超YOLOv3
‑
Tiny。具体的说：
[0026]1、在网络的前端，采用非对称卷积，提升网络在前期提取特征时，能在水平和竖直方向上分别进行特征的提取，增强网络对本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于轻量化网络的工业产品表面缺陷检测方法，其特征在于，所述方法包括以下步骤：S1确定深度学习网络模型的结构；S2将训练样本输入网络模型进行训练，其中，在使用网络对样本进行检测之前，需要对网络的参数进行训练，使每一个参数达到合适的数值；S3保存训练获得的参数文件；S4将参数赋值给网络并对测试样本进行检测；S5根据检测结果对样本图像中的缺陷进行标注。2.根据权利要求1所述的基于轻量化网络的工业产品表面缺陷检测方法，其特征在于，还包括在步骤1与步骤2之间的优化训练样本的标记步骤，其中，在将训练样本输入进网络模型前，根据不同的网络模型，需要对训练样本的标记进行一定的预处理。3.根据权利要求2所述的基于轻量化网络的工业产品表面缺陷检测方法，其特征在于，所述预处理是求出一个矩形，将椭圆标记包括在内，具体的求法是，分别找到椭圆内所有点横纵坐标的最大值和最小值，以此分别求得矩形标记框的位置和大小。4.根据权利要求1所述的基于轻量化网络的工业产品表面缺陷检测方法，其特征在于，所述步骤S3包括在每一轮训练的过程中，会将训练样本的一部分，作为训练阶段的测试样本。5.根据权利要求1所述的基于轻量化网络的工业产品表面缺陷检测方法，其特征在于，所述步骤4中包括将训练获得的参数，赋值给网络模型，将测试样本作为网络模型的输入，网络模型会经过运算后，得到...

【专利技术属性】
技术研发人员：赖剑煌，曹阳，冯展祥，
申请(专利权)人：中山大学，
类型：发明
国别省市：