[发明专利]三维户型模型生成方法、装置、设备及存储介质

说明书

本发明涉及三维建模技术领域，提供一种三维户型模型生成方法、装置、设备及存储介质，所述方法包括以下步骤：当接收到二维户型图时，对所述二维户型图进行预处理，得到所述二维户型图对应的目标户型图；对所述目标户型图进行卷积和池化，获取对应的墙体特征图；通过转置卷积，基于所述墙体特征图，生成目标墙体图；基于所述二维户型图对应的标注信息对所述目标墙体图进行处理，得到所述目标墙体图对应的三维墙体，并基于所述三维墙体，将所述二维户型图生成三维户型模型。本发明通过转置卷积，将墙体特征图生成目标墙体图，进而获取对应的三维墙体，最后根据三维墙体生成三维户型模型，实现了二维户型图到三维户型模型的转换，方便用户了解户型。

技术领域

本发明涉及三维建模技术领域，尤其涉及一种三维户型模型生成方法、装置、设备及计算机可读存储介质。

背景技术

在房地产行业，客户在不前往现场查看的情况下，想要了解户型时，一般是通过销售人员发放的二维户型图纸，或者观看设计专家使用实物模具，靠人力搭建的三维户型模型。前者由于是在平面上，不能给客户带来空间上的直观感受；后者由于技术门槛高，因此产出小，缺少自动生成三维户型模型的手段。相似的专利中，如201710673386.6用户通过在图纸中进行点击，逐步确定户型中组件位置，这种方式还是需要人力参与的，明显还不够智能，而201510156080.4则是通过传统图形处理手段处理照片，虽然节省了人力，但并不能得到准确位置，因此没有在市场普及和应用。因此，目前急需一种既节省专家人力，又能快速准确生成户型三维模型的方法，进而普及到市场解决客户的痛点需求。

发明内容

本发明的主要目的在于提供一种三维户型模型生成方法、装置、设备及计算机可读存储介质，旨在解决现有三维户型模型建立比较复杂，而且建立速度慢的技术问题。

为实现上述目的，本发明提供一种三维户型模型生成方法，所述三维户型模型生成方法应用于三维户型模型生成设备，所述三维户型模型生成方法包括以下步骤：

当接收到二维户型图时，对所述二维户型图进行预处理，得到所述二维户型图对应的目标户型图；

对所述目标户型图进行卷积和池化，获取对应的墙体特征图；

通过转置卷积，基于所述墙体特征图生成目标墙体图；

基于所述二维户型图对应的标注信息，对所述目标墙体图进行处理，得到所述目标墙体图对应的三维墙体，并基于所述三维墙体将所述二维户型图生成三维户型模型。

可选地，所述对所述目标户型图进行卷积和池化，获取对应的墙体特征图的步骤包括：

将所述目标户型图输入预先训练好的神经网络模型中，基于所述神经网络模型，对所述目标户型图进行卷积和池化，得到待分类特征图；

根据预设分类器，获取所述待分类特征图中的墙体特征图。

可选地，所述通过转置卷积，基于所述墙体特征图生成目标墙体图的步骤包括：

获取所述墙体特征图在所述目标户型图中对应的位置；

根据所述位置，通过装置卷积，基于所述墙体特征图反向生成目标墙体图。

可选地，所述当接收到二维户型图时，对所述二维户型图进行预处理，得到所述二维户型图对应的目标户型图的步骤包括：

当接收到二维户型图时，基于预设图片大小，确定所述二维户型图的放缩倍数；

根据所述放缩倍数，对所述二维户型图进行放缩处理，得到所述二维户型图对应的目标户型图。

可选地，所述根据所述放缩倍数，对所述二维户型图进行放缩处理，得到所述二维户型图对应的目标户型图的步骤包括：

确定所述二维户型图的宽高比；