[发明专利]产品缺陷检测方法和装置

权利要求书

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

构建可变形卷积缺陷检测模型，所述可变形卷积缺陷检测模型包括第一检测模块、第二检测模块、第三检测模块和第四检测模块，所述第二检测模块分别与所述第一检测模块和所述第三检测模块相连，所述第二检测模块和所述第三检测模块通过ROI Align网络连接所述第四检测模块，其中，

所述第一检测模块为残差网络，所述第二检测模块为FPN网络，所述第三检测模块为RPN网络，所述第四检测模块为Head网络，具体地，所述第一检测模块为残差网络包括第一梯度、第二梯度、第三梯度和第四梯度，并且所述第一梯度、所述第二梯度、所述第三梯度和所述第四梯度均由层叠残差网络构成，并且所述层叠残差网络包括第一类层叠残差网络和第二类层叠残差网络，其中，所述第一类层叠残差网络包含DCL网络，所述第二类层叠残差网络不包含DCL网络，此外，所述第一类层叠残差网络和所述第二类层叠残差网络均包含X1-Y1、X2-Y2、X3-Y3、X4-Y4四个映射通道，并且在所述第一类层叠残差网络的所述X2-Y2、所述X3-Y3、所述X4-Y4映射通道中层叠设置3*3DCBR网络，在所述第二类层叠残差网络的所述X2-Y2、所述X3-Y3、所述X4-Y4映射通道中层叠设置3*3CBR网络；

获取待检测产品的图像数据；

均衡处理所述图像数据以得到样本数据；

根据所述样本数据训练所述可变形卷积缺陷检测模型；

采用训练后所述可变形卷积缺陷检测模型对所述待检测产品进行缺陷检测。

2.根据权利要求1所述的产品缺陷检测方法，其特征在于，其中，

所述第一梯度包括三个所述第二类层叠残差网络；

所述第二梯度包括四个所述第一类层叠残差网络；

所述第三梯度包括六个所述第一类层叠残差网络；

所述第四梯度包括三个所述第一类层叠残差网络。

3.根据权利要求2所述的产品缺陷检测方法，其特征在于，所述待检测产品包括多个项目平台产品，所述图像数据包括所述多个项目平台产品的图像数据。

4.根据权利要求3所述的产品缺陷检测方法，其特征在于，均衡处理所述图像数据以得到样本数据，具体包括以下步骤：

采用CycleGAN网络均衡处理所述图像数据以得到样本数据。

5.根据权利要求4所述的产品缺陷检测方法，其特征在于，根据所述样本数据训练所述可变形卷积缺陷检测模型，具体包括以下步骤：

对所述样本数据进行标注分配以得到训练集、验证集和测试集；

对所述训练集在线数据增强操作，以得到待输入训练集；

对所述测试集和所述验证集分别进行TTA操作，以对应得到待输入测试集和待输入验证集；

根据所述待输入训练集、所述待输入测试集和所述待输入验证集训练所述可变形卷积缺陷检测模型。