[发明专利]一种标签定位模型训练方法、装置、电子设备及存储介质

权利要求书

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

确定训练数据集，所述训练数据集包括标注数据集和公共数据集；

基于所述训练数据集训练标签定位模型，使得所述标签定位模型能够确定所述训练数据集中的标签信息；所述标签信息包括各个标签的位置信息和相应位置的类别信息。

2.根据权利要求1所述的方法，其特征在于，所述确定训练数据集包括：

对数据集标签图像进行标注，得到标注数据集；

将任意文本形状的数据集确定为公共数据集；

将所述标注数据集和所述公共数据集作为所述训练数据集。

3.根据权利要求2所述的方法，其特征在于，所述对数据集标签图像进行标注，得到标注数据集，包括：

将所述数据集标签图像中的各个标签按不同的形式划分为不同的标签类别，所述标签类别包括：文本标签和图像标签；

确定标注所述数据集标签图像过程中生成的文本文件的格式；

基于所述文本文件的格式，以所述标签类别的英文名称替换所述数据集标签图像中各个标签的文本内容，得到所述标注数据集。

4.根据权利要求3所述的方法，其特征在于，所述文本文件的格式包括所述数据集标签图像中各个标签边框的四个顶点的横纵坐标值，以及各个所述标签边框所属标签的类别。

5.根据权利要求2-4任一项所述的方法，其特征在于，所述数据集标签图像包括：第一待测图像以及与所述第一待测图像相似度满足相似阈值的第二待测图像；

所述第一待测图像与所述第二待测图像中的各个标签一一对应，所述第二待测图像包括：与所述第一待测图像中对应的标签边框全图占比不同的图像、与所述第一待测图像中对应的标签边框位置存在差异的图像，以及与所述第一待测图像中对应的文字位置存在差异的图像。

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

获取验证数据集，所述验证数据集包括多个携带标注的标签图像；

将所述验证数据集输入至所述标签定位模型，得到所述标签定位模型输出的标签信息，所述标签信息包括多个所述携带标注的标签图像中各个标签的位置信息和相应位置的标签的类别信息；

基于所述标签信息验证所述验证数据集的准确性，并调整所述标签定位模型的参数。

7.一种标签定位方法，其特征在于，所述方法包括：

获取待处理数据，所述待处理数据包括多个待测标签图像；

利用标签定位模型确定所述待处理数据中各个标签的位置信息和相应位置的标签类别信息。

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

数据集确定模块，用于确定训练数据集，所述训练数据集包括标注数据集和公共数据集；

模型训练模块，用于基于所述训练数据集训练标签定位模型，使得所述标签定位模型能够确定所述训练数据集中的标签信息；所述标签信息包括各个标签的位置信息和相应位置的类别信息。

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

数据获取模块，用于获取待处理数据，所述待处理数据包括多个待测标签图像；

确定模块，用于利用标签定位模型确定所述待处理数据中各个标签的位置信息和相应位置的标签类别信息。

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

至少一个处理器；以及

与所述至少一个处理器通信连接的存储器；其中，

所述存储器存储有可被所述至少一个处理器执行的指令，所述指令被所述至少一个处理器执行，以使所述至少一个处理器能够执行权利要求1-6中任一项所述的标签定位模型训练方法；

或者，所述至少一个处理器能够执行权利要求7所述的标签定位方法。

11.一种存储有计算机指令的非瞬时计算机可读存储介质，其特征在于，所述计算机指令用于使所述计算机执行根据权利要求1-6中任一项所述的标签定位模型训练方法；或者，执行权利要求7所述的标签定位方法。