一种自动生成矢量化户型平面图的方法

本发明公开了一种自动生成矢量化户型平面图的方法，包括以下步骤：步骤1、采用节点生成模型生成表示户型平面图中墙交叉点的节点集合，每个节点包含其位置坐标、周围房间的语义标签，同时通过新颖的损失函数项提高了生成节点集合的几何质量；步骤2、以步骤1得到的节点集合中的节点组合形成的侯选边集合作为输入，预测得到边缘集合，同时通过新颖的随机自监督几何增强方法提高了预测边缘集合的几何质量；步骤3、基于和角点关联的房间的语义信息，通过遍历图的最小环操作得到所有的最小环即为所有的房间。本发明减少繁琐的人工设计，智能化程度高。