[发明专利]零售商品货架图像的生成方法、系统、设备和存储介质

说明书

本公开涉及图像处理技术领域，特别涉及一种用于模型训练的零售商品货架图像的生成方法。本公开提供的用于模型训练的零售商品货架图像的生成方法中，通过制作模拟实际场景中商品和货架视觉状态的三维合成图像，从而替代了传统的人工拍摄图片来获取用于训练的图像数据的方式，大大提高了模型训练图像数据的制作时间。并且由于在制作合成图像时，可以在三维图像制作软件的环境下，单独制作商品模型和货架模型，并且可以利用现有三维图像制作软件中的脚本编写功能，实现多种摆放样式的商品模型摆放至货架模型中，进而实现商品种类和摆放方式的多样化，使得最终得到的零售商品货架图像效果可以更加接近采用拍照设备实际拍摄的图像效果。

技术领域

本公开涉及图像处理技术领域，特别涉及一种用于模型训练的零售商品货架图像的生成方法、系统、设备和存储介质。

背景技术

机器学习是一门多领域交叉学科。它是专门用来研究计算机怎样模拟或实现人类的学习行为，以获取新的知识或技能，重新组织已有的知识结构使之不断改善自身的性能。它是人工智能的核心，是使计算机具有智能的根本途径。

在机器学习中，常常会采用大量的图像数据进行图像检测、识别训练。而在获取这些图像数据时，通常是采用相机等拍摄设备进行拍摄而得到，并通过不断累加拍摄的图像，而使图像数据满足所需要的数量级。

发明内容

本公开的一方面提供了一种用于模型训练的零售商品货架图像的生成方法。所述零售商品货架图像的生成方法包括如下步骤：

制作模拟实际场景中商品和货架视觉状态的三维合成图像；

获取所述合成图像中的预定位置的商品模型的高度和宽度的数据信息；

利用获取的所述数据信息形成边界框，并围在所述合成图像中的对应商品模型上，以得到带有所述数据信息和所述边界框的零售商品货架图像。

在一实施例中，所述实际场景中商品和货架视觉状态至少包括光线角度、商品和货架上的反射图案

在一实施例中，所述货架是超市中用于摆放商品的摆放架或用于冷藏饮料的立式冰柜。

在一实施例中，所述制作模拟实际场景中商品和货架视觉状态的合成图像的步骤包括：

构建目标商品和货架的三维模型，分别得到商品模型和货架模型；

随机选取不同种类和数量的所述商品模型，并将选取的所述商品模型随机摆放至所述货架模型中；

利用预先制作的包装图案和背景图案对所述货架模型和所述货架模型中的所述商品模型进行渲染，以得到合成图像，其中所述合成图像符合现实场景中商品和货架的视觉状态。

在一实施例中，所述获取合成图像中的预定位置的商品模型的高度和宽度的数据信息的步骤包括：

对所述合成图像中商品模型占用的区域、货架模型与背景图案占用的区域利用不同颜色进行替代，以得到商品模型与货架模型和背景图案之间、商品模型与商品模型之间能够通过颜色进行区分的分割图像；

获取所述分割图像中预定位置的所述商品模型的高度和宽度的数据信息。

在一实施例中，所述获取分割图像中预定位置的商品模型的高度和宽度的数据信息的步骤包括：

在所述分割图像上建立直角坐标系；

获取预定位置的所述商品模型按照其颜色所占用的Y轴方向最大值和最小值、X轴方向最大值和最小值，以得到预定位置的所述商品模型的高度和宽度数据信息；其中，

Y轴方向最大值和最小值之差为对应所述商品模型的高度数据信息，X轴方向最大值和最小值为对应所述商品模型的宽度数据信息。