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

权利要求书

1.一种工业检测中的工件表面小缺陷检测方法，其特征在于，包括以下步骤：

步骤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值，对第一放大图像I2的所有GT框的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，采用增强后的数据集进行所述工件表面缺陷检测模型的训练。

2.根据权利要求1所述的工业检测中的工件表面小缺陷检测方法，其特征在于，采用邻近插值法将所述工件图像放大设定倍数和将所述第二Mask图像M2放大设定倍数。

3.根据权利要求2所述的工业检测中的工件表面小缺陷检测方法，其特征在于，所述设定倍数为16倍。

4.一种非临时性计算机可读存储介质，其特征在于，其上存储有计算机程序，该程序被处理器执行时实现根据权利要求1-3中任一项所述的工业检测中的工件表面小缺陷检测方法。