[发明专利]数据增强方法、装置、计算设备及计算机存储介质

权利要求书

1.一种数据增强方法，包括：

随机生成N组掩模数据，基于所述N组掩模数据生成随机掩模；

对待增强训练图像进行背景提取处理，得到对应的第一背景图像、第一前景图像；

对辅助图像进行背景提取处理，得到对应的第二背景图像，其中，所述辅助图像为从训练图像样本中随机选取的一训练图像；

根据所述随机掩模，对所述第一背景图像及所述第二背景图像进行背景拼接处理，得到第三背景图像；

对第一前景图像及第三背景图像进行叠加处理，得到数据增强图像。

2.根据权利要求1所述的方法，其中，所述对待增强训练图像进行背景提取处理，得到对应的第一背景图像、第一前景图像进一步包括：

基于前景框位置标注信息从待增强训练图像中提取出第一前景图像；

基于前景框位置标注信息，利用第一阈值对待增强训练图像中的前景图像进行填充处理，得到第一背景图像。

3.根据权利要求1或2所述的方法，其中，所述对辅助图像进行背景提取处理，得到对应的第二背景图像进一步包括：

基于前景框位置标注信息，利用第一阈值对辅助图像中的前景图像进行填充处理，得到第二背景图像。

4.根据权利要求1或2所述的方法，其中，根据所述随机掩模，对所述第一背景图像及所述第二背景图像进行背景拼接处理，得到第三背景图像进一步包括：

根据随机掩模的像素值进行背景图像选择处理；

若随机掩模的像素值为第二阈值，则选择第一背景图像；若随机掩模的像素值为第三阈值，则选择第二背景图像，得到第三背景图像。

5.根据权利要求1或2所述的方法，其中，所述对第一前景图像及第三背景图像进行前景叠加处理，得到数据增强图像进一步包括：

将第一前景图像与第三背景图像进行逐像素对比，基于比对结果将第一前景图像粘贴至第三背景图像相应位置，得到数据增强图像。

6.根据权利要求1或2所述的方法，其中，掩模数据包含坐标值及颜色值；

基于所述N组掩模数据生成随机掩模进一步包括：

针对N组掩模数据，分别根据相应的掩模数据中的坐标值及颜色值在N个灰度值为第一阈值的灰度图上绘制对应颜色值的色块，得到N个掩模；

对N个掩模的颜色值求均值，并利用第一阈值对均值处理后的掩模进行二值化处理，得到随机掩模。

7.根据权利要求1或2所述的方法，其中，在得到数据增强图像之后，所述方法还包括：基于所述数据增强图像进行目标检测模型训练，以根据训练得到目标检测模型进行目标检测。

8.一种数据增强装置，包括：

随机掩模生成模块，适于随机生成N组掩模数据，基于所述N组掩模数据生成随机掩模；

背景提取模块，适于对待增强训练图像进行背景提取处理，得到对应的第一背景图像、第一前景图像；以及，对辅助图像进行背景提取处理，得到对应的第二背景图像，其中，所述辅助图像为从训练图像样本中随机选取的一训练图像；

背景拼接模块，适于根据所述随机掩模，对所述第一背景图像及所述第二背景图像进行背景拼接处理，得到第三背景图像；

前景叠加模块，适于对第一前景图像及第三背景图像进行前景叠加处理，得到数据增强图像。

9.一种计算设备，包括：处理器、存储器、通信接口和通信总线，所述处理器、所述存储器和所述通信接口通过所述通信总线完成相互间的通信；

所述存储器用于存放至少一可执行指令，所述可执行指令使所述处理器执行如权利要求1-7中任一项所述的数据增强方法对应的操作。

10.一种计算机存储介质，所述存储介质中存储有至少一可执行指令，所述可执行指令使处理器执行如权利要求1-7中任一项所述的数据增强方法对应的操作。