[发明专利]数据增强策略搜索方法、装置、设备以及存储介质

权利要求书

1.一种数据增强策略搜索方法，包括：

获取模型的训练脚本、训练数据集和数据增强策略配置信息；

基于所述数据增强策略配置信息，生成数据增强策略搜索空间；

在所述数据增强策略搜索空间中对所述训练数据集进行数据增强，以及利用数据增强的训练数据集对所述训练脚本进行模型训练，自动搜索所述数据增强策略搜索空间中的数据增强策略的目标增强值。

2.根据权利要求1所述的方法，其中，所述在所述数据增强策略搜索空间中对所述训练数据集进行数据增强，以及利用数据增强的训练数据集对所述训练脚本进行模型训练，自动搜索所述数据增强策略搜索空间中的数据增强策略的目标增强值，包括：

对于所述数据增强策略搜索空间中的数据增强策略的候选增强值，将所述训练数据集作为第一组件，调用该候选增强值进行数据增强，得到该候选增强值对应的训练数据集，以及将所述训练脚本作为第二组件，调用该的候选增强值对应的训练数据集进行模型训练，得到该候选增强值对应的模型；

将满足预设效果的模型对应的候选增强值确定为所述目标增强值。

3.根据权利要求1所述的方法，其中，所述获取数据增强策略配置信息，包括：

将所述数据增强策略配置信息作为第三组件，以及从组件化配置信息提交接口获取所述数据增强策略配置信息，其中，所述数据增强策略配置信息的提交方式包括在线提交或命令行提交。

4.根据权利要求1所述的方法，其中，所述方法还包括：

利用所述目标增强值对所述训练数据集进行数据增强，得到目标训练数据集；

利用所述目标训练数据集进行训练，得到目标模型。

5.根据权利要求1-4之一所述的方法，其中，数据增强策略包括以下至少一种：水平翻转、垂直翻转、随机剪裁、高斯模糊、图像锐度和图像旋转。

6.一种数据增强策略搜索装置，包括：

模型信息获取模块，被配置成获取模型的训练脚本、训练数据集和数据增强策略配置信息；

搜索空间生成模块，被配置成基于所述数据增强策略配置信息，生成数据增强策略搜索空间；

增强策略搜索模块，被配置成在所述数据增强策略搜索空间中对所述训练数据集进行数据增强，以及利用数据增强的训练数据集对所述训练脚本进行模型训练，自动搜索所述数据增强策略搜索空间中的数据增强策略的目标增强值。

7.根据权利要求6所述的装置，其中，所述增强策略搜索模块进一步被配置成：

对于所述数据增强策略搜索空间中的数据增强策略的候选增强值，将所述训练数据集作为第一组件，调用该候选增强值进行数据增强，得到该候选增强值对应的训练数据集，以及将所述训练脚本作为第二组件，调用该候选增强值对应的训练数据集进行模型训练，得到该候选增强值对应的模型；

将满足预设效果的模型对应的候选增强值确定为所述目标增强值。

8.根据权利要求6所述的装置，其中，所述模型信息获取模块进一步被配置成：

将所述数据增强策略配置信息作为第三组件，以及从组件化配置信息提交接口获取所述数据增强策略配置信息，其中，所述数据增强策略配置信息的提交方式包括在线提交或命令行提交。

9.根据权利要求6所述的装置，其中，所述装置还包括：

数据增强模块，被配置成利用所述目标增强值对所述训练数据集进行数据增强，得到目标训练数据集；

模型训练模块，被配置成利用所述目标训练数据集进行训练，得到目标模型。

10.根据权利要求6-9之一所述的装置，其中，数据增强策略包括以下至少一种：水平翻转、垂直翻转、随机剪裁、高斯模糊、图像锐度和图像旋转。