[发明专利]工业检测中的工件表面小缺陷检测方法

说明书

本发明涉及工业检测技术领域，提供一种工业检测中的工件表面小缺陷检测方法，首先利用缺陷搜索算法获取模型最优阈值thr_best，然后根据模型最优阈值thr_best搜索出工件表面缺陷检测模型的数据集潜在的缺陷位置，再根据低秩矩阵增强算法仅对潜在的缺陷位置图像进行低秩矩阵增强，采用增强后的图像进行检测模型训练，不仅可以降低算法处理的时间，还可以有效提高模型对小缺陷的检出率，满足实际工业生产要求，且无需高分辨率的相机，可以有效控制成本。

技术领域

本发明涉及工业检测技术领域，具体涉及一种工业检测中的工件表面小缺陷检测方法和一种非临时性计算机可读存储介质。

背景技术

工业质检中，小缺陷（像素总数小于400的缺陷）属于面积小但又比较严重的缺陷。目前，工业质检模型在提取图像特征的过程中会对图片进行多次下采样操作，小缺陷很容易在下采样操作的过程中被过滤掉，导致模型最终检测不出这些小缺陷。

相关技术中，为了提高小缺陷的检出率，通过采用高分辨的相机进行图像采集，以提高小缺陷的所占像素的量，该方式存在以下问题：1、高分辨率的相机比一般的相机成本高出数倍，导致工业检测的成本增加；2、高分辨的相机拍出的图片的数据量很大，导致模型在计算时候的需要处理大量的数据，花费大量的时间，影响检测效率。

发明内容

为解决上述技术问题，本发明的第一个目的在于提出工业检测中的工件表面小缺陷检测方法。

本发明的第二个目的在于提出一种非临时性计算机可读存储介质。

本发明采用的技术方案如下：

本发明第一方面的实施例提出了一种工业检测中的工件表面小缺陷检测方法，包括以下步骤：步骤S1，采集工件图像并标注缺陷；步骤S2，将所述工件图像与高斯滤波核进行卷积；步骤S3，将卷积后的图像采用2*2核进行最大池化操作；步骤S4，重复步骤S2-S3两次，以提取特征图像；步骤S5，将所述特征图像与四种方向的局部极值算子进行卷积操作并获取卷积结果，所述四种方向包括：+x方向、+y方向、-x方向和-y方向；步骤S6，对所述卷积结果进行判断，其中，如果所述卷积结果大于阈值thr，则将当前位置像素值设置为1，如果所述卷积结果小于或者等于阈值thr，则将当前位置像素值设置为0，以生成4张二值化图；步骤S7，将所述4张二值化图相同位置的像素值相加得到每个位置的像素值T，并所述每个位置的像素值T进行判断，其中，如果T=4，则将当前位置的像素值设置1，如果T≠4，则将当前位置的像素值设置0，以生成所述工件图像对应的第一Mask图像M1；步骤S8，对所述第一Mask图像M1求最大内接矩形轮廓，将所述内接矩形轮廓内的像素置为1，内接矩形轮廓外的像素置为0，得到第二Mask图像M2；步骤S9，将所述工件图像放大设定倍数生成第一放大图像I2，将所述第二Mask图像M2放大设定倍数生成第二放大图像M3；步骤S10，将所述第二放大图像M3的最大内接矩形轮廓与所述第一放大图像I2的每个GT框求IOU（Intersectionover Union，交并比）值，对第一放大图像I2的所有GT（Ground Truth，缺陷标注）框的IOU值求均值IOU_MEAN；步骤S11，将所述阈值thr从255到0遍历一遍，重复执行步骤S2~S10，将满足IOU_MEAN0.5的阈值thr中的最大值，作为当前工件图像的最优阈值；步骤S12，对所述步骤S1采集的所有工件图像分别执行步骤S2~S11，统计所有工件图像的最优阈值，将所述所有工件图像的最优阈值中的最小的值作为模型最优阈值thr_best；步骤S13，获取工件表面缺陷检测模型的数据集，将所述数据集的图像执行步骤S2-S9，其中，在执行步骤S2-S9时将步骤S5中的阈值设定为所述模型最优阈值thr_best；步骤S14，获取所述数据集的图像中第二放大图像M3像素值为1对应在第一放大图像I2位置的图像块Y，采用低秩矩阵增强算法对所述图像块Y进行图像增强；步骤S15，采用增强后的数据集进行所述工件表面缺陷检测模型的训练。

本发明上述提出的工业检测中的工件表面小缺陷检测方法还可以具有如下附加技术特征：