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

权利要求书

1.一种图像数据增强的方法，其特征在于，包括：

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

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

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

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

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

2.如权利要求1所述的图像数据增强的方法，其特征在于，所述基于所述预设顺序对所述图像样本执行所述目标增强策略，得到增强图像，包括：

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

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

3.如权利要求1所述的图像数据增强的方法，其特征在于，还包括：

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

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

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

4.如权利要求3所述的图像数据增强的方法，其特征在于，所述判断所述增强图像是否需要进行矩形框变换，包括：

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

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

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

5.如权利要求4所述的图像数据增强的方法，其特征在于，所述判断所述增强图像的矩形框是否发生位移，包括：

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

将所述关键图像坐标与所述矩形框坐标进行比对，判断所述关键图像坐标是否与所述矩形框坐标重叠，或，判断所述关键图像坐标是否位于所述矩形框坐标内；

当所述关键图像坐标与所述矩形框坐标重叠，或，所述关键图像坐标位于所述矩形框坐标内时，所述增强图像无需进行矩形框变换；

当所述关键图像坐标不与所述矩形框坐标重叠，或，所述关键图像坐标不位于所述矩形框坐标内时，所述增强图像需要进行矩形框变换。

6.如权利要求5所述的图像数据增强的方法，其特征在于，所述当所述增强图像需要进行矩形框变换时，将所述增强图像进行矩形框变换，得到目标图像，包括：

当所述关键图像坐标不与所述矩形框坐标重叠时，移动矩形框直至所述关键图像坐标与所述矩形框坐标重叠，或，放大矩形框直至所述关键图像坐标在所述矩形框坐标内，得到第一目标图像；

当所述关键图像坐标不位于所述矩形框坐标内时，移动矩形框直至所述关键图像坐标与所述矩形框坐标重叠，或，放大矩形框直至所述关键图像坐标在所述矩形框坐标内，得到第二目标图像。

7.如权利要求2所述的图像数据增强的方法，其特征在于，所述判断所述增强图像是否需要进行矩形框变换，包括：

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

8.一种图像数据增强的装置，其特征在于，包括：

获取模块，用以获取图像样本、若干个图像增强策略；

分类模块，用以将所述若干个图像增强策略进行分类；

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

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

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