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

权利要求书

1.一种三维户型模型生成方法，其特征在于，所述三维户型模型生成方法应用于三维户型模型生成设备，所述三维户型模型生成方法包括以下步骤：

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

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

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

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

2.如权利要求1所述的三维户型模型生成方法，其特征在于，所述对所述目标户型图进行卷积和池化，获取对应的墙体特征图的步骤包括：

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

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

3.如权利要求1所述的三维户型模型生成方法，其特征在于，所述通过转置卷积，基于所述墙体特征图生成目标墙体图的步骤包括：

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

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

4.如权利要求1所述的三维户型模型生成方法，其特征在于，所述当接收到二维户型图时，对所述二维户型图进行预处理，得到所述二维户型图对应的目标户型图的步骤包括：

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

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

5.如权利要求4所述的三维户型模型生成方法，其特征在于，所述根据所述放缩倍数，对所述二维户型图进行放缩处理，得到所述二维户型图对应的目标户型图的步骤包括：

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

根据所述宽高比和所述放缩倍数，对所述二维户型图进行放缩处理，得到放缩处理后的二维户型图；

根据所述预设图片大小，对放缩处理后的二维户型图对应的空白边缘进行填充，得到目标户型图。

6.如权利要求4所述的三维户型模型生成方法，其特征在于，所述获取所述目标墙体图对应的三维墙体，并基于所述三维墙体将所述二维户型图生成三维户型模型的步骤包括：

基于所述放缩倍数，对所述目标墙体图进行放缩处理，得到放缩处理后的目标墙体图；

获取所述放缩后的目标墙体图中的墙体；

基于所述二维户型图对应的标注信息，将所述墙体拉伸为三维墙体，生成三维户型模型。

7.如权利要求1至6任意一项所述的三维户型模型生成方法，其特征在于，所述三维户型模型生成方法还包括：

将所述三维户型模型显示于所述三维户型模型生成设备对应的显示界面；

当接收到基于所述显示界面的放缩指令时，根据所述放缩指令，对所述三维户型模型进行放缩。

8.一种三维户型模型生成装置，其特征在于，所述三维户型模型生成装置包括：

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

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

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

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

9.一种三维户型模型生成设备，其特征在于，所述三维户型模型生成设备包括处理器、存储器、以及存储在所述存储器上并可被所述处理器执行的三维户型模型生成程序，其中所述三维户型模型生成程序被所述处理器执行时，实现如权利要求1至7中任一项所述的三维户型模型生成方法的步骤。

10.一种计算机可读存储介质，其特征在于，所述计算机可读存储介质上存储有三维户型模型生成程序，其中所述三维户型模型生成程序被处理器执行时，实现如权利要求1至7中任一项所述的三维户型模型生成方法的步骤。