[发明专利]一种基于深度学习的接箍外壁检测方法及系统

说明书

本发明公开了一种基于深度学习的接箍外壁检测方法及系统，包括如下步骤：获取缺陷样本集，对缺陷样本集中的缺陷位置进行标注，所述缺陷样本集由多个均具有缺陷的原始接箍外壁图像组成；将缺陷样本集中的每个原始接箍外壁图像均划分为m个网格，分别获取待检测目标所处的网格作为候选框；将候选框输入已构建完成的神经网络模型中，输出携带有权重参数的模型，对携带有权重参数的候选框进行计算，得到并保存用于匹配样本缺陷位置的模型，完成对神经网络模型的训练；通过已训练完成的神经网络模型对实际生产中的待检测图像进行缺陷检测，输出标识出缺陷位置的待检测图像；该接箍外壁检测方法提高了接箍外壁检测的准确性和鲁棒性。

技术领域

本发明涉及接箍外壁检测技术领域，尤其涉及一种基于深度学习的接箍外壁检测方法及系统。

背景技术

目前行业内对于接箍的检测，采用传统荧光磁粉探伤机，通过人工上下料，基于半自动磁化与退磁，人工判断表面裂缝或褶皱缺陷。该种方法存在如下缺点：人员劳动强度大；缺陷检测不稳定；紫外荧光灯与荧光液对工人身体健康造成一定伤害。

发明内容

基于背景技术存在的技术问题，本发明提出了一种基于深度学习的接箍外壁检测方法及系统，提高了接箍外壁检测的准确性和鲁棒性。

本发明提出的一种基于深度学习的接箍外壁检测方法，包括如下步骤：

获取缺陷样本集，对缺陷样本集中的缺陷位置进行标注，所述缺陷样本集由多个均具有缺陷的原始接箍外壁图像组成；

将缺陷样本集中的每个原始接箍外壁图像均划分为m个网格，分别获取待检测目标所处的网格作为候选框；

将候选框输入已构建完成的神经网络模型中，输出携带有权重参数的模型，对携带有权重参数的候选框进行计算，得到并保存用于匹配样本缺陷位置的模型，完成对神经网络模型的训练；

通过已训练完成的神经网络模型对实际生产中的待检测图像进行缺陷检测，输出标识出缺陷位置的待检测图像。

进一步地，在将候选框输入已构建完成的神经网络模型中，输出携带有权重参数的模型，对携带有权重参数的候选框进行计算，得到并保存匹配样本缺陷位置的模型，完成对神经网络模型的训练之后，包括：

获取训练样本集，训练样本集中由多个均具有缺陷的原始接箍外壁图像组成；

将训练样本集输入到已训练完成的神经网络模型中，分别输出标识出缺陷位置的图像；

对标识出缺陷位置的图像进行正确率验证，以验证神经网络模型；

当正确率不大于设定值，则对神经网络模型重新训练后，再进行正确率验证验证；

当正确率大于设定值，则神经网络模型验证完成。

进一步地，在将将缺陷样本集中的每个原始接箍外壁图像均划分为m个网格，分别获取待检测目标所处的网格作为候选框，具体包括：

获取已标注的缺陷位置所处的网格；

选择缺陷位置所处的网格作为待检测区域，输出待检测区域所处的网格作为候选框。

进一步地，将候选框输入已构建完成的神经网络模型中，输出携带有权重参数的模型，对携带有权重参数的候选框进行计算，得到并保存用于匹配样本缺陷位置的模型，完成对神经网络模型的训练，其中，对携带有权重参数的候选框进行计算，具体包括：

获取携带有权重参数的候选框所处网格的目标类别；

将携带有权重参数的候选框与所处网格的目标类别的乘积最大值作为携带有权重参数的候选框的类别，将携带有权重参数的候选框的类别作为所述接箍外壁图像缺陷位置的置信度；

利用置信度与非极大值抑制过滤冗余边框，得到用于匹配样本缺陷位置的模型。