[发明专利]结构图的结构重构方法、装置、电子设备及存储介质

说明书

本发明公开了一种结构图的结构重构方法，对结构图片进行节点检测，获取所述结构图片的节点信息；对所述结构图片进行导引线检测，获取所述结构图片的导引线信息；根据所述节点信息和所述导引线信息进行结构还原，获取所述结构图片的还原结构图。本发明公开的结构图的结构重构方法、装置、电子设备及存储介质，能够还原结构图片中的还原结构图，实现了对结构图片中的还原结构图进行编辑的效果。

技术领域

本发明涉及图像处理技术领域，特别涉及一种结构图的结构重构方法、装置、电子设备及存储介质。

背景技术

现有技术中，结构图中例如思维导图和流程图会经常出现在用户日常工作及学习生活中，其通常以图像的形式存在，便于文件传输和阅读。然而，结构图以图像格式作为文件存储与传输方式时，存在一个很大的缺陷——只能支持查阅、无法支持编辑，如此，为了方便对结构图进行编辑，亟需一种能够还原结构图片中的还原结构图的方法。

发明内容

本发明实施例提供一种结构图的结构重构方法、装置、电子设备及存储介质，能够还原结构图片中的还原结构图，实现了对结构图片中的还原结构图进行编辑的效果。

本发明实施例第一方面提供一种结构图的结构重构方法，所述方法包括：

对结构图片进行节点检测，获取所述结构图片的节点信息；

对所述结构图片进行导引线检测，获取所述结构图片的导引线信息；

根据所述节点信息和所述导引线信息进行结构还原，获取所述结构图片的还原结构图。

可选的，所述对结构图片进行节点检测，获取所述结构图片的节点信息，包括：

通过图像检测网络对所述结构图片进行节点检测，得到所述节点信息。

可选的，所述通过图像检测网络对所述结构图片进行节点检测，得到所述节点信息，包括：

将所述结构图片输入到所述图像检测网络中，得到所述节点信息，其中，所述节点信息包括节点位置信息和节点类型。

可选的，所述对所述结构图片进行导引线识别，获取所述结构图片的导引线信息，包括：

通过图像分割网络对所述结构图片进行导引线检测，得到所述导引线信息。

可选的，所述根据所述节点信息和所述导引线信息进行结构还原，获取所述结构图片的还原结构图，包括：

根据所述导引线信息，生成所述结构图片的掩膜图像；

根据所述节点信息，通过结构还原方法在所述掩膜图像上进行结构还原，得到所述还原结构图。

可选的，所述根据所述节点信息，通过结构还原方法在所述掩膜图像上进行结构还原，得到所述还原结构图，包括：

将所述节点信息的根节点保存在节点栈中，并以所述根节点作为起点在所述掩膜图像上进行结构还原，结构还原步骤如下；

从所述节点栈中选取位于第j节点内的第p搜索点，若所述第p搜索点的节点标识与所述第j节点的节点标识不同，则确定所述第j节点为所述第p搜索点的子节点，且所述第p搜索点为节点；当所述第p搜索点的节点标识与所述第j节点的节点标识相同，则以所述第p搜索点作为中心在所述掩膜图像上进行连通域分析，当搜索到与所述第p搜索点连通的第q搜索点，则将所述第q搜索点保存在所述节点栈中，且将所述第p搜索点的节点标识作为所述第q搜索点的节点标识，其中，所述第j节点为所述节点信息中的节点；

在通过所述结构还原步骤搜索到所述节点信息中的每个节点之后，获取所述还原结构图。

可选的，在获取所述结构图片的还原结构图之后，所述方法还包括：

输出所述还原结构图，并以文件方式保存所述还原结构图。