[发明专利]工件缺陷检测方法和装置

权利要求书

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

获取待检测工件的缺陷标注数据；

采用图像像素分解重构算法对所述缺陷标注数据进行增强；

根据增强后的所述缺陷标注数据构建缺陷检测模型；

根据所述缺陷检测模型对所述待检测工件进行缺陷检测；

其中，采用图像像素分解重构算法对所述缺陷标注数据进行增强，包括以下步骤：对所述缺陷标注数据进行逐像素拆分并聚类得到原子数据；采用多种数据增强算法分别对所述原子数据进行数据增强得到相应的扩张数据；采用卷积神经网络对所述扩张数据的准确率进行评判，并根据评判结果选择三种所述数据增强算法作为最终的数据增强算法；根据所述最终数据增强算法对所述缺陷标注数据进行增强。

2.根据权利要求1所述的工件缺陷检测方法，其特征在于，获取待检测工件的缺陷标注数据，包括以下步骤：

获取所述待检测工件的原始数据；

采用语义分割算法对所述原始数据进行缺陷标注得到所述缺陷标注数据。

3.根据权利要求2所述的工件缺陷检测方法，其特征在于，对所述缺陷标注数据进行逐像素拆分并聚类得到原子数据，包括以下步骤：

根据缺陷类别对所述缺陷标注数据进行逐像素拆分；

将含有相同缺陷类别的像素聚类；

判断聚类后的含有相同缺陷类别的像素之间的距离是否超过预设距离；

若是，则对聚类后的含有相同缺陷类别的像素进行拆分转换得到相应的原子数据。

4.根据权利要求3所述的工件缺陷检测方法，其特征在于，根据增强后的所述缺陷标注数据构建缺陷检测模型，包括以下步骤：

根据增强后的所述缺陷标注数据得到训练样本和测试样本；

根据所述训练样本进行迭代训练得到所述缺陷检测模型，并采用所述测试样本对所述缺陷检测模型进行测试。

5.一种工件缺陷检测装置，其特征在于，包括：

获取模块，所述获取模块用于获取待检测工件的缺陷标注数据；

数据增强模块，所述数据增强模块采用图像像素分解重构算法对所述缺陷标注数据进行增强；

建模模块，所述建模模块用于根据增强后的所述缺陷标注数据构建缺陷检测模型；

检测模块，所述检测模块用于根据所述缺陷检测模型对所述待检测工件进行缺陷检测；

其中，所述数据增强模块具体用于对所述缺陷标注数据进行逐像素拆分并聚类得到原子数据，采用多种数据增强算法分别对所述原子数据进行数据增强得到相应的扩张数据，采用卷积神经网络对所述扩张数据的准确率进行评判，并根据评判结果选择三种所述数据增强算法作为最终的数据增强算法，根据所述最终数据增强算法对所述缺陷标注数据进行增强。

6.一种计算机设备，包括存储器、处理器及存储在存储器上并可在处理.器上运行的计算机程序，其特征在于，所述处理器执行所述计算机程序时，实现根据权利要求1-4中任一项所述的工件缺陷检测方法。

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