[发明专利]定位模型的训练和定位方法、装置、设备以及存储介质

权利要求书

1.一种定位模型的训练方法，其特征在于，包括：

将预先获取的图形样本输入初始定位模型；所述图形样本携带有预设的至少一个定位目标的坐标标签，每个坐标标签标记了每个定位目标的多个顶点的坐标；

预测得到所述至少一个定位目标的定位信息，所述定位信息包括多个顶点坐标；

根据所述至少一个定位目标的定位信息和坐标标签，确定损失函数的最小值；

根据所述损失函数对所述初始定位模型进行训练，直至损失函数的最小值小于预设值，得到所述定位模型。

2.根据权利要求1所述的方法，其特征在于，所述根据所述至少一个定位目标的定位信息和坐标标签，确定损失函数的最小值，包括：

将所述定位信息的多个顶点坐标与所述坐标标签的多个顶点的坐标逐个组合，得到所述损失函数的多组参数；

针对每组参数，计算得到多个损失函数值，并获取所述多个损失函数值中的最小值。

3.根据权利要求2所述的方法，其特征在于，若所述至少一个定位目标为四边形，则所述损失函数为：

其中，为预测得到的所述定位信息的四个顶点坐标，为所述坐标标签的四个顶点的坐标，Pos为所述图形样本中检测得到的至少一个定位目标的集合，为匹配指示矩阵中的系数，smooth_L1为平滑L1损失函数。

4.根据权利要求3所述的方法，其特征在于，所述方法还包括：

根据所述定位信息，将所述坐标标签中每个顶点的坐标转换为相对坐标；所述定位信息包括包围盒的中心点坐标，所述包围盒为根据所述图形样本中的图形生成的矩形框。

5.一种四边形顶点相对位置确定方法，其特征在于，所述方法包括：

将待定位的目标图形输入定位模型，获取所述目标图形中每个定位目标的四个顶点坐标；

根据所述四个顶点坐标和预设置的坐标系，确定所述定位目标的定位信息，所述定位信息包括每个顶点的坐标和顶点位置；所述四个顶点坐标均处于所述预设置的坐标系的x轴正半轴和y轴正半轴。

6.根据权利要求5所述的方法，其特征在于，所述根据所述四个顶点坐标和预设置的坐标系，确定所述定位目标的定位信息，包括：

根据所述预设置的坐标系，将所述四个顶点坐标中x坐标值最小的点作为基准点；

将所述基准点与所述定位目标中的其余顶点连线，计算每条线与x轴正半轴的夹角；

根据所述每条线与x轴正半轴的夹角的大小，确定所述定位目标中每个顶点的顶点位置。

7.根据权利要求6所述的方法，其特征在于，若所述基准点的顶点位置为左上顶点，所述根据所述每条直线与x轴正半轴的夹角的大小，确定所述定位目标中每个顶点的顶点位置，包括：

按照每条直线与x轴正半轴的夹角从小到大的顺序，依次确定与每个夹角对应的顶点的顶点位置为右上顶点、右下顶点和左下顶点。

8.一种定位模型的训练装置，其特征在于，所述装置包括：

输入模块，将预先获取的图形样本输入初始定位模型；所述图形样本携带有预设的至少一个定位目标的坐标标签，每个坐标标签标记了每个定位目标的多个顶点的坐标；

预测模块，用于预测得到所述至少一个定位目标的定位信息，所述定位信息包括多个顶点坐标；

处理模块，用于根据所述至少一个定位目标的定位信息和坐标标签，确定损失函数的最小值；

所述处理模块还用于根据所述损失函数对所述初始定位模型进行训练，直至损失函数的最小值小于预设值，得到所述定位模型。

9.一种定位装置，其特征在于，所述装置包括：

输入模块，用于将待定位的目标图形输入定位模型，获取所述目标图形中每个定位目标的四个顶点坐标；

处理模块，用于根据所述四个顶点坐标和预设置的坐标系，确定所述定位目标的定位信息，所述定位信息包括每个顶点的坐标和顶点位置；所述四个顶点坐标均处于所述预设置的坐标系的x轴正半轴和y轴正半轴。

10.一种电子设备，其特征在于，包括：

存储器，用于存储程序；

处理器，用于执行所述存储器存储的所述程序，当所述程序被执行时，所述处理器用于执行如权利要求1至7中任一所述的方法。