[发明专利]楼宇场景的建模方法和装置

说明书

本发明实施例提供一种楼宇场景的建模方法和装置，该方法包括：获取目标楼宇场景中每个楼层的楼宇结构图栅格数据；根据预先训练得到的实例分割模型对所述楼宇结构图栅格数据进行实例分割，得到对所述楼宇结构图的实例分割结果；将实例分割结果与所述楼宇结构图的楼宇结构矢量数据进行配准，得到配准的楼宇结构矢量数据图；最后通过楼层模型库自动适配与所述楼宇结构矢量数据图匹配的楼宇模型；基于三维建模工具对所述匹配到的楼宇模型建模，完成楼宇场景的在线快速建模，本实施例提供的方法对场景的适配度比对比方法具有较大的弹性空间且成本低、建模效率高。

技术领域

本发明实施例涉及三维建模技术领域，尤其涉及一种楼宇场景的建模方法和装置。

背景技术

随着现代城市的飞速发展，越来越多的高楼建筑拔地而起。相关人员需要在建筑建设之前先建立三维的楼宇模型，比如三维楼宇外观模型、楼宇场景中的每个楼层内的三维户型模型等，以便于在建设过程中参考三维模型。并且，建筑建成之后，还需要向客户展示楼宇建筑的每层楼内的三维户型模型，以便于客户全方位的了解该建筑。

相关技术中，楼宇场景建模的方法主要分为人工和自动化建模两种。其中，自动化建模通过图像获取对空间的整体理解，然后，在采用超文本标记语言HTML编辑浏览器页面中的HTML5画布上，采用以3D编程工具ThreeJS的形式封装的3D绘图协议WebGL显示所建立的三维立体房产模型，然后采用脚本语言JavaScript的方式来控制三维立体房产模型的显示效果。

但是，现有两种建模方法中，人工建模虽然具有较好的建模质量，但是建模周期长、成本高；自动化建模通过传统的ThreeJS在前端页面建模，不仅效率低下，且对操作的人员的专业要求度较高，需要消耗较大的学习成本。

发明内容

本发明实施例提供一种楼宇场景的建模方法和装置，以克服现有技术中自动化建模效率低且消耗成本高的术问题。

第一方面，本发明实施例提供一种楼宇场景的建模方法，包括：

获取目标楼宇场景中每个楼层的楼宇结构图栅格数据；

根据预先训练得到的实例分割模型，对所述楼宇结构图栅格数据进行实例分割，以得到对所述楼宇结构图的实例分割结果；

将所述实例分割结果与所述楼宇结构图的楼宇结构矢量数据进行配准，得到配准的楼宇结构矢量数据图；

根据所述楼宇结构矢量数据图，在楼层模型库中匹配相应的楼宇模型；

基于三维建模工具对所述匹配到的楼宇模型建模，得到所述目标楼宇场景的三维模型。

可选的，所述方法还包括：

获取训练样本集，所述训练样本集包括多个楼层的楼宇结构图栅格数据样本，所述楼宇结构图栅格数据样本是标注有楼层的楼宇结构矢量数据和经过掩膜制作的楼宇结构图；

基于实例分割算法Mask Scoring R-CNN对所述楼宇结构图栅格数据样本进行训练，得到所述实例分割模型。

可选的，所述基于实例分割算法Mask Scoring R-CNN对所述楼宇结构图栅格数据样本进行训练，包括：

将所述楼宇结构图栅格数据样本输入神经网络ResNet-101-FPN中，得到所述神经网络ResNet-101-FPN输出的所述楼宇结构图对应的楼宇结构特征图；

在所述楼宇结构特征图中划分处多个候选兴趣区域；

将所述多个候选兴趣区域输入区域建议网络RPN，所述区域建议网络RPN对所述候选兴趣区域进行二值分类和边框回归操作，并通过非极大值抑制算法对所述多个候选兴趣区域进行筛选，得到目标兴趣区域；

对所述目标兴趣区域进行区域特征聚集操作；