[发明专利]文本标签的确定方法及相关装置

说明书

本申请实施例公开了一种文本标签的确定方法及相关装置，该方法包括：获取目标图片中的目标文本与目标文本对应的目标文本框特征向量，将目标文本与目标文本框特征向量输入预先训练好的标签确定模型，确定目标文本对应的第一标签，标签确定模型由多个训练文本、训练文本对应的训练文本框特征向量拼接后的数据，以及训练文本对应的第二标签训练得到，第二标签为预先设定的标签。本申请将训练文本框特征向量以及训练文本拼接，通过拼接后的训练文本框特征向量以及训练文本来训练标签确定模型，使得训练好的标签确定模型输出结果为训练文本的对应的第二标签，提高确定文本标签的准确性。

技术领域

本申请涉及数据处理技术领域，具体涉及一种文本标签的确定方法及相关装置。

背景技术

随着科技的发展，包括文本图片在内的多媒体资源规模越来越大。文本检索逐渐成为自然语言处理领域中的一个研究热点，产生了许多基于光学字符识别(OCR)技术的文本检索方法，该方法从图片中识别文本内容，然后利用文本检索技术实现文本图片检索系统。通过现有的图片提取技术，从图片中提取完文本后，再提取文本的信息，从而根据识别与提取的结果为该文本打上标签，图片文本的提取与识别文本是两个独立的过程，导致为图片中文本打上对应标签的效率低，且准确度低。

发明内容

本申请实施例提供了一种文本标签的确定方法及相关装置，能够将目标图片中的目标文本与目标文本框特征向量拼接后，来确定目标图片中目标文本的标签，从而提高确定文本标签时的准确性。

第一方面，本申请实施例提供一种文本标签的确定方法，所述方法包括：

获取目标图片；

提取所述目标图片中的目标文本与所述目标文本对应的目标文本框特征向量；

将所述目标文本与所述目标文本框特征向量输入预先训练好的标签确定模型，确定所述目标文本对应的第一标签，所述标签确定模型由多个训练文本、所述训练文本对应的训练文本框特征向量拼接后的数据，以及所述训练文本对应的第二标签训练得到，所述训练文本框特征向量包括所述训练文本所在的区域在训练图片中的顶点坐标，以及所述区域的斜边长与所述训练图片的斜边长的比值，所述第二标签为预先设定的标签。

第二方面，本申请实施例提供的一种文本标签的确定装置，所述装置包括：

第一获取单元，用于获取目标图片；

提取单元，用于提取所述目标图片中的目标文本与所述目标文本对应的目标文本框特征向量；

第一输入单元，用于将所述目标文本与所述目标文本框特征向量输入预先训练好的标签确定模型，确定所述目标文本对应的第一标签，所述标签确定模型由多个训练文本、所述训练文本对应的训练文本框特征向量拼接后的数据，以及所述训练文本对应的第二标签训练得到，所述训练文本框特征向量包括所述训练文本所在的区域在训练图片中的顶点坐标，以及所述区域的斜边长与所述训练图片的斜边长的比值，所述第二标签为预先设定的标签。

第三方面，本申请实施例提供一种终端设备，所述终端设备包括处理器、存储器、通信接口，以及一个或多个程序，所述一个或多个程序被存储在所述存储器中，并且被配置由所述处理器执行，所述程序包括用于执行上述第一方面所述的方法中所描述的部分或全部步骤的指令。

第四方面，本申请实施例提供了一种计算机可读存储介质，存储用于电子数据交换的计算机程序，其中，所述计算机程序使得计算机执行如本实施例第一方面中的部分或全部步骤。