[发明专利]一种样本生成方法、装置、电子设备及计算机可读存储介质

权利要求书

1.一种样本生成方法，其特征在于，包括：

获取包含猪体的初始样本图像及多个背景图像；

在所述初始样本图像中提取猪体掩膜图像；

对所述猪体掩膜图像进行前景增强，得到多个增强猪体掩膜图像；

将每个所述增强猪体掩膜图像分别与多个所述背景图像进行融合处理，得到多个训练样本图像及标注信息。

2.根据权利要求1所述的样本生成方法，其特征在于，所述将每个所述增强猪体掩膜图像分别与多个所述背景图像进行融合处理，得到多个训练样本图像及标注信息，包括：

在将每个所述增强猪体掩膜图像与任一背景图像融合时，随机生成所述增强猪体掩膜图像在所述背景图像中的添加位置；

将所述增强猪体掩膜图像按照所述添加位置覆盖在所述背景图像上；

将所述增强猪体掩膜图像与所述背景图像进行图像融合，得到训练样本图像；

根据所述添加位置生成所述标注信息。

3.根据权利要求2所述的样本生成方法，其特征在于，将所述增强猪体掩膜图像按照所述添加位置覆盖在所述背景图像上，包括：

获取所述增强猪体掩膜图像的图像尺寸；

根据所述添加位置及所述增强猪体掩膜图像的图像尺寸计算添加区域；

将所述增强猪体掩膜图像添加至所述添加区域中。

4.根据权利要求3所述的样本生成方法，其特征在于，将所述增强猪体掩膜图像与所述背景图像进行图像融合，得到训练样本图像，包括：

根据所述增强猪体掩膜图像每个像素点的像素值计算所述增强猪体掩膜图像的第一梯度场；

根据所述背景图像中除所述添加区域外的每个像素点的计算所述背景图像中除所述添加区域外的区域图像的第二梯度场；

根据所述第一梯度场和所述第二梯度场计算待重建图像的梯度场；

根据所述待重建图像的梯度场计算散度；

根据所述散度、预设的系数矩阵和泊松方程计算训练样本图像的每个像素点的RGB值。

5.根据权利要求2所述的样本生成方法，其特征在于，根据所述添加位置生成所述标注信息，包括：

获取所述增强猪体掩膜图像的图像尺寸；

根据所述添加位置及所述图像尺寸确定标注范围；

根据所述标注范围生成所述标注信息。

6.根据权利要求1所述的样本生成方法，其特征在于，前景增强的方式包括：平移、旋转、缩放、水平翻转、垂直翻转、模糊处理和添加椒盐噪声中的一种或多种。

7.一种样本生成装置，其特征在于，包括：

获取模块，用于获取包含猪体的初始样本图像及多个背景图像；

提取模块，用于在所述初始样本图像中提取猪体掩膜图像；

前景增强模块，用于对所述猪体掩膜图像进行前景增强，得到多个增强猪体掩膜图像；

融合处理模块，用于将每个所述增强猪体掩膜图像分别与多个所述背景图像进行融合处理，得到多个训练样本图像及标注信息。

8.根据权利要求7所述的样本生成装置，其特征在于，所述融合处理模块，包括：

位置生成单元，用于在将每个所述增强猪体掩膜图像与任一背景图像融合时，随机生成所述增强猪体掩膜图像在所述背景图像中的添加位置；

覆盖单元，用于将所述增强猪体掩膜图像按照所述添加位置覆盖在所述背景图像上；

融合单元，用于将所述增强猪体掩膜图像与所述背景图像进行图像融合，得到训练样本图像；

生成单元，用于根据所述添加位置生成所述标注信息。

9.根据权利要求8所述的样本生成装置，其特征在于，所述覆盖单元，包括：

第一获取子单元，用于获取所述增强猪体掩膜图像的图像尺寸；

第一计算子单元，用于根据所述添加位置及所述增强猪体掩膜图像的图像尺寸计算添加区域；

添加子单元，用于将所述增强猪体掩膜图像添加至所述添加区域中。