基于随机神经网络搜索的机油罐标签划痕检测系统及方法

技术领域

本发明涉及机油罐标签划痕检测技术领域，具体为基于随机神经网络搜索的机油罐标签划痕检测系统及方法。

背景技术

机油罐标签划痕其产生主要由生产过程中的机械成因或人为成因造成，其中划痕属于成型不良的一种常见形式。其机械成因主要是由于器械间转移过程中会出现位移偏差，进而造成碰撞，机油罐与器械相互碰撞，容易产生长条状深划痕。人为成因在自动化生产线上出现几率较少，部分情形需要人工转移机油罐时，可能产生磕碰，此外一些误操会产生部分人为划痕，包括工具使用不当造成擦碰造成划伤。而无论是机械成因还是人为成因造成的划痕成型，在外观显示上会严重影响美观。进而影响口碑以及销量。而目前机油罐生产厂的生产线均采用人工外观检测方式，其有以下几个缺点：（1）劳动强度大，生产效率低，长时间劳作尤其夜班工作不仅会造成检测质量会下降，更重要的是身体机能也会随之下降，影响身体健康；（2）人工外检的质量不高，容易产生失误，错分有划痕的产品。

总之，现在的机油罐生产是一个速度很高、高度自动化的生产过程，传统的人工划痕检测已经远远不能满足精细生产的需要，而基于神经网络架构搜索和图像处理技术的机油罐标签划痕目标检测系统，能够有效保证划痕的检测精度。

传统的缺陷检测方法准确率不高。随着深度学习技术的发展，基于卷积神经网络的缺陷检测算法逐渐兴起，其超参数往往难以确定，得不到最好的检测效果。

发明内容

鉴于现有技术中所存在的问题，本发明公开了基于随机神经网络搜索的机油罐标签划痕检测系统，采用的技术方案是，包括：第一传送带、第二传送带、设备计算机、第一扫描摄像头和第二扫描摄像头，所述第一扫描摄像头和第二扫描摄像头均与所述设备计算机连接，所述第一传送带和所述第二传送带垂直设置，所述第一传送带的两侧设置有图像采集单元，通过图像采集单元的设置，能够采集机油罐图像，所述第一传送带之间设有扫描空间，所述第一扫描摄像头设于所述扫描空间的左侧，所述第二扫描摄像头设于所述扫描空间的右侧，所述第一扫描摄像头与所述第二扫描摄像头相对设置，所述第一传送带的右侧设置有报警灯，通过报警灯的设置，能够根据处理单元反馈的信息来决定是否报警，并将识别的有划痕机油罐转移到第二传送带。

本发明还公开了基于随机神经网络搜索的机油罐标签划痕检测方法，采用的技术方案是，包括以下步骤：

S1：收集机油罐标签数据集，并将其构建为训练集、验证集、测试集，对采集的图像做灰度转换，获取其灰度图；

S2：创建CB模块和PB模块，并确定神经网络搜索空间和运算操作类型空间；

S3：以模块结构为最小单元进行堆叠构建主干网络，确定网络结构；

S4：在搜索空间中均匀随机采样邻接矩阵，即得到一个有向无环图(DAG)，利用所述数据集搜索每个所述结构单元中待搜索结构单元的每两个内部节点之间的最优操作；

S5：在运算操作类型空间中，均匀随机采样以确定DAG中除了输入输出外的每一个顶点的运算操作类型；

S6：将所划分的数据集输入到网络模型，利用所述训练集确定所述搜索空间中每个模块的学习参数；利用所述验证集确定所述每个单元模块；

S7：计算网络架构损失并更新学习参数；

S8：根据邻接矩阵和运算操作向量请求模型的性能指标，并且进行循环S5～S8步骤；

S9：根据上述循环迭代的验证集性能指标，返回最优网络模型，并且保存其网络模型权重参数和架构参数。

作为本发明的一种优选技术方案，所述步骤S1中训练集用于训练神经网络的权重参数，验证集用于训练神经网络的架构参数，所述权重参数指神经网络的操作中的权重，所述架构参数指代表待搜索操作的重要程度的参数，所述训练数据集、验证集、测试集都包括若干有划痕机油罐标签图像与无划痕的机油罐标签图像，所述训练数据集、验证集、测试集的具体方法为：采集若干张有划痕机油罐标签图像与无划痕的机油罐标签图像，对其进行裁剪得到相同大小的图像，将所述图像通过灰度处理算法生成对应的灰度图像，将所得到的灰度图像来划分训练数据集、验证集、测试集；其中有划痕的标签图像的类别标签为0，无划痕的机油罐标签图像的类别标签为1。

作为本发明的一种优选技术方案，所述步骤S2中所述神经网络搜索空间包括卷积模块CB和池化模块PB，所述特征提取模块CB一般由三个卷积层和一个跳跃连接组成，第一个卷积层(conv1)输入瓶颈卷积层，其卷积核大小固定为 1×1，第二个卷积层(conv2)采用深度卷积提取特征信息，采用两种深度卷积，即残差可分离卷积和残差空洞可分离卷积，以形成两种conv2，第三层卷积(conv3)层采用点式卷积，其卷积核大小固定为 1×1；根据上述卷积模块CB的说明，设置了四种不同的卷积模块，其中CB1模块包含三个卷积层，conv1为输入瓶颈卷积层，其卷积核大小固定为 1×1，conv2采用残差可分离卷积，其卷积核大小固定为 3×3，conv3层采用点式卷积，其卷积核大小固定为 1×1；CB2模块为包含三个的卷积层，conv1为输入瓶颈卷积层，其卷积核大小固定为 1×1，conv2采用残差可分离卷积，其卷积核大小固定为 5×5，conv3层采用点式卷积，其卷积核大小固定为 1×1；CB3模块包含三个卷积层，conv1为输入瓶颈卷积层，其卷积核大小固定为 1×1，conv2采用残差空洞可分离卷积，其卷积核大小固定为 3×3，conv3层采用点式卷积，其卷积核大小固定为 1×1；CB4模块为包含三个的卷积层，conv1为输入瓶颈卷积层，其卷积核大小固定为 1×1，conv2采用残差空洞可分离卷积，其卷积核大小固定为 5×5，conv3层采用点式卷积，其卷积核大小固定为 1×1，上述卷积层的步长都为1，每个卷积层的中间连接为激活层，激活层是RELU函数；所述池化模块PB包括PB1平均池化模块和PB2最大值池化模块，PB1平均池化模块内核大小设置为 3×3，步长设置为 2×2，PB2最大值池化模块内核大小设置为 3×3，步长设置为 2×2。

作为本发明的一种优选技术方案，所述步骤S3中所述网络结构包括第一网络单元、第二网络单元和第三网络单元，所述第一网络单元包含一个特征提取模块CB和一个池化模块PB组成，然后通过两次最大值合并得到统一特征图；所述第二网络单元包含三个特征提取模块CB和一个池化模块PB组成；第三网络单元包含依次连接的Dropout层、两个全连接层和输出层，输出层的激活函数为Softmax函数，通过全连接层和输出层反馈性能指标。

作为本发明的一种优选技术方案，所述步骤S4中的有向无环图(DAG)由一个集合定义，该集合中包含了由 5个节点所组成的所有可能的 DAG，所述节点即为CB模块或者PB模块，DAG 中的每个节点对应的运算操作有2种可能，即3×3卷积操作和5×5卷积操作，取其中的一种作为该节点的运算操作，此外，DAG 需要有两个特殊的输入和输出节点，他们分别被标记为 in 和 out，分别表示网络结构的输入和输出张量，其中所有的卷积运算都利用了批归一化（Batch Normalization, BN）且激活函数使用 ReLU函数。

作为本发明的一种优选技术方案，所述步骤S7中所述网络架构损失是利用架构损失函数基于所述神经网络的输出和所述图像对应的类别标签向量，计算每个所述结构单元中待搜索结构单元的每两个内部节点之间每个操作对应的损失值；模型通过其架构损失函数损失来表现性能指标，其计算公式为：

其中J(θ)为损失值，

所述更新学习参数是所述神经网络的权重参数，基于所述架构损失计算每个所述结构单元中待搜索结构单元对应的架构损失梯度，基于所述架构损失梯度更新每个所述结构单元中待搜索模块单元对应的学习参数，使得上述损失函数达到最优，其计算公式为：

其中α为学习率，θ

作为本发明的一种优选技术方案，所述步骤S8中所述性能指标是指通过当前的网络结构模型计算训练准确率和验证准确率，在每一轮迭代中，记录验证准确率最高的个体变化曲线，来评估算法的有效性。

本发明的有益效果：本发明通过采集机油罐正反两面的标签图像对其进行划痕检测，并实现有划痕自动报警，根据检测结果对机油罐进行自动分类，相比采用人工的方式节省了大量的人力、物力和财力，同时采用随机神经网络架构搜索方法搜索得最优网络模型，检测准确度更高，能够更加规范生产管理和操作流程。

附图说明

图1为本发明系统的结构示意图；

图2为本发明系统的工作流程示意图；

图3为本发明方法中网络搜索空间中卷积模块和池化模块示意图；

图4为本发明方法中的一种划痕检测任务的神经网络的结构图；

图5为本发明方法的一种随机神经网络架构搜索方法的流程图。

图中：1、设备计算机；2、第一扫描摄像头；3、第二扫描摄像头；4、报警灯；5、第一传送带；6、第二传送带。

实施方式

实施例

如图1至图5所示，本发明公开了基于随机神经网络搜索的机油罐标签划痕检测系统，采用的技术方案是，包括：第一传送带5、第二传送带6、设备计算机1、第一扫描摄像头2、第二扫描摄像头3和报警灯4，所述第一扫描摄像头2和第二扫描摄像头3均与所述设备计算机1连接，所述第一传送带5和所述第二传送带6垂直设置，所述第一传送带5的两侧设置有图像采集单元，所述第一传送带5之间设有扫描空间，所述第一扫描摄像头2设于所述扫描空间的左侧，所述第二扫描摄像头3设于所述扫描空间的右侧，所述第一扫描摄像头2与所述第二扫描摄像头3相对设置，所述第一传送带5的右侧设置有报警灯4。

本发明还公开了基于随机神经网络搜索的机油罐标签划痕检测方法，采用的技术方案是，包括以下步骤：

S1：收集机油罐标签数据集，并将其构建为训练集、验证集、测试集，对采集的图像做灰度转换，获取其灰度图；

S2：创建CB模块和PB模块，并确定神经网络搜索空间和运算操作类型空间；

S3：以模块结构为最小单元进行堆叠构建主干网络，确定网络结构；

S4：在搜索空间中均匀随机采样邻接矩阵，即得到一个有向无环图(DAG)，利用所述数据集搜索每个所述结构单元中待搜索结构单元的每两个内部节点之间的最优操作；

S5：在运算操作类型空间中，均匀随机采样以确定DAG中除了输入输出外的每一个顶点的运算操作类型；

S6：将所划分的数据集输入到网络模型，利用所述训练集确定所述搜索空间中每个模块的学习参数；利用所述验证集确定所述每个单元模块；

S7：计算网络架构损失并更新学习参数；

S8：根据邻接矩阵和运算操作向量请求模型的性能指标，并且进行循环S5～S8步骤；

S9：根据上述循环迭代的验证集性能指标，返回最优网络模型，并且保存其网络模型权重参数和架构参数。

作为本发明的一种优选技术方案，所述步骤S1中训练集用于训练神经网络的权重参数，验证集用于训练神经网络的架构参数，所述权重参数指神经网络的操作中的权重，所述架构参数指代表待搜索操作的重要程度的参数，所述训练数据集、验证集、测试集都包括若干有划痕机油罐标签图像与无划痕的机油罐标签图像，所述训练数据集、验证集、测试集的具体方法为：采集若干张有划痕机油罐标签图像与无划痕的机油罐标签图像，对其进行裁剪得到相同大小的图像，例如1024*1024像素大小，将所述图像通过灰度处理算法生成对应的灰度图像，将所得到的灰度图像来划分训练数据集、验证集、测试集；其中有划痕的标签图像的类别标签为0，无划痕的机油罐标签图像的类别标签为1。

作为本发明的一种优选技术方案，所述步骤S2中所述神经网络搜索空间包括卷积模块CB和池化模块PB，所述特征提取模块CB一般由三个卷积层和一个跳跃连接组成，第一个卷积层(conv1)输入瓶颈卷积层，其卷积核大小固定为 1×1，第二个卷积层(conv2)采用深度卷积提取特征信息，采用两种深度卷积，即残差可分离卷积和残差空洞可分离卷积，以形成两种conv2，第三层卷积(conv3)层采用点式卷积，其卷积核大小固定为 1×1；根据上述卷积模块CB的说明，设置了四种不同的卷积模块，其中CB1模块包含三个卷积层，conv1为输入瓶颈卷积层，其卷积核大小固定为 1×1，conv2采用残差可分离卷积，其卷积核大小固定为 3×3，conv3层采用点式卷积，其卷积核大小固定为 1×1；CB2模块为包含三个的卷积层，conv1为输入瓶颈卷积层，其卷积核大小固定为 1×1，conv2采用残差可分离卷积，其卷积核大小固定为 5×5，conv3层采用点式卷积，其卷积核大小固定为 1×1；CB3模块包含三个卷积层，conv1为输入瓶颈卷积层，其卷积核大小固定为 1×1，conv2采用残差空洞可分离卷积，其卷积核大小固定为 3×3，conv3层采用点式卷积，其卷积核大小固定为 1×1；CB4模块为包含三个的卷积层，conv1为输入瓶颈卷积层，其卷积核大小固定为 1×1，conv2采用残差空洞可分离卷积，其卷积核大小固定为 5×5，conv3层采用点式卷积，其卷积核大小固定为 1×1，上述卷积层的步长都为1，每个卷积层的中间连接为激活层，激活层是RELU函数；所述池化模块PB包括PB1平均池化模块和PB2最大值池化模块，PB1平均池化模块内核大小设置为 3×3，步长设置为 2×2，PB2最大值池化模块内核大小设置为 3×3，步长设置为 2×2。

作为本发明的一种优选技术方案，所述步骤S3中所述网络结构包括第一网络单元、第二网络单元和第三网络单元，所述第一网络单元包含一个特征提取模块CB和一个池化模块PB组成，然后通过两次最大值合并得到统一特征图；所述第二网络单元包含三个特征提取模块CB和一个池化模块PB组成；第三网络单元包含依次连接的Dropout层、两个全连接层和输出层，输出层的激活函数为Softmax函数，通过全连接层和输出层反馈性能指标。

作为本发明的一种优选技术方案，所述步骤S4中的有向无环图(DAG)由一个集合定义，该集合中包含了由 5个节点所组成的所有可能的 DAG，所述节点即为CB模块或者PB模块，DAG 中的每个节点对应的运算操作有2种可能，即3×3卷积操作和5×5卷积操作，取其中的一种作为该节点的运算操作，此外，DAG 需要有两个特殊的输入和输出节点，他们分别被标记为 in 和 out，分别表示网络结构的输入和输出张量，其中所有的卷积运算都利用了批归一化（Batch Normalization, BN）且激活函数使用 ReLU函数。

作为本发明的一种优选技术方案，所述步骤S7中所述网络架构损失是利用架构损失函数基于所述神经网络的输出和所述图像对应的类别标签向量，计算每个所述结构单元中待搜索结构单元的每两个内部节点之间每个操作对应的损失值；模型通过其架构损失函数损失来表现性能指标，其计算公式为：

其中J(θ)为损失值，

所述更新学习参数是所述神经网络的权重参数，基于所述架构损失计算每个所述结构单元中待搜索结构单元对应的架构损失梯度，基于所述架构损失梯度更新每个所述结构单元中待搜索模块单元对应的学习参数，使得上述损失函数达到最优，其计算公式为：

其中α为学习率，θ

作为本发明的一种优选技术方案，所述步骤S8中所述性能指标是指通过当前的网络结构模型计算训练准确率和验证准确率，在每一轮迭代中，记录验证准确率最高的个体变化曲线，来评估算法的有效性。

本发明的工作原理：图像采集单元将采集信息发送给处理单元进行划痕目标检测，当罐体经过所述扫描空间时，所述设备计算机1用于接收所述第一扫描摄像头2和所述第二扫描摄像头3采集图像信号，处理单元将采集的机油罐图像转换为灰度图，将其输入到训练好的通过随机神经网络架构搜索算法搜索到的最优神经网络，检测所述待检测罐体商标是否有划痕，并将检测结果信息反馈到报警单元的报警灯4，报警灯4实现有划痕自动报警并将罐体移动至第二传送带6。

尽管本文较多地使用了传送带、图像采集单元、报警单元等术语，但并不排除使用其它术语的可能性。使用这些术语当且仅当是为了更好地描述和解释本发明。

本发明涉及的电路和机械连接为本领域技术人员采用的惯用手段，可通过有限次试验得到技术启示，属于公知常识。

本文中未详细说明的部件为现有技术。

上述虽然对本发明的具体实施例作了详细说明，但是本发明并不限于上述实施例，在本领域普通技术人员所具备的知识范围内，还可以在不脱离本发明宗旨的前提下做出各种变化，而不具备创造性劳动的修改或变形仍在本发明的保护范围以内。