本技术公开了一种基于图的三维扫描室内平面图重建方法,其特征在于,包括以下步骤:(1)获取开源数据集;(2)构建基元估计网络SeqPNet得到墙点热图;基元估计网络SeqPNet包括:墙体热图回归层、房间热图回归层及点热图回归层;依次生成三种类型的热图墙B、房间R和点P掩码得到墙点热图;(3)对墙点热图进行三角化、三角剪枝处理,得到初始图;(4)根据初始图求解最优子图,得到最终的重建结构;(5)评估重建结果;本发明有效地捕捉了细节,并保持了平面图的整体完整性。
背景技术
目前平面重建方法有两种:第一种使用自底向上的基元检测方法,直接从输入中提取角位置,并通过组合、预测边缘关系获得一组线段作为平面结构表示。这种方法通常采用端到端式网络进行推理,具有较快的推理速度,然而结果受到第一阶段原始检测精度的高度影响,并且缺乏全局结构信息导致结果完整性低(存在不闭合的房间)。第二种方法通常是自上而下的采用启发式房间优化方法,基于图像分割提取房间蒙版并启发式优化房间的轮廓,最终组合为总体结构。然而,这种启发式方法计算时间成本过高,并且结果离CAD式平面表示有一定差距。
实现思路