[发明专利]图像拉正方法、系统、设备和存储介质

说明书

本公开涉及图像处理技术领域，特别涉及一种图像拉正方法、系统、设备和存储介质。在本公开提供的图像拉正方法、系统、设备和存储介质中，通过检测货架图片中的商品和置物层，进而调整置物层在不同方向上的长度得到二次调整图片；最后对二次调整图片进行整体的比例调整，以使二次调整图片中商品的标记框的长宽比与货架图片中对应商品的标记框的长宽比一致，从而得到拉正图片的方式，提高了图片拉正的精度和效率。

技术领域

本公开涉及图像处理技术领域，特别涉及一种图像拉正方法、系统、设备和存储介质。

背景技术

随着零售领域智能化和信息化进程的加快，越来越多的图像处理技术被应用到零售领域中，也因此对零售领域中的相关图像的处理技术提出了更高的要求。

以商超中的零售商品货架的图像来说，由于所有货架的图像均为采用相机拍摄得到，因此所有货架的图像不可避免地都会出现变形、图像中的商品比例不一致等各种问题。所以，对货架的图像调整，以使货架的图像变形程度降低，比例不一致现象减少是零售领域智能化进程中的重要环节。

发明内容

本公开的一方面提供了一种图像拉正方法，用于拉正在零售货架正前方的竖直线上拍摄得到的货架图片。所述图像拉正方法包括如下步骤：

检测所述货架图片中的每个置物层和每个置物层中的商品；

在第一方向上比例拉伸所述货架图片中的置物层，以使每个置物层在第一方向上的长度一致，得到初次调整图片；

在第二方向上比例拉伸所述初次调整图片中的置物层，以使每个置物层在第二方向上的长度一致，得到二次调整图片；

调整所述二次调整图片的长宽比，以使所述二次调整图片内商品的标记框长宽比与所述货架图片内对应商品的标记框长宽比一致，得到拉正图片。

在一实施例中，所述检测货架图片中的每个置物层和每个置物层中的商品的步骤包括：

训练一模型，以得到能够检测出货架图像中的商品，并能够对每个商品进行标记框标记的商品检测模型；

训练另一模型，以得到能够检测出货架图像中的置物层，并能够对每个置物层进行边界框标记的置物层检测模型；

将所述货架图片输入所述商品检测模型中，输出得到商品区域带有标记框的所述货架图片；

将所述带有标记框的货架图片输入所述置物层检测模型中，输出得到带有标记框和边界框的所述货架图片。

在一实施例中，所述的在第一方向上比例拉伸所述货架图片中的置物层，以使每个置物层在第一方向上的长度一致，得到初次调整图片的步骤包括：

在每个置物层对应的所述边界框中建立一趋势线段，其中所述趋势线段过所述边界框的中心，两端与所述边界框相交，并且延伸方向与所述边界框的延伸方向平行；

选择其中一个所述边界框中的趋势线段为基准趋势线段；

在第一方向上比例拉伸其余所述边界框对应的所述置物层，以使所述其余边界框上的趋势线段的两端与所述基准趋势线段的两端对齐，得到所述初次调整图片。

在一实施例中，所述其中一个所述边界框是所有所述边界框中在第一方向上长度最长的一个。

在一实施例中，所述在第二方向比例拉伸初次调整图片中的置物层，以使每个置物层在第二方向上的长度一致，得到二次调整图片的步骤包括：选择其中一个所述置物层为基准置物层；

在第二方向上比例拉伸其余所述置物层，以使每个所述置物层在第二方向上的长度一致，得到二次调整图片

在一实施例中，所述调整二次调整图片的长宽比，以使二次调整图片内商品的标记框长宽比与货架图片内对应商品的标记框长宽比一致，得到拉正图片的步骤包括：