[发明专利]一种图像数据增强的方法、装置及电子设备

说明书

本发明提供了一种图像数据增强的方法、装置及电子设备，涉及图像处理的技术领域，包括获取图像样本、若干个图像增强策略；将所述若干个图像增强策略进行分类；基于所述若干个图像增强策略的类别进行优先级排序，得到预设顺序；从所述若干个图像增强策略随机选取符合预设数量的目标增强策略；基于所述预设顺序对所述图像样本执行所述目标增强策略，得到增强图像。本申请减少了图像数据增强需人工调整的参数数量，降低了图像数据增强的应用门槛。

技术领域

本发明涉及图像数据处理的技术领域，尤其涉及一种图像数据增强的方法、装置及电子设备。

背景技术

近年来，神经网络在图像识别、物体检测、场景分割等任务上取得了显著的成效。为了提高神经网络的鲁棒性和泛化能力，往往需要利用大规模的数据对神经网络进行训练。在现实情况中，经常会存在训练数据不足的情况，此时，数据增强技术就显得尤为重要。利用数据增强技术对训练数据进行数据增强，能够增大训练数据的数量并且提高训练数据的多样性。

目前，数据增强步骤一般都需要人工调整较多的参数，在若干个数据增强策略中勾选数据增强策略、选择执行数据增强策略的执行顺序、考虑数据增强策略出现的概率、选择数据增强策略的扭曲程度等。

因此，提出一种图像数据增强的方法、装置及电子设备。

发明内容

本说明书提供一种图像数据增强的方法、装置及电子设备，减少了图像数据增强需人工调整的参数数量，降低了图像数据增强的应用门槛。

本申请提供的一种图像数据增强的方法采用如下的技术方案，包括：

获取图像样本、若干个图像增强策略；

将所述若干个图像增强策略进行分类；

基于所述若干个图像增强策略的类别进行优先级排序，得到预设顺序；

从所述若干个图像增强策略随机选取符合预设数量的目标增强策略；

基于所述预设顺序对所述图像样本执行所述目标增强策略，得到增强图像。

可选的，所述基于所述预设顺序对所述图像样本执行所述目标增强策略，得到增强图像，包括：

当所述目标增强策略包含多个类别的图像增强策略时，将所述图像样本按照预设顺序执行所述目标增强策略，得到第一增强图像；

当所述目标增强策略包含一个类别的图像增强策略时，将所述图像样本按照随机顺序执行所述目标增强策略，得到第二增强图像。

可选的，所述图像数据增强的方法还包括：

基于关键图像在所述图像样本加入矩形框；

判断所述增强图像是否需要进行矩形框变换；

当所述增强图像需要进行矩形框变换时，将所述增强图像进行矩形框变换，得到目标图像。

可选的，所述判断所述增强图像是否需要进行矩形框变换，包括：

所述若干个图像增强策略的类别包括第一特征分类的数据增强策略、第二特征分类的数据增强策略、第三特征分类的数据增强策略；

判断所述目标增强策略是否包括所述第二特征分类的图像增强策略或所述第三特征分类的图像增强策略；

当所述目标增强策略包括所述第二特征分类的图像增强策略或所述第三特征分类的图像增强策略，判断所述增强图像的矩形框是否发生位移。

可选的，所述判断所述增强图像的矩形框是否发生位移，包括：

获取矩形框坐标、关键图像坐标；