[发明专利]一种陈列图像样本的生成方法及装置

权利要求书

1.一种陈列图像样本的生成方法，其特征在于，包括：

获取不同视觉角度下的待融合SKU的SKU图像，并根据视觉角度对每一所述SKU图像进行分类；

获取陈列场景的场景图像，并根据视觉角度对所述场景图像中的每一待替换SKU进行分类；其中，所述场景图像中的待替换SKU与所述SKU图像中的SKU为不同品类的SKU；

以所述SKU图像为前景图像，以所述场景图像为背景图像，根据所述前景图像所属的视觉角度类别，将所述前景图像中的待融合SKU与所述背景图像中，视觉角度类别相同的待替换SKU进行替换，生成所述陈列图像样本。

2.根据权利要求1所述陈列图像样本的生成方法，其特征在于，所述以所述SKU图像为前景图像，以所述场景图像为背景图像，根据所述前景图像所属的视觉角度类别，将所述前景图像中的待融合SKU与所述背景图像中，视觉角度类别相同的待替换SKU进行替换，生成所述陈列图像样本，具体包括：

步骤A:以一所述SKU图像，作为所述前景图像，并提取所述前景图像中待融合SKU的mask；

步骤B:以一所述场景图像，作为所述背景图像，并提取所述背景图像中所有待替换SKU对应的标注框的位置信息和视觉角度类别信息；

步骤C:选取一标注框，并判断所选取的标注框的视觉角度类别信息是否与所述前景图像的视觉角度类别一致；若一致，则将所选取的标注框对应的待替换SKU与所述待融合SKU的mask进行替换，若不一致，则重新选取一标注框，重新判断所选取的标注框的视觉角度类别信息是否与所述前景图像的视觉角度类别一致；

步骤D:重复步骤A和步骤C直至所替换的前景图像的数量达到预设值，生成所述陈列图像样本。

3.根据权利要求1所述陈列图像样本的生成方法，其特征在于，在获取不同视觉角度下的待融合SKU的SKU图像之后，还包括：将所有所述SKU图像输入预设的对抗网络模型中，以使所述对抗网络模型生成新的SKU图像。

4.根据权利要求1所述陈列图像样本的生成方法，其特征在于，在生成所述陈列图像样本后，还包括，通过预设的StyleGAN模型对所述陈列图像样本进行处理。

5.根据权利要求2所述陈列图像样本的生成方法，其特征在于，还包括：修改所述陈列图像样本中待融合SKU的mask所对应的标注框的位置信息，继而计算所述陈列图像样本中每两个标注框的iou值，并根据所述iou值进行去重。

6.根据权利要求1所述陈列图像样本的生成方法，其特征在于，所述陈列场景包括：货架场景、冰柜场景、端架场景或地推场景。

7.一种陈列图像样本的生成装置，其特征在于，包括SKU图像获取及处理模块，场景图像获取及处理模块和图像融合模块；

所述SKU图像获取及处理模块，用于获取不同视觉角度下的待融合SKU的SKU图像，并根据视觉角度对每一所述SKU图像进行分类；

所述场景图像获取及处理模块，用于获取陈列场景的场景图像，并根据视觉角度对所述场景图像中的每一待替换SKU进行分类；其中，所述场景图像中的待替换SKU与所述SKU图像中的SKU为不同品类的SKU；

所述图像融合模块，用于以所述SKU图像为前景图像，以所述场景图像为背景图像，根据所述前景图像所属的视觉角度类别，将所述前景图像中的待融合SKU与所述背景图像中，视觉角度类别相同的待替换SKU进行替换，生成所述陈列图像样本。