[发明专利]基于语义分割的表格重建方法、装置、设备及存储介质

说明书

本申请涉及人工智能领域，具体公开了一种基于语义分割的表格重建方法、装置、设备及存储介质。该方法包括：获取待重建图像，所述待重建图像中存在待重建表格；对所述待重建图像进行语义分割，得到多个掩码图，其中，所述多个掩码图与多个预设分割类别一一对应，所述多个掩码图中的每个掩码图用于表示所述待重建图像中的各个像素点属于与所述每个掩码图对应的预设分割类别的概率；根据所述多个掩码图构建待重建表格；从所述待重建图像中提取与所述待重建表格对应的文字内容；将所述文字内容填入所述待重建表格，得到重建好的表格。本申请实施例有利于提高表格重建的精度。

技术领域

本申请涉及人工智能技术领域，具体涉及一种基于语义分割的表格重建方法、装置、设备及存储介质。

背景技术

表格作为一种有效的数据组织与展现方法被广泛应用，也成为各类文档中常见的页面对象。表格通常以结构化的方式展示基本信息，因而表格检测和识别是诸多文件分析应用中的一项重要任务。而且，随着文档数目的爆炸性增长，如何高效地从文档中找到表格并获取内容与结构信息即表格识别，成为了一个亟待解决的问题。

目前，对于图像中的表格的检测与重建大多通过边缘检测的方式来实现，即通过边缘检测技术识别出表格的边界，基于表格的边界重建表格。但是，实际应用中有些图像可能会存在变形，甚至有些表格没有明显的边界，这都给边缘检测带来巨大的挑战，导致重建出的表格精度低。

发明内容

本申请实施例提供了一种基于语义分割的表格重建方法、装置、设备及存储介质。通过语义分割的方式，提高表格的重建精度。

第一方面，本申请实施例提供一种基于语义分割的表格重建方法，包括：

获取待重建图像，所述待重建图像中存在待重建表格；

对所述待重建图像进行语义分割，得到多个掩码图，其中，所述多个掩码图与多个预设分割类别一一对应，所述多个掩码图中的每个掩码图用于表示所述待重建图像中的各个像素点属于与所述每个掩码图对应的预设分割类别的概率；

根据所述多个掩码图构建待重建表格；

从所述待重建图像中提取与所述待重建表格对应的文字内容；

将所述文字内容填入所述待重建表格，得到重建好的表格。

第二方面，本申请实施例提供一种表格重建装置，包括：

获取单元，用于获取待重建图像，所述待重建图像中存在待重建表格；

处理单元，用于对所述待重建图像进行语义分割，得到多个掩码图，其中，所述多个掩码图与多个预设分割类别一一对应，所述多个掩码图中的每个掩码图用于表示所述待重建图像中的各个像素点属于与所述每个掩码图对应的预设分割类别的概率；

根据所述多个掩码图构建待重建表格；

从所述待重建图像中提取与所述待重建表格对应的文字内容；

将所述文字内容填入所述待重建表格，得到重建好的表格。

第三方面，本申请实施例提供一种电子设备，包括：处理器，所述处理器与存储器相连，所述存储器用于存储计算机程序，所述处理器用于执行所述存储器中存储的计算机程序，以使得所述电子设备执行如第一方面所述的方法。

第四方面，本申请实施例提供一种计算机可读存储介质，所述计算机可读存储介质存储有计算机程序，所述计算机程序使得计算机执行如第一方面所述的方法。

第五方面，本申请实施例提供一种计算机程序产品，所述计算机程序产品包括存储了计算机程序的非瞬时性计算机可读存储介质，所述计算机可操作来使计算机执行如第一方面所述的方法。

实施本申请实施例，具有如下有益效果：