[发明专利]图像处理方法、装置、计算机设备及存储介质

权利要求书

1.一种图像处理方法，其特征在于，所述方法包括：

获取第一图像和第一掩码图像，所述第一图像为第一产品的表面纹理图像，所述第一掩码图像用于标识所述第一图像中缺陷所在的目标区域；

在所述第一掩码图像中，对所述目标区域进行变换，得到第二掩码图像；

基于所述第二掩码图像，将所述目标区域与第二图像进行融合，得到目标图像，所述第二图像为第二产品的表面纹理图像；

基于多个所述目标图像和多个所述第二掩码图像，对初始分割模型进行参数调整，得到缺陷分割模型，所述缺陷分割模型用于识别第二产品的表面纹理图像中缺陷所在的目标区域。

2.根据权利要求1所述的方法，其特征在于，所述在所述第一掩码图像中，对所述目标区域进行变换，得到第二掩码图像包括：

从缺陷角度区间中，随机采样得到目标角度；

在所述第一掩码图像中，按照所述目标角度对所述目标区域进行旋转变换，得到所述第二掩码图像。

3.根据权利要求1所述的方法，其特征在于，所述在所述第一掩码图像中，对所述目标区域进行变换，得到第二掩码图像包括：

从缺陷尺寸区间中，随机采样得到目标尺寸；

在所述第一掩码图像中，将所述目标区域的尺寸变换为所述目标尺寸，得到所述第二掩码图像。

4.根据权利要求1所述的方法，其特征在于，所述在所述第一掩码图像中，对所述目标区域进行变换，得到第二掩码图像包括：

响应于所述第二图像与所述第一掩码图像尺寸相同，从所述第一掩码图像中随机采样得到目标位置；在所述第一掩码图像中，将所述目标区域平移至所述目标位置，得到所述第二掩码图像；或者，

响应于所述第二图像与所述第一掩码图像尺寸不相同，在所述第二图像中随机采样得到目标位置；将所述第一掩码图像的尺寸缩放成与所述第二图像相同的尺寸；在尺寸缩放后的第一掩码图像中，将所述目标区域平移至所述目标位置，得到所述第二掩码图像。

5.根据权利要求4所述的方法，其特征在于，所述将所述目标区域平移至所述目标位置之后，所述方法还包括：

响应于平移后的目标区域的边缘超出所述第一掩码图像的边缘，基于所述第一掩码图像的边缘对所述平移后的目标区域进行截断。

6.根据权利要求1所述的方法，其特征在于，所述基于所述第二掩码图像，将所述目标区域与第二图像进行融合，得到目标图像包括：

基于所述第二掩码图像，获取变换后的目标区域；

将所述变换后的目标区域叠加至所述第二图像，得到第三图像；

对所述第三图像进行平滑处理，得到所述目标图像。

7.根据权利要求6所述的方法，其特征在于，所述将所述变换后的目标区域叠加至所述第二图像，得到第三图像包括：

对所述变换后的目标区域与所述第二图像进行泊松图像融合，得到所述第三图像。

8.一种图像处理装置，其特征在于，所述装置包括：

获取模块，用于获取第一图像和第一掩码图像，所述第一图像为第一产品的表面纹理图像，所述第一掩码图像用于标识所述第一图像中缺陷所在的目标区域；

变换模块，用于在所述第一掩码图像中，对所述目标区域进行变换，得到第二掩码图像；

融合模块，用于基于所述第二掩码图像，将所述目标区域与第二图像进行融合，得到目标图像，所述第二图像为第二产品的表面纹理图像；

参数调整模块，用于基于多个所述目标图像和多个所述第二掩码图像，对初始分割模型进行参数调整，得到缺陷分割模型，所述缺陷分割模型用于识别第二产品的表面纹理图像中缺陷所在的目标区域。

9.一种计算机设备，其特征在于，所述计算机设备包括一个或多个处理器和一个或多个存储器，所述一个或多个存储器中存储有至少一条计算机程序，所述至少一条计算机程序由所述一个或多个处理器加载并执行以实现如权利要求1至权利要求7任一项所述的图像处理方法。

10.一种存储介质，其特征在于，所述存储介质中存储有至少一条计算机程序，所述至少一条计算机程序由处理器加载并执行以实现如权利要求1至权利要求7任一项所述的图像处理方法。