[发明专利]一种建筑图纸的文本信息识别方法及装置

权利要求书

1.一种建筑图纸的文本信息识别方法，其特征在于，包括：

获取待识别建筑图纸的原始图像、文本区域颜色及目标文本颜色；

根据所述文本区域颜色和所述目标文本颜色对所述原始图像进行图像分割，确定目标文本图像；

将所述目标文本图像输入至文本识别模型进行文本识别，确定待识别建筑图纸对应的目标文本信息。

2.根据权利要求1所述的方法，其特征在于，所述根据所述文本区域颜色和所述目标文本颜色对所述原始图像进行图像分割，确定目标文本图像，包括：

根据所述文本区域颜色对应的第一RGB颜色分量，对所述原始图像进行颜色阈值分割，确定文本区域图像；

根据所述目标文本颜色对应的第二RGB颜色分量，对所述文本区域图像进行颜色阈值分割，确定所述目标文本图像。

3.根据权利要求2所述的方法，其特征在于，所述根据所述文本区域颜色对应的第一RGB颜色分量，对所述原始图像进行颜色阈值分割，确定文本区域图像，包括：

根据所述第一RGB颜色分量，对所述原始图像进行颜色阈值分割，确定第一图像；

根据所述第一图像的像素信息确定所述文本区域图像对应的第一边界位置信息；

根据所述第一边界位置信息及所述原始图像，确定所述文本区域图像。

4.根据权利要求2所述的方法，其特征在于，所述将所述目标文本图像输入至文本识别模型进行文本识别，确定待识别建筑图纸对应的目标文本信息，包括：

获取所述目标文本图像的像素信息；

根据所述目标文本图像的像素信息，确定目标中文本对应的第二边界信息；

根据所述第二边界信息及所述目标文本图像，确定更新后的目标文本图像；

将更新后的目标文本图像输入至文本识别模型进行文本识别，确定待识别建筑图纸对应的目标文本信息。

5.根据权利要求1所述的方法，其特征在于，所述文本识别模型通过以下方式进行训练得到的：

获取历史建筑图纸的历史目标文本图像及对应的历史目标文本信息；

将历史目标文本图像输入文本识别模型中，输出预测目标文本信息；

根据所述预测目标文本信息及所述目标文本信息，更新所述文本识别模型中的参数。

6.根据权利要求1所述的方法，其特征在于，所述获取待识别建筑图纸的原始图像，包括：

获取所述待识别建筑图纸；

根据预设格式转换方法对所述待识别建筑图纸进行格式转换，确定所述待识别建筑图纸的原始图像。

7.根据权利要求1所述的方法，其特征在于，所述文本识别模型为基于CNN+softmax的分类器模型。

8.一种建筑图纸的文本信息识别装置，其特征在于，包括：

获取模块，用于获取待识别建筑图纸的原始图像、文本区域颜色及目标文本颜色；

第一处理模块，用于根据所述文本区域颜色和所述目标文本颜色对所述原始图像进行图像分割，确定目标文本图像；

第二处理模块，用于将所述目标文本图像输入至文本识别模型进行文本识别，确定待识别建筑图纸对应的目标文本信息。

9.一种非暂态计算机可读存储介质，其特征在于，所述非暂态计算机可读存储介质存储计算机指令，所述计算机指令被处理器执行时实现如权利要求1-7中任一项所述的方法。

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

存储器和处理器，所述存储器和所述处理器之间互相通信连接，所述存储器中存储有计算机指令，所述处理器通过执行所述计算机指令，从而执行如权利要求1-7中任一项所述方法。