[发明专利]一种表面缺陷数据增强方法、装置、电子设备及存储介质

权利要求书

1.一种表面缺陷数据增强方法，其特征在于，包括：

获取待增强的表面缺陷数据；所述待增强的表面缺陷数据包括：待增强的齿轮表面缺陷数据；

获取待增强部位信息；

根据待增强部位信息对表面缺陷数据进行划分；所述根据待增强部位信息对表面缺陷数据进行划分，包括：根据待增强部位信息将齿轮表面缺陷数据划分为齿面缺陷数据和齿底缺陷数据；

对划分后的表面缺陷数据分别进行数据增强，以获取用于对表面缺陷检测网络进行训练的训练数据，使得所述表面缺陷检测网络对不同的待增强部位进行缺陷检测。

2.根据权利要求1所述的表面缺陷数据增强方法，其特征在于，在所述根据待增强部位信息对表面缺陷数据进行划分前，所述方法还包括：

对所述表面缺陷数据进行初次数据增强，获得经过初次数据增强后的表面缺陷数据；

所述根据待增强部位信息对表面缺陷数据进行划分，包括：根据待增强部位信息对初次数据增强后的表面缺陷数据进行划分。

3.根据权利要求2所述的表面缺陷数据增强方法，其特征在于，其中，所述初次数据增强的方式包括：基于几何变换的数据增强、基于色彩变换的数据增强、以及基于像素变换的数据增强中的至少一者。

4.根据权利要求1所述的表面缺陷数据增强方法，其特征在于，所述对划分后的表面缺陷数据分别进行数据增强的增强方式，包括：基于几何变换的数据增强、基于色彩变换的数据增强、以及基于像素变换的数据增强中的至少一者。

5.根据权利要求1所述的表面缺陷数据增强方法，其特征在于，所述待增强部位信息包括：齿面数据模板与齿底数据模板；

所述根据待增强部位信息对齿轮表面缺陷数据进行划分，包括：

针对齿面数据模板和齿底数据模板，采用特征点匹配法将齿轮表面缺陷数据划分为齿面缺陷数据和齿底缺陷数据。

6.根据权利要求1所述的表面缺陷数据增强方法，其特征在于，所述对划分后的表面缺陷数据分别进行数据增强，包括：

分别获取划分后的表面缺陷数据的低频分量；

根据低频分量对划分后的表面缺陷数据分别进行数据增强。

7.一种表面缺陷数据增强装置，其特征在于，包括：

待增强表面缺陷数据获取模块，用于获取待增强的表面缺陷数据；所述待增强的表面缺陷数据包括：待增强的齿轮表面缺陷数据；

待增强部位信息获取模块，用于获取待增强部位信息；

数据划分模块，用于根据待增强部位信息对表面缺陷数据进行划分；所述根据待增强部位信息对表面缺陷数据进行划分，包括：根据待增强部位信息将齿轮表面缺陷数据划分为齿面缺陷数据和齿底缺陷数据；

数据增强模块，用于对划分后的表面缺陷数据分别进行数据增强，以获取用于对表面缺陷检测网络进行训练的训练数据，使得所述表面缺陷检测网络对不同的待增强部位进行缺陷检测。

8.一种电子设备，其特征在于，包括：处理器、存储器和总线，其中，

所述处理器和所述存储器通过所述总线完成相互间的通信；

所述存储器存储有可被所述处理器执行的程序指令，所述处理器调用所述程序指令能够执行如权利要求1~6任一项所述的方法。

9.一种计算机可读存储介质，其特征在于，该计算机可读存储介质上存储有计算机程序，该计算机程序被处理器运行时执行如权利要求1~6任一所述的方法。