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

权利要求书

1.一种图像拉正方法，其特征在于，用于拉正在零售货架正前方的竖直线上拍摄得到的货架图片，所述方法包括如下步骤：

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

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

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

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

其中置物层的延伸方向定义为第一方向，将垂直于第一方向的方向定义为第二方向。

2.如权利要求1所述的图像拉正方法，其特征在于，所述检测货架图片中的每个置物层和每个置物层中的商品的步骤包括：

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

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

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

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

3.如权利要求2所述的图像拉正方法，其特征在于，所述在第一方向上比例拉伸所述货架图片中的置物层，以使每个置物层在第一方向上的长度一致，得到初次调整图片的步骤包括：

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

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

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

4.如权利要求3所述的图像拉正方法，其特征在于，所述其中一个所述边界框是所有所述边界框中在第一方向上长度最长的一个。

5.如权利要求3所述的图像拉正方法，其特征在于，所述在第二方向比例拉伸初次调整图片中的置物层，以使每个置物层在第二方向上的长度一致，得到二次调整图片的步骤包括：选择其中一个所述置物层为基准置物层；

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

6.如权利要求1或3所述的图像拉正方法，其特征在于，所述调整二次调整图片的长宽比，以使二次调整图片内商品的标记框长宽比与货架图片内对应商品的标记框长宽比一致，得到拉正图片的步骤包括：

分别获取所述货架图片内中位数商品和所述二次调整图片内中位数商品的标记框长宽比；

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

7.如权利要求1或3所述的图像拉正方法，其特征在于，所述调整二次调整图片的长宽比，以使二次调整图片内商品的标记框长宽比与货架图片内对应商品的标记框长宽比一致，得到拉正图片的步骤包括：

获取所述货架图片内所有商品标记框长宽比的均值；

获取所述二次调整图片内所有商品标记框长宽比的均值；

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