[发明专利]信息提取方法、装置、设备、介质及产品

权利要求书

1.一种信息提取方法，其特征在于，所述方法包括：

对图像进行文本检测，获得所述图像中的文本区域，所述文本区域包括多个文本行；

根据所述文本区域构建图网络模型，所述文本区域中的每个文本行为所述图网络模型的一个节点；

通过节点分类模型对所述图网络模型中的节点进行分类，以及通过边分类模型对所述图网络模型中的节点之间的边进行分类；

根据对所述节点的分类结果以及对所述边的分类结果，获得所述图像中的至少一个键值对。

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

提取所述节点的特征，以及提取所述边的特征；

根据所述边的特征聚合所述节点的邻居节点的特征，获得所述节点的嵌入表示；

所述通过节点分类模型对所述图网络模型中的节点进行分类，以及通过边分类模型对所述图网络模型中的节点之间的边进行分类，包括：

根据所述节点的嵌入表示，通过节点分类模型对所述图网络模型中的节点进行分类；

将所述边对应的两个节点的嵌入表示拼接，根据拼接后的所述嵌入表示，通过边分类模型对所述图网络模型中的节点之间的边进行分类。

3.根据权利要求1所述的方法，其特征在于，所述通过边分类模型对所述图网络模型中的节点之间的边进行分类，包括：

根据对所述节点的分类结果，通过边分类模型对所述图网络模型中的节点之间的边进行分类。

4.根据权利要求1至3任一项所述的方法，其特征在于，所述对所述节点的分类结果包括如下标签中的一种：键、值和其他，所述对所述边的分类结果包括如下标签中的一种：键值边、值值边、键键边或其他中的一种。

5.根据权利要求4所述的方法，其特征在于，对所述节点的分类结果为键时，则对所述边的分类结果包括键值边或键键边；对所述节点的分类结果为值时，则对所述边的分类结果包括键值边或值值边。

6.根据权利要求1至5任一项所述的方法，其特征在于，所述节点的特征包括所述节点的图像特征、文本特征和位置特征中的至少一种，所述边的特征所述文本行之间的相对位置、相对宽高中的至少一种。

7.根据权利要求1至5任一项所述的方法，其特征在于，所述节点分类模型和所述边分类模型为端到端模型。

8.一种信息提取装置，其特征在于，所述装置包括：

检测模块，用于对图像进行文本检测，获得所述图像中的文本区域，所述文本区域包括多个文本行；

构建模块，用于根据所述文本区域构建图网络模型，所述文本区域中的每个文本行为所述图网络模型的一个节点；

分类模块，用于通过节点分类模型对所述图网络模型中的节点进行分类，以及通过边分类模型对所述图网络模型中的节点之间的边进行分类；

获取模块，用于根据对所述节点的分类结果以及对所述边的分类结果，获得所述图像中的至少一个键值对。

9.根据权利要求8所述的装置，其特征在于，所述装置还包括提取模块，所述提取模块用于：

提取所述节点的特征，以及提取所述边的特征；

根据所述边的特征聚合所述节点的邻居节点的特征，获得所述节点的嵌入表示；

所述分类模型具体用于：

根据所述节点的嵌入表示，通过节点分类模型对所述图网络模型中的节点进行分类；

将所述边对应的两个节点的嵌入表示拼接，根据拼接后的所述嵌入表示，通过边分类模型对所述图网络模型中的节点之间的边进行分类。

10.根据权利要求8所述的装置，其特征在于，所述分类模块具体用于：

根据对所述节点的分类结果，通过边分类模型对所述图网络模型中的节点之间的边进行分类。