一种基于图神经网络的结构化户型解析方法和系统与流程

1.本发明涉及计算机技术领域，尤其是一种基于图神经网络的结构化户型解析方法和系统。


背景技术：

2.在房地产行业和家装行业设计中，设计师的历史设计方案经常出现没有为房间类型命名的情况，另外很大一部分数量的平面户型图通常只包含房屋的结构信息而缺少房间类型，但房间类型对后续设计和分析有十分重要的意义，因此快速准确地分析出户型图内的各个房间的类型具有重要的应用价值。在传统的户型图分析任务中，数据主体一般是绘制好的二维平面户型图，往往需要投入大量的人力物力才能完成对户型图的整体分析，一般通过资料调查和人为校对和推理来确定每个房间的类别和位置，这种方式的效率较低而且准确率低。


技术实现要素：

3.有鉴于此，本发明实施例提供一种效率高且准确率高的，基于图神经网络的结构化户型解析方法和系统。
4.本发明实施例的一方面提供了一种基于图神经网络的结构化户型解析方法，包括：
5.根据户型数据库获取所有户型结构信息，并对所述户型结构信息进行结构化处理，得到结构化的户型结构信息；
6.根据结构化的户型结构信息构建户型连通气泡图并进行特征工程，获取气泡图中所有节点的特征项和类别标签作为训练数据；
7.搭建基于图神经网络模型的分类模型，输入训练数据进行训练，保存最优模型权重；
8.根据最优模型权重对测试户型数据进行预测，计算分类精度，进而根据分类精度构建得到目标模型；
9.根据目标模型对待解析的户型信息进行解析，得到房间分类结果与户型结构数据后进行可视化展示。
10.可选地，所述根据户型数据库获取所有户型结构信息，并对所述户型结构信息进行结构化处理，得到结构化的户型结构信息，包括：
11.获取地产公司和家装公司积累的户型数据；
12.筛选出具有房间类型标注的户型图数据，针对无坐标信息的平面户型图矢量数据和带有坐标信息的户型数据分别进行处理；
13.针对无坐标信息的平面户型图矢量数据进行矢量数据栅格化，提取户型内的内外墙体和门窗的信息；
14.针对带有坐标信息的户型数据，依次提取房间类别，房间包含的墙体线段坐标，墙
体包含的门窗线段坐标；
15.根据数据分析和统计，将待分析的房间类型进行归类；
16.以一个户型为单位，将处理后的户型数据以文本的形式整合到一起，保存为结构化户型数据集合。
17.可选地，所述方法还包括：
18.分别以线段表示每面墙体、门和窗，并保存每一面墙体所包含的门窗线段坐标，提取每个房间的墙体闭合多边形轮廓，以二维点串的形式表示每个房间多边形，并保存房间类别和多边形轮廓的对应关系以及每一个房间所包含的墙体线段坐标。
19.可选地，所述根据数据分析和统计，将待分析的房间类型进行归类，包括：
20.根据数据分析和统计，将待分析的房间类型归纳为10类，分别是客厅、辅助房、阳台、卧室、厨房、浴室、走廊、多功能房、入户花园和自定义房间。
21.可选地，所述根据结构化的户型结构信息构建户型连通气泡图并进行特征工程，获取气泡图中所有节点的特征项和类别标签作为训练数据，包括：
22.获取结构化户型数据集合；
23.遍历结构化户型数据集合，对于集合中的每一个户型数据，执行以下处理：以每个房间作为节点，节点标注为对应的房间类型；将房间之间的连通关系作为边，边类型只限定为连通关系；构建出一个户型连通气泡图；依次计算各个房间节点的面积、房间包围框、房间长边、房间短边、房间周长、房间中心坐标、房间门窗占比这7项数据作为节点特征；根据气泡图内连通关系构建出一个关系列表，连通节点特征一起保存为一份训练样本；
24.将遍历结束后得到户型数据集合划分为训练数据和测试数据。
25.可选地，所述搭建基于图神经网络模型的分类模型，输入训练数据进行训练，保存最优模型权重，包括：
26.获取训练数据的集合；
27.根据气泡图结构初始化整体图结构，记为g(v,e),其中v代表图中的边集，e代表图中的点集，输入节点特征，记为{xv},初始化第k阶权重矩阵，记为wk，k从1开始取值，选取均值方法作为聚合函数，记为aggk，设置图中的节点采样函数，记为n，其中，聚合和采样的操作上限设置为二阶，即k＝2；
28.构建图神经网络模型，图中所有节点的输入特征，记为h
v0
，首先利用采样函数n分别采样节点h
v0
的一阶邻居节点、二阶邻居节点，分别记为h
v1
、h
v2
，然后利用聚合函数aggk和权重矩阵wk分别对二阶邻居节点h
v2
和一阶邻居节点h
v1
先后进行聚合和加权，得到节点的特征向量h
vk
，最后将得到的节点向量进行二范数归一化；
29.将节点特征向量h
vk
输入维度为(7，10)的全连接层进行分类；
30.读取训练数据并批次载入模型进行训练，采用交叉熵损失对模型预测的分类结果做损失计算；
31.采用adam优化器作为模型，初始学习率为0.004，采用阶梯下降的学习率衰减方法，batch size定义为32个房间，总共训练100轮次。
32.可选地，所述根据目标模型对待解析的户型信息进行解析，得到房间分类结果与户型结构数据后进行可视化展示，包括：
33.读取经过后处理的房间分类结果；
34.根据测试数据索引，从获取的结构化户型数据集合中读取测试数据的户型结构信息；
35.遍历整个户型的所有房间，按照房间索引，对应地绘制出房间二维结构图并标注房间类型；
36.展示带有房间类型的平面户型图。
37.本发明实施例的另一方面还提供了一种基于图神经网络的结构化户型解析系统，包括：
38.第一模块，用于根据户型数据库获取所有户型结构信息，并对所述户型结构信息进行结构化处理，得到结构化的户型结构信息；
39.第二模块，用于根据结构化的户型结构信息构建户型连通气泡图并进行特征工程，获取气泡图中所有节点的特征项和类别标签作为训练数据；
40.第三模块，用于搭建基于图神经网络模型的分类模型，输入训练数据进行训练，保存最优模型权重；
41.第四模块，用于根据最优模型权重对测试户型数据进行预测，计算分类精度，进而根据分类精度构建得到目标模型；
42.第五模块，用于根据目标模型对待解析的户型信息进行解析，得到房间分类结果与户型结构数据后进行可视化展示。
43.本发明实施例的另一方面还提供了一种电子设备，包括处理器以及存储器；
44.所述存储器用于存储程序；
45.所述处理器执行所述程序实现如前面所述的方法。
46.本发明实施例的另一方面还提供了一种计算机可读存储介质，所述存储介质存储有程序，所述程序被处理器执行实现如前面所述的方法。
47.本发明实施例还公开了一种计算机程序产品或计算机程序，该计算机程序产品或计算机程序包括计算机指令，该计算机指令存储在计算机可读存储介质中。计算机设备的处理器可以从计算机可读存储介质读取该计算机指令，处理器执行该计算机指令，使得该计算机设备执行前面的方法。
48.本发明的实施例根据户型数据库获取所有户型结构信息，并对所述户型结构信息进行结构化处理，得到结构化的户型结构信息；根据结构化的户型结构信息构建户型连通气泡图并进行特征工程，获取气泡图中所有节点的特征项和类别标签作为训练数据；搭建基于图神经网络模型的分类模型，输入训练数据进行训练，保存最优模型权重；根据最优模型权重对测试户型数据进行预测，计算分类精度，进而根据分类精度构建得到目标模型；根据目标模型对待解析的户型信息进行解析，得到房间分类结果与户型结构数据后进行可视化展示。本发明的效率高且准确率高。
附图说明
49.为了更清楚地说明本技术实施例中的技术方案，下面将对实施例描述中所需要使用的附图作简单地介绍，显而易见地，下面描述中的附图仅仅是本技术的一些实施例，对于本领域普通技术人员来讲，在不付出创造性劳动的前提下，还可以根据这些附图获得其他的附图。
50.图1为本发明实施例提供的一种基于图神经网络的结构化户型解析方法的整体步骤流程图；
51.图2为本发明实施例提供的基于图神经网络的结构化户型图理解的方法的模型示意图。
具体实施方式
52.为了使本技术的目的、技术方案及优点更加清楚明白，以下结合附图及实施例，对本技术进行进一步详细说明。应当理解，此处所描述的具体实施例仅仅用以解释本技术，并不用于限定本技术。
53.针对现有技术存在的问题，本发明实施例提供了一种基于图神经网络的结构化户型解析方法，包括：
54.根据户型数据库获取所有户型结构信息，并对所述户型结构信息进行结构化处理，得到结构化的户型结构信息；
55.根据结构化的户型结构信息构建户型连通气泡图并进行特征工程，获取气泡图中所有节点的特征项和类别标签作为训练数据；
56.搭建基于图神经网络模型的分类模型，输入训练数据进行训练，保存最优模型权重；
57.根据最优模型权重对测试户型数据进行预测，计算分类精度，进而根据分类精度构建得到目标模型；
58.根据目标模型对待解析的户型信息进行解析，得到房间分类结果与户型结构数据后进行可视化展示。
59.可选地，所述根据户型数据库获取所有户型结构信息，并对所述户型结构信息进行结构化处理，得到结构化的户型结构信息，包括：
60.获取地产公司和家装公司积累的户型数据；
61.筛选出具有房间类型标注的户型图数据，针对无坐标信息的平面户型图矢量数据和带有坐标信息的户型数据分别进行处理；
62.针对无坐标信息的平面户型图矢量数据进行矢量数据栅格化，提取户型内的内外墙体和门窗的信息；
63.针对带有坐标信息的户型数据，依次提取房间类别，房间包含的墙体线段坐标，墙体包含的门窗线段坐标；
64.根据数据分析和统计，将待分析的房间类型进行归类；
65.以一个户型为单位，将处理后的户型数据以文本的形式整合到一起，保存为结构化户型数据集合。
66.可选地，所述方法还包括：
67.分别以线段表示每面墙体、门和窗，并保存每一面墙体所包含的门窗线段坐标，提取每个房间的墙体闭合多边形轮廓，以二维点串的形式表示每个房间多边形，并保存房间类别和多边形轮廓的对应关系以及每一个房间所包含的墙体线段坐标。
68.可选地，所述根据数据分析和统计，将待分析的房间类型进行归类，包括：
69.根据数据分析和统计，将待分析的房间类型归纳为10类，分别是客厅、辅助房、阳
台、卧室、厨房、浴室、走廊、多功能房、入户花园和自定义房间。
70.可选地，所述根据结构化的户型结构信息构建户型连通气泡图并进行特征工程，获取气泡图中所有节点的特征项和类别标签作为训练数据，包括：
71.获取结构化户型数据集合；
72.遍历结构化户型数据集合，对于集合中的每一个户型数据，执行以下处理：以每个房间作为节点，节点标注为对应的房间类型；将房间之间的连通关系作为边，边类型只限定为连通关系；构建出一个户型连通气泡图；依次计算各个房间节点的面积、房间包围框、房间长边、房间短边、房间周长、房间中心坐标、房间门窗占比这7项数据作为节点特征；根据气泡图内连通关系构建出一个关系列表，连通节点特征一起保存为一份训练样本；
73.将遍历结束后得到户型数据集合划分为训练数据和测试数据。
74.可选地，所述搭建基于图神经网络模型的分类模型，输入训练数据进行训练，保存最优模型权重，包括：
75.获取训练数据的集合；
76.根据气泡图结构初始化整体图结构，记为g(v,e),其中v代表图中的边集，e代表图中的点集，输入节点特征，记为{xv},初始化第k阶权重矩阵，记为wk，k从1开始取值，选取均值方法作为聚合函数，记为aggk，设置图中的节点采样函数，记为n，其中，聚合和采样的操作上限设置为二阶，即k＝2；
77.构建图神经网络模型，图中所有节点的输入特征，记为h
v0
，首先利用采样函数n分别采样节点h
v0
的一阶邻居节点、二阶邻居节点，分别记为h
v1
、h
v2
，然后利用聚合函数aggk和权重矩阵wk分别对二阶邻居节点h
v2
和一阶邻居节点h
v1
先后进行聚合和加权，得到节点的特征向量h
vk
，最后将得到的节点向量进行二范数归一化；
78.将节点特征向量h
vk
输入维度为(7，10)的全连接层进行分类；
79.读取训练数据并批次载入模型进行训练，采用交叉熵损失对模型预测的分类结果做损失计算；
80.采用adam优化器作为模型，初始学习率为0.004，采用阶梯下降的学习率衰减方法，batch size定义为32个房间，总共训练100轮次。
81.可选地，所述根据目标模型对待解析的户型信息进行解析，得到房间分类结果与户型结构数据后进行可视化展示，包括：
82.读取经过后处理的房间分类结果；
83.根据测试数据索引，从获取的结构化户型数据集合中读取测试数据的户型结构信息；
84.遍历整个户型的所有房间，按照房间索引，对应地绘制出房间二维结构图并标注房间类型；
85.展示带有房间类型的平面户型图。
86.本发明实施例的另一方面还提供了一种基于图神经网络的结构化户型解析系统，包括：
87.第一模块，用于根据户型数据库获取所有户型结构信息，并对所述户型结构信息进行结构化处理，得到结构化的户型结构信息；
88.第二模块，用于根据结构化的户型结构信息构建户型连通气泡图并进行特征工
程，获取气泡图中所有节点的特征项和类别标签作为训练数据；
89.第三模块，用于搭建基于图神经网络模型的分类模型，输入训练数据进行训练，保存最优模型权重；
90.第四模块，用于根据最优模型权重对测试户型数据进行预测，计算分类精度，进而根据分类精度构建得到目标模型；
91.第五模块，用于根据目标模型对待解析的户型信息进行解析，得到房间分类结果与户型结构数据后进行可视化展示。
92.本发明实施例的另一方面还提供了一种电子设备，包括处理器以及存储器；
93.所述存储器用于存储程序；
94.所述处理器执行所述程序实现如前面所述的方法。
95.本发明实施例的另一方面还提供了一种计算机可读存储介质，所述存储介质存储有程序，所述程序被处理器执行实现如前面所述的方法。
96.本发明实施例还公开了一种计算机程序产品或计算机程序，该计算机程序产品或计算机程序包括计算机指令，该计算机指令存储在计算机可读存储介质中。计算机设备的处理器可以从计算机可读存储介质读取该计算机指令，处理器执行该计算机指令，使得该计算机设备执行前面的方法。
97.下面结合说明书附图，对本发明的具体实施过程进行详细描述：
98.本发明提供了一种基于图神经网络的结构化户型数据的理解方法和系统，该系统从户型图的结构化组织入手，以图结构为媒介，依据户型内房间的连通性为规则构建结构化户型数据，以图神经网络模型为方法对结构化户型数据进行分析处理，能够快速推理出户型内各个房间的类型和位置，最终输出带有标注的户型图可视化结果。该系统具备两方面优点，首先，通过结构化数据描述户型图相比二维户型图像，既可以极大程度上节省储存空间，又可以方便图神经网络模型的数据吞吐。其次，利用图神经网络分析户型既能隐含得学习了户型内的房间连通关系又能抽象出各类房间再高维空间中的特征向量。
99.本发明提供一种基于图神经网络的结构化户型理解方法和系统，如图1所示，方法包括以下步骤：
100.1、根据户型数据库获取所有户型结构信息，并将对应的结构信息进行结构化处理；
101.2、根据结构化的户型数据构建户型连通气泡图并进行特征工程，获取气泡图中所有节点的特征项和类别标签作为训练数据；
102.3、搭建基于图神经网络分类的模型，输入训练数据进行训练，保存最优模型权重；
103.4、根据保存的模型对测试户型数据进行预测，并计算分类精度；
104.5、将经过后处理的房间分类结果与户型结构数据进行可视化。
105.第一步，根据户型数据库获取所有户型结构信息，并将对应的结构信息进行结构化处理。
106.a.获取地产公司和家装公司积累的户型数据；
107.b.筛选出具有房间类型标注的户型图数据，针对无坐标信息的平面户型图矢量数据和带有坐标信息的户型数据分别进行处理；
108.c.针对无坐标信息的平面户型图矢量数据进行矢量数据栅格化，提取户型内的内
外墙体和门窗，分别以线段表示每面墙体、门和窗，并保存每一面墙体所包含的门窗线段坐标，提取每个房间的墙体闭合多边形轮廓，以二维点串的形式表示每个房间多边形，并保存房间类别和多边形轮廓的对应关系以及每一个房间所包含的墙体线段坐标。
109.d.针对带有坐标信息的户型数据，依次提取房间类别，房间包含的墙体线段坐标，墙体包含的门窗线段坐标，组织方式同上。
110.e.根据数据分析和统计，将待分析的房间类型归纳为10类，分别是客厅、辅助房(主卫或衣帽间都归为此类)、阳台、卧室、厨房、浴室(淋浴室和卫生间都归为此类)、走廊、多功能房、入户花园和自定义房间(代指未知分类的房间)。
111.f.将上述数据以一个户型为单位，将处理后的户型数据以文本的形式整合到一起，保存为结构化户型数据集合。
112.第二步，根据结构化的户型数据构建户型连通气泡图并进行特征工程，获取气泡图中所有节点的特征项和类别标签作为训练数据。
113.a.获取第一步已经构建的结构化户型数据集合；
114.b.遍历结构化户型数据集合，对于集合中的每一个户型数据，都做以下处理：
115.b1.首先以每个房间作为节点，节点标注为对应的房间类型；
116.b2.房间之间的连通关系作为边，边类型只限定为连通关系；
117.b3.构建出一个户型连通气泡图；
118.b4.依次计算各个房间节点的面积、房间包围框、房间长边、房间短边、房间周长、房间中心坐标、房间门窗占比这7项数据作为节点特征；
119.b5.根据气泡图内连通关系构建出一个关系列表，连通节点特征一起保存为一份训练样本；c.将遍历结束后得到户型数据集合划分为训练数据和测试数据。
120.第三步，如图2所示，搭建图神经网络模型的分类模型，输入训练数据进行训练，保存最优模型权重。
121.a.获取第二步的训练数据集合；
122.b.根据气泡图结构初始化整体图结构，记为g(v,e),其中v代表图中的边集，e代表图中的点集，输入节点特征，记为{xv},初始化第k阶权重矩阵，记为wk，k从1开始取值，选取均值方法作为聚合函数，记为aggk，设置图中的节点采样函数，记为n，其中，聚合和采样的操作上限设置为二阶，即k＝2；
123.c.构建图神经网络模型，图中所有节点的输入特征，记为h
v0
，首先利用采样函数n分别采样节点h
v0
的一阶邻居节点、二阶邻居节点，分别记为h
v1
、h
v2
，然后利用聚合函数aggk和权重矩阵wk分别对二阶邻居节点h
v2
和一阶邻居节点h
v1
先后进行聚合和加权，得到节点的特征向量h
vk
，最后将得到的节点向量进行二范数归一化，以方便空间上的相似度计算。
124.d.将节点特征向量h
vk
输入维度为(7，10)的全连接层进行分类；
125.e.读取第二步划分的训练数据并批次载入模型进行训练，对模型预测的分类结果做损失计算，采用交叉熵损失进行；
126.f.模型采用adam优化器，初始学习率为0.004，采用阶梯下降的学习率衰减方法，batch size定义为32个房间，总共训练100轮次；
127.第四步，根据保存的模型对测试户型数据进行预测，并计算分类精度。
128.a.加载第三步保存的模型；
129.b.读取第二步划分的测试数据，载入模型并获取分类结果，选取精准度作为指标，根据第一步设置的10个类别分别进行精度计算。
130.第五步，将经过后处理的房间分类结果与户型结构数据进行可视化。
131.a.读取经过后处理的房间分类结果；
132.b.根据测试数据索引，从第一步获取的结构化户型数据集合中读取测试数据的户型结构信息；
133.c.遍历整个户型的所有房间，按照房间索引，对应地绘制出房间二维结构图并标注房间类型；
134.d.最终展示带有房间类型的平面户型图。
135.本发明具有以下特点：
136.1、基于图神经网络模型的户型图理解方法，即该模型训练流程；
137.2、对户型图数据的收集和处理方法以及结构化方法，包括无坐标信息的平面户型图矢量数据和带有坐标信息的户型数据；
138.3、对户型图结构数据的基于图结构的特征向量表示方法。
139.综上所述，本发明通过图神经网络模型，辅以结构化的户型数据处理，迅速准确地分析出户型内各个房间的类别信息，并展示带有房间类型标注的平面户型图，可以帮助人和机器快速理解整个户型。利用此种方法和系统，可以大大节约生产生活在户型理解上所耗费的人力物力，提升上游任务的工作效率。
140.在一些可选择的实施例中，在方框图中提到的功能/操作可以不按照操作示图提到的顺序发生。例如，取决于所涉及的功能/操作，连续示出的两个方框实际上可以被大体上同时地执行或所述方框有时能以相反顺序被执行。此外，在本发明的流程图中所呈现和描述的实施例以示例的方式被提供，目的在于提供对技术更全面的理解。所公开的方法不限于本文所呈现的操作和逻辑流程。可选择的实施例是可预期的，其中各种操作的顺序被改变以及其中被描述为较大操作的一部分的子操作被独立地执行。
141.此外，虽然在功能性模块的背景下描述了本发明，但应当理解的是，除非另有相反说明，所述的功能和/或特征中的一个或多个可以被集成在单个物理装置和/或软件模块中，或者一个或多个功能和/或特征可以在单独的物理装置或软件模块中被实现。还可以理解的是，有关每个模块的实际实现的详细讨论对于理解本发明是不必要的。更确切地说，考虑到在本文中公开的装置中各种功能模块的属性、功能和内部关系的情况下，在工程师的常规技术内将会了解该模块的实际实现。因此，本领域技术人员运用普通技术就能够在无需过度试验的情况下实现在权利要求书中所阐明的本发明。还可以理解的是，所公开的特定概念仅仅是说明性的，并不意在限制本发明的范围，本发明的范围由所附权利要求书及其等同方案的全部范围来决定。
142.所述功能如果以软件功能单元的形式实现并作为独立的产品销售或使用时，可以存储在一个计算机可读取存储介质中。基于这样的理解，本发明的技术方案本质上或者说对现有技术做出贡献的部分或者该技术方案的部分可以以软件产品的形式体现出来，该计算机软件产品存储在一个存储介质中，包括若干指令用以使得一台计算机设备(可以是个人计算机，服务器，或者网络设备等)执行本发明各个实施例所述方法的全部或部分步骤。而前述的存储介质包括：u盘、移动硬盘、只读存储器(rom，read-only memory)、随机存取存
储器(ram，random access memory)、磁碟或者光盘等各种可以存储程序代码的介质。
143.在流程图中表示或在此以其他方式描述的逻辑和/或步骤，例如，可以被认为是用于实现逻辑功能的可执行指令的定序列表，可以具体实现在任何计算机可读介质中，以供指令执行系统、装置或设备(如基于计算机的系统、包括处理器的系统或其他可以从指令执行系统、装置或设备取指令并执行指令的系统)使用，或结合这些指令执行系统、装置或设备而使用。就本说明书而言，“计算机可读介质”可以是任何可以包含、存储、通信、传播或传输程序以供指令执行系统、装置或设备或结合这些指令执行系统、装置或设备而使用的装置。
144.计算机可读介质的更具体的示例(非穷尽性列表)包括以下：具有一个或多个布线的电连接部(电子装置)、便携式计算机盘盒(磁装置)、随机存取存储器(ram)、只读存储器(rom)、可擦除可编辑只读存储器(eprom或闪速存储器)、光纤装置以及便携式光盘只读存储器(cdrom)。另外，计算机可读介质甚至可以是可在其上打印所述程序的纸或其他合适的介质，因为可以例如通过对纸或其他介质进行光学扫描，接着进行编辑、解译或必要时以其他合适方式进行处理来以电子方式获得所述程序，然后将其存储在计算机存储器中。
145.应当理解，本发明的各部分可以用硬件、软件、固件或它们的组合来实现。在上述实施方式中，多个步骤或方法可以用存储在存储器中且由合适的指令执行系统执行的软件或固件来实现。例如，如果用硬件来实现，和在另一实施方式中一样，可用本领域公知的下列技术中的任一项或他们的组合来实现：具有用于对数据信号实现逻辑功能的逻辑门电路的离散逻辑电路，具有合适的组合逻辑门电路的专用集成电路，可编程门阵列(pga)，现场可编程门阵列(fpga)等。
146.在本说明书的描述中，参考术语“一个实施例”、“一些实施例”、“示例”、“具体示例”、或“一些示例”等的描述意指结合该实施例或示例描述的具体特征、结构、材料或者特点包含于本发明的至少一个实施例或示例中。在本说明书中，对上述术语的示意性表述不一定指的是相同的实施例或示例。而且，描述的具体特征、结构、材料或者特点可以在任何的一个或多个实施例或示例中以合适的方式结合。
147.尽管已经示出和描述了本发明的实施例，本领域的普通技术人员可以理解：在不脱离本发明的原理和宗旨的情况下可以对这些实施例进行多种变化、修改、替换和变型，本发明的范围由权利要求及其等同物限定。
148.以上是对本发明的较佳实施进行了具体说明，但本发明并不限于所述实施例，熟悉本领域的技术人员在不违背本发明精神的前提下还可做出种种的等同变形或替换，这些等同的变形或替换均包含在本技术权利要求所限定的范围内。

技术特征：
1.一种基于图神经网络的结构化户型解析方法，其特征在于，包括：根据户型数据库获取所有户型结构信息，并对所述户型结构信息进行结构化处理，得到结构化的户型结构信息；根据结构化的户型结构信息构建户型连通气泡图并进行特征工程，获取气泡图中所有节点的特征项和类别标签作为训练数据；搭建基于图神经网络模型的分类模型，输入训练数据进行训练，保存最优模型权重；根据最优模型权重对测试户型数据进行预测，计算分类精度，进而根据分类精度构建得到目标模型；根据目标模型对待解析的户型信息进行解析，得到房间分类结果与户型结构数据后进行可视化展示。2.根据权利要求1所述的一种基于图神经网络的结构化户型解析方法，其特征在于，所述根据户型数据库获取所有户型结构信息，并对所述户型结构信息进行结构化处理，得到结构化的户型结构信息，包括：获取地产公司和家装公司积累的户型数据；筛选出具有房间类型标注的户型图数据，针对无坐标信息的平面户型图矢量数据和带有坐标信息的户型数据分别进行处理；针对无坐标信息的平面户型图矢量数据进行矢量数据栅格化，提取户型内的内外墙体和门窗的信息；针对带有坐标信息的户型数据，依次提取房间类别，房间包含的墙体线段坐标，墙体包含的门窗线段坐标；根据数据分析和统计，将待分析的房间类型进行归类；以一个户型为单位，将处理后的户型数据以文本的形式整合到一起，保存为结构化户型数据集合。3.根据权利要求2所述的一种基于图神经网络的结构化户型解析方法，其特征在于，所述方法还包括：分别以线段表示每面墙体、门和窗，并保存每一面墙体所包含的门窗线段坐标，提取每个房间的墙体闭合多边形轮廓，以二维点串的形式表示每个房间多边形，并保存房间类别和多边形轮廓的对应关系以及每一个房间所包含的墙体线段坐标。4.根据权利要求2所述的一种基于图神经网络的结构化户型解析方法，其特征在于，所述根据数据分析和统计，将待分析的房间类型进行归类，包括：根据数据分析和统计，将待分析的房间类型归纳为10类，分别是客厅、辅助房、阳台、卧室、厨房、浴室、走廊、多功能房、入户花园和自定义房间。5.根据权利要求2所述的一种基于图神经网络的结构化户型解析方法，其特征在于，所述根据结构化的户型结构信息构建户型连通气泡图并进行特征工程，获取气泡图中所有节点的特征项和类别标签作为训练数据，包括：获取结构化户型数据集合；遍历结构化户型数据集合，对于集合中的每一个户型数据，执行以下处理：以每个房间作为节点，节点标注为对应的房间类型；将房间之间的连通关系作为边，边类型只限定为连通关系；构建出一个户型连通气泡图；依次计算各个房间节点的面积、房间包围框、房间长
边、房间短边、房间周长、房间中心坐标、房间门窗占比这7项数据作为节点特征；根据气泡图内连通关系构建出一个关系列表，连通节点特征一起保存为一份训练样本；将遍历结束后得到户型数据集合划分为训练数据和测试数据。6.根据权利要求1所述的一种基于图神经网络的结构化户型解析方法，其特征在于，所述搭建基于图神经网络模型的分类模型，输入训练数据进行训练，保存最优模型权重，包括：获取训练数据的集合；根据气泡图结构初始化整体图结构，记为g(v,e),其中v代表图中的边集，e代表图中的点集，输入节点特征，记为{x
v
},初始化第k阶权重矩阵，记为w
k
，k从1开始取值，选取均值方法作为聚合函数，记为agg
k
，设置图中的节点采样函数，记为n，其中，聚合和采样的操作上限设置为二阶，即k＝2；构建图神经网络模型，图中所有节点的输入特征，记为h
v0
，首先利用采样函数n分别采样节点h
v0
的一阶邻居节点、二阶邻居节点，分别记为h
v1
、h
v2
，然后利用聚合函数agg
k
和权重矩阵w
k
分别对二阶邻居节点h
v2
和一阶邻居节点h
v1
先后进行聚合和加权，得到节点的特征向量h
vk
，最后将得到的节点向量进行二范数归一化；将节点特征向量h
vk
输入维度为(7，10)的全连接层进行分类；读取训练数据并批次载入模型进行训练，采用交叉熵损失对模型预测的分类结果做损失计算；采用adam优化器作为模型，初始学习率为0.004，采用阶梯下降的学习率衰减方法，batch size定义为32个房间，总共训练100轮次。7.根据权利要求1所述的一种基于图神经网络的结构化户型解析方法，其特征在于，所述根据目标模型对待解析的户型信息进行解析，得到房间分类结果与户型结构数据后进行可视化展示，包括：读取经过后处理的房间分类结果；根据测试数据索引，从获取的结构化户型数据集合中读取测试数据的户型结构信息；遍历整个户型的所有房间，按照房间索引，对应地绘制出房间二维结构图并标注房间类型；展示带有房间类型的平面户型图。8.一种基于图神经网络的结构化户型解析系统，其特征在于，包括：第一模块，用于根据户型数据库获取所有户型结构信息，并对所述户型结构信息进行结构化处理，得到结构化的户型结构信息；第二模块，用于根据结构化的户型结构信息构建户型连通气泡图并进行特征工程，获取气泡图中所有节点的特征项和类别标签作为训练数据；第三模块，用于搭建基于图神经网络模型的分类模型，输入训练数据进行训练，保存最优模型权重；第四模块，用于根据最优模型权重对测试户型数据进行预测，计算分类精度，进而根据分类精度构建得到目标模型；第五模块，用于根据目标模型对待解析的户型信息进行解析，得到房间分类结果与户型结构数据后进行可视化展示。
9.一种电子设备，其特征在于，包括处理器以及存储器；所述存储器用于存储程序；所述处理器执行所述程序实现如权利要求1至7中任一项所述的方法。10.一种计算机可读存储介质，其特征在于，所述存储介质存储有程序，所述程序被处理器执行实现如权利要求1至7中任一项所述的方法。

技术总结
本发明公开了一种基于图神经网络的结构化户型解析方法和系统，方法包括：根据户型数据库获取所有户型结构信息，并对所述户型结构信息进行结构化处理，得到结构化的户型结构信息；根据结构化的户型结构信息构建户型连通气泡图并进行特征工程，获取气泡图中所有节点的特征项和类别标签作为训练数据；搭建基于图神经网络模型的分类模型，输入训练数据进行训练，保存最优模型权重；根据最优模型权重对测试户型数据进行预测，计算分类精度，进而根据分类精度构建得到目标模型；根据目标模型对待解析的户型信息进行解析，得到房间分类结果与户型结构数据后进行可视化展示。本发明的效率高且准确率高，可广泛应用于计算机技术领域。可广泛应用于计算机技术领域。可广泛应用于计算机技术领域。


技术研发人员：王兵 柯建生 戴振军 陈学斌
受保护的技术使用者：广州极点三维信息科技有限公司
技术研发日：2023.04.11
技术公布日：2023/8/13