本技术涉及图像处理,尤其涉及一种图像处理方法及装置。
背景技术:
1、房屋户型图简称户型图,是指房屋的平面空间布局图。通常,户型图会对房屋的各个独立空间的使用功能、相应位置、尺寸进行图示,通过户型图可以直观的了解房屋的结构。
2、传统技术中,户型图的生成是由相关工作人员对房屋进行实地测量,并根据测量结果人工绘制的。为了避免人进行户型图的绘制,相关技术中提出根据房屋的全景图生成房屋的三维模型(点云模型、三维网格模型)以及根据房屋的三维模型(点云模型、三维网格模型)的俯视图来获取户型图。然而,这种户型图绘制方案极容易因为房屋内的衣柜、餐桌、茶几等物品的影响而得到错误的三维模型,进而导致获取的户型图与实际不符。
技术实现思路
1、为了解决上述技术问题或者至少部分地解决上述技术问题,本技术提供了一种图像处理方法及装置,用于提升根据房屋的全景图获取的房屋的户型图的准确性。
2、为了实现上述目的,本技术实施例提供的技术方案如下:
3、第一方面,本技术实施例提供了一种图像处理方法,包括:
4、对房屋的第一全景图进行实例分割,以得到实例掩膜图像,所述实例掩膜图像中包括:物品掩膜、墙体掩膜、天花板掩膜和/或地板掩膜;
5、从所述物品掩膜中确定待删除掩膜,所述待删除掩膜为与所述墙体掩膜、所述天花板掩膜以及所述地板掩膜中的至少两个存在相邻像素的物品掩膜;
6、删除所述实例掩膜图像中的所述待删除掩膜,获取待填充掩膜图像;
7、对所述待填充掩膜图像中的目标图像区域进行图像填充,获取所述第一全景图对应的房屋结构图,所述目标图像区域为所述待删除掩膜对应的图像区域;
8、基于所述第一全景图对应的房屋结构图,生成所述房屋的户型图。
9、作为本技术实施例一种可选的实施方式,所述对房屋的第一全景图进行实例分割,以得到实例掩膜图像,包括:
10、通过图像分割网络模型对所述第一全景图进行实例分割,以得到所述实例掩膜图像;
11、其中,所述图像分割网络模型为基于样本数据对预设机器学习模型进行训练获取的模型,所述样本数据包括多个样本全景图像、各个样本全景图像对应的样本实例掩膜图像以及各个样本实例掩膜图像中的各个掩膜的标签,所述样本实例掩膜图像中任一掩膜的标签用于指示该掩膜图像为物品掩膜或墙体掩膜或天花板掩膜或地板掩膜。
12、作为本技术实施例一种可选的实施方式,所述方法还包括:
13、在删除所述实例掩膜图像中的所述待删除掩膜,获取待填充掩膜图像之前,对所述待删除掩膜进行图像膨胀处理。
14、作为本技术实施例一种可选的实施方式,所述基于所述第一全景图对应的房屋结构图,生成所述房屋的户型图,包括:
15、识别所述房屋结构图中天花板与墙体之间的第一交界线、地板与墙体之间的第二交界线,以及墙面与墙体之间的第三交界线;
16、基于所述房屋结构图、所述第一交界线、所述第二交界线以及所述第三交界线,生成所述第一全景图对应的功能间立体模型;
17、基于所述第一全景图对应的功能间立体模型,生成所述房屋的户型图。
18、作为本技术实施例一种可选的实施方式,所述基于所述第一全景图对应的功能间立体模型,生成所述房屋的户型图,包括:
19、获取所述房屋的第二全景图的拍摄位置,以及所述第二全景图对应的功能间立体模型,所述第二全景图的拍摄位置与所述第一全景图的拍摄位置不同;
20、基于所述第一全景图的拍摄位置和所述第二全景图的拍摄位置,合并所述第一全景图对应的功能间立体模型和所述第二全景图对应的功能间立体模型生成所述房屋的立体模型;
21、基于所述房屋的立体模型,生成所述房屋的户型图。
22、作为本技术实施例一种可选的实施方式,所述基于所述房屋的立体模型,生成所述房屋的户型图,包括:
23、基于所述房屋的立体模型的俯视图,生成所述房屋的户型图。
24、作为本技术实施例一种可选的实施方式,所述方法还包括:
25、基于所述第一全景图确定所述第一全景图对应的功能间中的物品,以及基于所述第二全景图确定所述第二全景图对应的功能间中的物品;
26、根据所述第一全景图对应的功能间中的物品的类型以及所述第二全景图对应的功能间中的物品的类型,确定所述户型图中的各个功能间的类型。
27、第二方面,本技术实施例提供了一种图像处理装置,包括:
28、分割单元,用于对房屋的第一全景图进行实例分割,以得到实例掩膜图像,所述实例掩膜图像中包括:物品掩膜、墙体掩膜、天花板掩膜和/或地板掩膜;
29、处理单元,用于从所述物品掩膜中确定待删除掩膜,所述待删除掩膜为与所述墙体掩膜、所述天花板掩膜以及所述地板掩膜中的至少两个存在相邻像素的物品掩膜;
30、擦除单元,用于删除所述实例掩膜图像中的所述待删除掩膜,获取待填充掩膜图像;
31、填充单元,用于对所述待填充掩膜图像中的目标图像区域进行图像填充,获取所述第一全景图对应的房屋结构图,所述目标图像区域为所述待删除掩膜对应的图像区域;
32、生成单元,用于基于所述第一全景图对应的房屋结构图,生成所述房屋的户型图。
33、作为本技术实施例一种可选的实施方式,所述分割单元,具体用于通过图像分割网络模型对所述第一全景图进行实例分割,以得到所述实例掩膜图像;
34、其中,所述图像分割网络模型为基于样本数据对预设机器学习模型进行训练获取的模型,所述样本数据包括多个样本全景图像、各个样本全景图像对应的样本实例掩膜图像以及各个样本实例掩膜图像中的各个掩膜的标签,所述样本实例掩膜图像中任一掩膜的标签用于指示该掩膜图像为物品掩膜或墙体掩膜或天花板掩膜或地板掩膜。
35、作为本技术实施例一种可选的实施方式,所述处理单元,还用于在删除所述实例掩膜图像中的所述待删除掩膜,获取待填充掩膜图像之前,对所述待删除掩膜进行图像膨胀处理。
36、作为本技术实施例一种可选的实施方式,所述生成单元,具体用于识别所述房屋结构图中天花板与墙体之间的第一交界线、地板与墙体之间的第二交界线,以及墙面与墙体之间的第三交界线;基于所述房屋结构图、所述第一交界线、所述第二交界线以及所述第三交界线,生成所述第一全景图对应的功能间立体模型,基于所述第一全景图对应的功能间立体模型,生成所述房屋的户型图。
37、作为本技术实施例一种可选的实施方式,所述生成单元,具体用于获取所述房屋的第二全景图的拍摄位置,以及所述第二全景图对应的功能间立体模型,所述第二全景图的拍摄位置与所述第一全景图的拍摄位置不同;基于所述第一全景图的拍摄位置和所述第二全景图的拍摄位置,合并所述第一全景图对应的功能间立体模型和所述第二全景图对应的功能间立体模型生成所述房屋的立体模型;基于所述房屋的立体模型,生成所述房屋的户型图。
38、作为本技术实施例一种可选的实施方式,所述生成单元,具体用于基于所述房屋的立体模型的俯视图,生成所述房屋的户型图。
39、作为本技术实施例一种可选的实施方式,所述生成单元,还用于基于所述第一全景图确定所述第一全景图对应的功能间中的物品,以及基于所述第二全景图确定所述第二全景图对应的功能间中的物品;根据所述第一全景图对应的功能间中的物品的类型以及所述第二全景图对应的功能间中的物品的类型,确定所述户型图中的各个功能间的类型。
40、第三方面,提供一种电子设备,包括:处理器、存储器及存储在所述存储器上并可在所述处理器上运行的计算机程序,所述计算机程序被所述处理器执行时实现上述任一项所述的图像处理方法。
41、第四方面,提供一种计算机可读存储介质,包括:所述计算机可读存储介质上存储计算机程序,所述计算机程序被处理器执行时实现上述任一项所述的图像处理方法。
42、第五方面,提供一种计算机程序产品,所述计算机程序产品中存储计算机程序,所述计算机程序被处理器执行时实现上述任一项所述的图像处理方法。
43、本技术实施例提供的图像处理方法在获取房屋的第一全景图之后,首先对房屋的第一全景图进行实例分割以得到实例掩膜图像,然后确定所述实例掩膜图像中与所述墙体掩膜、所述天花板掩膜以及所述地板掩膜中的至少两个存在相邻像素的物品掩膜,以及删除所述实例掩膜图像中与所述墙体掩膜、所述天花板掩膜以及所述地板掩膜中的至少两个存在相邻像素的物品掩膜,再对所述待填充掩膜图像中与所述墙体掩膜、所述天花板掩膜以及所述地板掩膜中的至少两个存在相邻像素的物品掩膜对应的图像区域进行图像填充,获取所述第一全景图对应的房屋结构图,以及基于所述房屋的立体模型,生成所述房屋的户型图。由于本技术实施例在得到房屋的实例掩膜图像后,会先将与所述墙体掩膜、所述天花板掩膜以及所述地板掩膜中的至少两个存在相邻像素的物品掩膜删除,以及对删除物品掩膜的区域进行填充,因此本技术实施例可以避免房屋内的物品影响房屋结构的确定,提升根据全景图确定的房屋结构的准确性,进而提升根据房屋的全景图获取的房屋的户型图的准确性。
1.一种图像处理方法,其特征在于,包括:
2.根据权利要求1所述的方法,其特征在于,所述对房屋的第一全景图进行实例分割,以得到实例掩膜图像,包括:
3.根据权利要求1所述的方法,其特征在于,所述方法还包括:
4.根据权利要求1所述的方法,其特征在于,所述基于所述第一全景图对应的房屋结构图,生成所述房屋的户型图,包括:
5.根据权利要求4所述的方法,其特征在于,所述基于所述第一全景图对应的功能间立体模型,生成所述房屋的户型图,包括:
6.根据权利要求5所述的方法,其特征在于,所述基于所述房屋的立体模型,生成所述房屋的户型图,包括:
7.根据权利要求5所述的方法,其特征在于,所述方法还包括:
8.一种图像处理装置,其特征在于,包括:
9.一种电子设备,其特征在于,包括:处理器、存储器及存储在所述存储器上并可在所述处理器上运行的计算机程序,所述计算机程序被所述处理器执行时实现如权利要求1-7任一项所述的图像处理方法。
10.一种计算机可读存储介质,其特征在于,包括:所述计算机可读存储介质上存储计算机程序,所述计算机程序被处理器执行时实现如权利要求1-7任一项所述的图像处理方法。
