[发明专利]一种装修方案生成方法、装置、电子设备及存储介质

权利要求书

1.一种装修方案生成方法，其特征在于，包括：

将获取的待装修图纸输入至预先训练的特征提取模型，得到所述待装修图纸的各房间的特征数据；其中，所述特征提取模型包括第一特征提取模型和至少一个第二特征提取模型；所述特征数据具体通过如下方式得到：将所述待装修图纸输入至所述第一特征提取模型，得到所述待装修图纸的各房间的房间类型标签信息和第一坐标信息；将所述房间类型标签信息、第一坐标信息和所述房间类型标签信息对应的待装修图纸输入至所述房间类型标签信息对应的第二特征提取模型，得到所述待装修图纸的各房间的特征数据；所述特征数据包括所述房间中部件的标签信息、第二坐标信息和尺寸信息；

将所述特征数据转换为所述待装修图纸的各房间的中间序列数据，其中，所述中间序列数据包括所述各房间的标签词向量和空间特征信息；所述中间序列数据具体通过如下方式得到：将所述特征数据中的标签信息输入至预先训练好的词向量模型，得到所述待装修图纸的各房间的标签词向量；基于所述标签词向量与所述特征数据中的第二坐标信息和尺寸信息，得到所述待装修图纸的各房间的初始序列数据；按照特定拼接顺序将所述初始序列数据进行拼接，得到所述各房间的中间序列数据；

分别计算所述中间序列数据与预先存储的至少一个样本序列数据之间的序列数据距离；所述序列数据距离具体通过如下方式得到：基于所述中间序列数据中任一序列的空间特征信息和所述样本序列数据中任一序列的空间特征信息，确定所述中间序列数据和所述样本序列数据的两序列间转移量和两序列中标签向量距离；基于所述两序列间转移量和两序列中标签向量距离，确定所述中间序列数据与所述样本序列数据之间的序列数据距离，其中，所述序列数据距离为两序列间转移量和两序列中标签向量距离的乘积和的最小值；

根据所述序列数据距离从所述样本序列数据中确定目标序列数据，并将所述目标序列数据对应的布局信息确定为所述待装修图纸的装修方案。

2.根据权利要求1所述的方法，其特征在于，所述根据所述序列数据距离从所述样本序列数据中确定目标序列数据，包括：

将小于预设阈值的序列数据距离对应的样本序列数据确定为所述目标序列数据；

或者，

基于计算得到的所述序列数据距离对所述样本序列数据进行排序；

将预设排序位置的序列数据距离对应的样本序列数据确定为所述目标序列数据。

3.根据权利要求1所述的方法，其特征在于，在所述将所述目标序列数据对应的布局信息确定为所述待装修图纸的装修方案之前，还包括：

获取所述目标序列数据的标签词向量；

对所述目标序列数据的标签词向量进行解析，根据解析结果确定所述目标序列数据对应的排布信息。

4.根据权利要求1所述的方法，其特征在于，所述方法还包括：

根据所述特征提取模型的样本装修图纸的各房间的空间特征信息，对所述样本装修图纸进行区域分割，得到至少一个分割区域；

以各所述分割区域的设定的种子点开始，遍历各所述分割区域的像素点对应的像素值，根据遍历结果对所述分割区域进行颜色填充；

根据颜色填充后的样本装修图纸对初始模型进行训练，得到所述特征提取模型。