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

说明书

本发明公开了一种陈列图像样本的生成方法及装置，所述方法包括：获取不同视觉角度下的待融合SKU的SKU图像，并根据视觉角度对每一所述SKU图像进行分类；获取陈列场景的场景图像，并根据视觉角度对所述场景图像中的每一待替换SKU进行分类；其中，所述场景图像中的待替换SKU与所述SKU图像中的SKU为不同品类的SKU；以所述SKU图像为前景图像，以所述场景图像为背景图像，根据所述前景图像所属的视觉角度类别，将所述前景图像中的待融合SKU与所述背景图像中，视觉角度类别相同的待替换SKU进行替换，生成所述陈列图像样本，通过实施本发明的实施例能降低新SKU陈列图像样本采集的人力物力成本并提高陈列图像样本的获取效率。

技术领域

本发明涉及图像处理技术领域，尤其是涉及一种陈列图像样本的生成方法及装置。

背景技术

快消品零售是一个很特殊的行业，商品销售节奏很快，它更多满足的是消费者的冲动性和即时性消费。充满视觉冲击力的货架陈列对推动销售业绩发挥着重要作用。很多大型快消品牌商每年花在陈列广告上的支出多达十几亿，光花在陈列检查上的费用就有近亿元。过去陈列检查完全依赖人眼和手工完成的，进店和记录需要花费大量的人力，效率很低，质量难以把控。随着技术的发展越来越多的陈列检测采用基于深度学些的商品识别的技术，对陈列进行智能检测。

而基于深度学习的商品识别在模型训练时需要依赖于大量陈列图片数据，作为训练样本，而这些作为训练样本的陈列图片主要由人工进行实地采集，而由于快消行业的快速变换的节奏，产品的更新比较块，若每变更一种新产品就要实地去采集新产品的陈列图像作为训练样本，需要花费大量人力物力成本，并且对于那些时效性特别高的，间歇性特别强的陈列场景，如节假日特别促销陈列场景等，由于陈列时间短，采用人工实地采集的方式可能不能在促销产品的陈列期间，采集足够多的样本。

发明内容

本发明实施例所要解决的技术问题在于，提供了一种陈列图像样本的生成方法及装置，能够在原有陈列场景的场景图像的基础上，生成与原场景图像不同的SKU的陈列图像，降低新SKU陈列图像样本采集的人力物力成本并提高陈列图像样本的获取效率。

为了解决上述技术问题，本发明实施例提供了一种陈列图像样本的生成方法，包括：获取不同视觉角度下的待融合SKU的SKU图像，并根据视觉角度对每一所述SKU图像进行分类；

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

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

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

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

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

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

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