[发明专利]一种基于条件卷积的建筑布局生成方法、装置及介质

权利要求书

1.一种基于条件卷积的建筑布局生成方法，其特征在于，包括以下步骤：

获取建筑布局图，对所述建筑布局图进行预处理，编码器根据预处理后的建筑布局图获取低维噪声；

将建筑边界图输入生成器，根据建筑边界图获取条件卷积核；

在生成器中，将低维噪声与条件卷积核进行条件卷积运算，直到生成的建筑布局设计图。

2.根据权利要求1所述的一种基于条件卷积的建筑布局生成方法，其特征在于，所述建筑布局图Ir为色块图，建筑布局图Ir中采用不同颜色来表示不同房间区域；

所述获取建筑布局图，对所述建筑布局图进行预处理，编码器根据预处理后的建筑布局图获取低维噪声z，包括：

将所述建筑布局图Ir按照预设比例进行伸缩，并进行张量化和归一化操作，获得预处理后的建筑布局图；

将预处理的建筑布局图输入编码器E，计算获得预处理建筑布局图的噪声均值mean和偏差var；

将噪声均值mean和偏差var进行重参数化，获得低维噪声z；

编码器E的输入输出公式如下：

z＝E(I_r)

式中，I_r为真实建筑布局图，z为噪声。

3.根据权利要求1所述的一种基于条件卷积的建筑布局生成方法，其特征在于，所述将建筑边界图输入生成器，根据建筑边界图获取条件卷积核，包括：

在生成器G中，采用下采样的降维卷积网络提取建筑边界图Ib的特征信息，获得不同尺度的特征图；

采用跳接层和上采样的二维卷积层，将建筑边界图Ib的特征图采样为条件卷积核K^cc；

其中，条件卷积核K^cc作为生成器G的调制信息。

4.根据权利要求1所述的一种基于条件卷积的建筑布局生成方法，其特征在于，所述将低维噪声与条件卷积核进行条件卷积运算，直到生成的建筑布局，包括：

采用线性层对低维噪声z进行预处理，以使低维噪声z的张量的形状符合图像数据类型；将预处理后的低维噪声z和条件卷积核K^cc进行条件卷积运算，经过上采样卷积层进行高宽的扩展，重复该过程预设次数，直到生成等同于建筑约束大小的建筑布局设计图Io。

5.根据权利要求4所述的一种基于条件卷积的建筑布局生成方法，其特征在于，所述将预处理后的低维噪声z和条件卷积核K^cc进行条件卷积运算，包括：

采用条件卷积模块，将预处理后的低维噪声z和条件卷积核K^cc进行条件卷积运算；

其中，条件卷积的计算公式如下：

在每一个条件卷积模块中，输入特征I通过卷积运算得到输出特征O，M为中间值，其中I∈R^C×H×W、O∈R^D×H×W；其中C为输入特征的通道数，D为输出特征的通道数，H和W分别是输入特征的高和宽；i和j分别表示输入特征I上的高和宽的坐标，数值范围分别在0至H和0至W之间；k为卷积核大小；K^cc为条件卷积核，K^p为点状卷积核；

生成器G的输入输出公式如下：

I_o＝G(z,I_b)

式中，I_o为输出图像，z为噪声，I_b为建筑边界图。