本发明属于计算机,具体涉及一种室内地图矢量化方法、装置和电子设备。
背景技术:
1、在相关技术中,室内地图生成方法包括:半自动化的室内地图数据矢量生成方法、基于图形学处理的室内地图的生成方法和基于平面图图像的室内自动制图与建模方法。
2、其中,半自动化的室内地图数据矢量生成方法,可以自动地对采集的室内区域进行规划化与模板化的后处理,但是室内区域的采集依旧需要人工参与,无法做到全自动的地图生产;基于图形学处理的室内地图的生成方法较多地依赖于传统的图像处理方法,需要有专业知识及复杂的调参过程,且无法处理较为复杂的图像,无法去除文字图标遮挡,无法解决遮挡带来的轮廓缺失等问题,泛化能力及鲁棒性较差;基于平面图图像的室内自动制图与建模方法采用蒙版递归卷积神经网络(mask recursive-convolutional neuralnetworks,mask r-cnn)模型直接对室内区域进行定位与分割,但需要采用定义较为复杂的规则如其中所述的共边检测处理房间之间的共边,以保证相邻房间之间的边对齐。
3、由上可知,相关技术中的室内地图生成方法,存在自动化程度低或模型复杂的问题,降低了矢量化室内地图的效率。
技术实现思路
1、本发明的目的是提供一种室内地图矢量化方法、装置和电子设备。能够基于深度学习的图像处理网络模型,对室内地图进行文字图标抹除、轮廓和关键点提取,并对关键点进行修正,并根据提取出的轮廓信息得到室内地图的矢量信息,从而提升了室内地图矢量化的效率。
2、为了解决上述技术问题,本发明是这样实现的:
3、第一方面,本发明提供了一种室内地图矢量化方法,包括:
4、将目标室内图像输入第一网络模型,获取所述第一网络模型输出的关键点热力图;
5、根据所述关键点热力图对所述目标室内图像的轮廓关键点进行修正,得到目标关键点;
6、根据所述目标室内图像的第一轮廓和所述目标关键点,确定所述目标室内图像的矢量信息。
7、第二方面,本发明还提供了一种室内地图矢量化装置,包括:
8、第一处理模块,用于将目标室内图像输入第一网络模型,获取所述第一网络模型输出的关键点热力图;
9、修正模块,用于根据所述关键点热力图对所述目标室内图像的轮廓关键点进行修正,得到目标关键点;
10、第一确定模块,用于根据所述目标室内图像的第一轮廓和所述目标关键点,确定所述目标室内图像的矢量信息。
11、第三方面,本发明还提供了一种电子设备,该电子设备包括处理器、存储器及存储在所述存储器上并可在所述处理器上运行的程序或指令,所述程序或指令被所述处理器执行时实现如第一方面所述的方法的步骤。
12、第四方面,本发明还提供了一种计算机可读存储介质,所述计算机可读存储介质上存储程序或指令,所述程序或指令被处理器执行时实现如第一方面所述的方法的步骤。
13、在本发明实施例中,将目标室内图像输入第一网络模型,获取所述第一网络模型输出的关键点热力图;根据所述关键点热力图对所述目标室内图像的轮廓关键点进行修正,得到目标关键点;根据所述目标室内图像的第一轮廓和所述目标关键点,确定所述目标室内图像的矢量信息。这样,可以利用网络模型训练得到的关键点热力图来反映目标室内图像中的轮廓关键点的分布概率,从而据此对目标室内图像的轮廓关键点进行修正,使得到的目标关键点更加准确,进而使得基于目标室内图像的第一轮廓和所述目标关键点所反映的目标室内图像的矢量信息更加准确。
1.一种室内地图矢量化方法,其特征在于,包括:
2.根据权利要求1所述的方法,其特征在于,在所述根据所述关键点热力图对所述目标室内图像的轮廓关键点进行修正,得到目标关键点之前,所述方法还包括:
3.根据权利要求2所述的方法,其特征在于,所述根据所述关键点热力图对所述目标室内图像的轮廓关键点进行修正,得到目标关键点,包括:
4.根据权利要求1至3中任一项所述的方法,其特征在于,在所述将目标室内图像输入第一网络模型,获取所述第一网络模型输出的关键点热力图之前,所述方法还包括:
5.根据权利要求4所述的方法,其特征在于,所述去除所述初始室内图像内的所述第一内容,得到中间图像,包括:
6.根据权利要求5所述的方法,其特征在于,所述第三网络模型的解码器包括基于注意力机制的特征聚合模块,所述将所述初始室内图像输入第三网络模型,并获取所述第三网络模型输出的与第一内容的位置对应的二值化掩膜,包括:
7.根据权利要求4所述的方法,其特征在于,所述对所述中间图像进行图像修复处理,得到所述目标室内图像,包括:
8.一种室内地图矢量化装置,其特征在于,包括:
9.一种电子设备,其特征在于,包括处理器、存储器及存储在所述存储器上并可在所述处理器上运行的计算机程序,所述计算机程序被所述处理器执行时实现如权利要求1至7中任一项所述的室内地图矢量化方法中的步骤。
10.一种计算机可读存储介质,其特征在于,所述计算机可读存储介质上存储有计算机程序,所述计算机程序被处理器执行时实现如权利要求1至7中任一项所述的室内地图矢量化方法中的步骤。