[发明专利]基于YoloV3模型的运单图片信息打码方法及系统

权利要求书

1.一种基于Yo1oV3模型的运单图片信息打码方法，其特征在于，包括：

S1、将运单图片输入至训练好的Yo1oV3模型进行检测，得到所述运单图片中的寄方框、寄方栏框、收方框和收方栏框的信息；所述运单图片中的寄方框、寄方栏框、收方框和收方栏框的信息包括各个框的高度、宽度以及中心点坐标、以及各个框的顶点的坐标；所述寄方框与寄方栏框的高度相同，收方框和收方栏框的高度相同；将属于同一联运单的寄方框、寄方栏框、收方框和收方栏框合成在同一框群中；

S2、选择寄方栏框或收方栏框，比较其高度与宽度两者间的大小，并结合寄方框和收方框的相对位置关系，识别出所述运单图片的当前姿态；选择的寄方栏框或收方栏框为任一框群中的寄方栏框或收方栏框；

S3、结合运单图片的当前姿态、寄方框、寄方栏框、收方框和收方栏框的信息，在寄方框和寄方栏框中设定打码区域，并对所述打码区域进行打码操作。

2.根据权利要求1所述的基于Yo1oV3模型的运单图片信息打码方法，其特征在于，所述步骤S2之后还包括：

根据当前姿态，将所述运单图片进行旋转摆正；摆正后的运单图片中，寄方框和寄方栏框并排，收方框在寄方框正下方，收方栏框在寄方栏框正下方。

3.根据权利要求2所述的基于Yo1oV3模型的运单图片信息打码方法，其特征在于，所述步骤S3包括：

S31、计算连接寄方框中心点到寄方栏框中心点的第一直线的斜率；

S32、以摆正后运单图片寄方框的左上角顶点为打码点的起点，沿第一直线的斜率走向，按照预设打码高度和步长，逐点打码至寄方栏框上边框的中心点处。

4.根据权利要求3所述的基于Yo1oV3模型的运单图片信息打码方法，其特征在于，所述步骤S32中沿第一直线的斜率走向逐点打码前包括：

判断当前打码点按照寄方框高度作为打码高度进行打码，打码后的区域下边界是否会超过收方框的上边框，若否，则预设打码高度为寄方框的高度；若是，则预设打码高度为寄方框上边框与收方框上边框两者间的距离。

5.根据权利要求1所述的基于Yo1oV3模型的运单图片信息打码方法，其特征在于，所述训练好的Yo1oV3模型通过如下步骤得到：

获取样本图片，将所述样本图片进行数据清洗以及数据增强，得到目标图片；

对所述目标图片中的寄方框、寄方栏框、收方框和收方栏框进行打标，生成标注信息；所述标注信息包括目标图片中各个框的中心点的X轴坐标、Y轴坐标，以及各个框的宽度和高度；

将所述目标图片和标注信息输入至Yo1oV3模型进行训练。

6.一种基于Yo1oV3模型的运单图片信息打码系统，其特征在于，包括：

检测模块，用于将运单图片输入至训练好的Yo1oV3模型进行检测，得到所述运单图片中的寄方框、寄方栏框、收方框和收方栏框的信息；所述运单图片中的寄方框、寄方栏框、收方框和收方栏框的信息包括各个框的高度、宽度以及中心点坐标、以及各个框的顶点的坐标；所述寄方框与寄方栏框的高度相同，收方框和收方栏框的高度相同；将属于同一联运单的寄方框、寄方栏框、收方框和收方栏框合成在同一框群中；

识别模块，用于选择寄方栏框或收方栏框，比较其高度与宽度两者间的大小，并结合寄方框和收方框的相对位置关系，识别出所述运单图片的当前姿态；选择的寄方栏框或收方栏框为任一框群中的寄方栏框或收方栏框；

打码模块，用于结合运单图片的当前姿态、寄方框、寄方栏框、收方框和收方栏框的信息，在寄方框和寄方栏框中设定打码区域，并对所述打码区域进行打码操作。

7.根据权利要求6所述的基于Yo1oV3模型的运单图片信息打码系统，其特征在于，其还包括：

摆正模块，用于根据当前姿态，将所述运单图片进行旋转摆正；摆正后的运单图片中，寄方框和寄方栏框并排，收方框在寄方框正下方，收方栏框在寄方栏框正下方。