技术摘要：
本发明实施例公开了一种家装设计方法、装置、电子设备及存储介质，该方法包括：获取待设计户型的房间信息；将所述房间信息输入至预先训练好的transformer网络模型，得到针对所述待设计户型的家装设计信息；其中，所述家装设计信息包括至少一种家具的类别信息以及所述家  全部
背景技术：
随着房地产行业的迅速发展，家装设计的市场需求也迅速增大，而传统的家装设 计行业是一个专业且耗时的行业，越来越无法满足市场对快速出图的需求。针对该市场需 求，越来越多的企业加大了对智能家装设计的研发与投入力度。 目前，在智能家装设计领域，主要还是运用基于规则的优化技术、先验概率、模板 匹配或者循环网络等技术。目前的智能家装设计技术普遍存在如下问题：无法满足家装设 计多样性的需求以及无法实现家具的自动排布。
技术实现要素：
本发明实施例提供一种家装设计方法、装置、电子设备及存储介质，实现了家装的 智能设计，且进一步实现了对家具进行自动排布的目的，提高了家装设计的智能化程度。 第一方面，本发明实施例提供了一种家装设计方法，该方法包括： 获取待设计户型的房间信息； 将所述房间信息输入至预先训练好的transformer网络模型，得到针对所述待设 计户型的家装设计信息； 其中，所述家装设计信息包括至少一种家具的类别信息以及所述家具的摆放位置 信息。 transformer网络具有较强的特征抽取能力，通过将transformer网络应用到家装 设计中，提高了家装设计的设计效率与效果，以及实现了对家具的自动排布，提高了家装设 计的智能化程度，地面家具的自动排布速度可达到秒级。 相比于传统的需要将输入编码成固定长度的seq2seq网络，transformer网络不会 造成数据特征的严重丢失，进而提高了家装设计的效果。 进一步的，所述将所述房间信息输入至预先训练好的transformer网络模型之前， 所述方法还包括： 基于所述待设计户型的房间信息，确定所述待设计户型的类别属性。 对应的，所述将所述房间信息输入至预先训练好的transformer网络模型，得到针 对所述待设计户型的家装设计信息，包括： 将所述房间信息输入至预先训练好的、与所述类别属性对应的transformer网络 模型； 其中，不同类别属性对应的transformer网络模型基于不同类别属性的户型的家 装设计信息训练得到。 所述类别属性包括：老人房间、儿童房间、青年人房间、书房、厨房、卫生间或者客 4 CN 111553012 A 说　明　书 2/12 页 厅； 所述家具的类别信息包括下述至少一种：床、书桌、衣柜、梳妆台、茶几、以及电视 柜； 所述房间信息包括下述至少一种：房门以及房门的位置信息、墙体以及墙体的位 置信息、飘窗以及飘窗的位置信息、窗户以及窗户的位置信息。 通过针对不同类别属性的房间分别建模训练，在对不同类别属性的房间进行家装 设计时，使用不同类别属性对应的网络模型，进一步提高了家装设计的设计效果。例如老人 房间的家装设计与儿童房间的家装设计存在较大差异，通过分别针对老人房间的家装设计 建立模型并预先训练，针对儿童房间的家装设计建立模型并预先训练。在需要对待设计的 儿童房间进行家装设计时，利用预先训练好的、适用儿童房间的网络模型进行智能设计，不 仅提高了家装设计速度，还可极大地提高家装设计效果。 进一步的，所述预先训练好的transformer网络模型包括一个编码网络Encoder和 两个解码网络Decoder； 其中，所述编码网络Encoder用于接收所述待设计户型的房间信息，并对所述房间 信息进行编码，将编码后得到的向量信息分别输入至所述两个解码网络Decoder； 所述两个解码网络Decoder中的一个用于对所述向量信息中的坐标序列进行解 码，以获得所述家装设计信息中家具的摆放位置信息，另一个解码网络Decoder用于对所述 向量信息中的类别序列进行解码，以获得所述家装设计信息中家具的类别信息。 通过设计两个解码网络Decoder，可提高网络模型的解码能力，实现准确解码出家 具的位置坐标序列和家具的类别序列，进而提高整个网络模型的家装设计性能。 进一步的，所述方法还包括： 基于所述家装设计信息，生成所述待设计户型的家装设计图。 例如利用3D显示软件基于所述家装设计信息，生成所述待设计户型的3D家装设计 图，不仅方便用户观看，还可极大地提高用户体验。 进一步的，所述方法还包括： 所述获取待设计户型的房间信息，包括： 基于所述待设计户型的户型图，通过图像识别模型确定所述房间信息。 例如，卷积神经网络在图像分类方面表现性能优越，可基于卷积神经网络预先训 练获得图像识别模型，用以基于待设计户型的户型图确定其中的房门、墙体、窗户等硬件实 体，即不可摆放家具的位置，或者理解为获得可摆放家具的空间信息，既可提高确定效率， 又可获得较高的确定精度。 第二方面，本发明实施例还提供了一种家装设计装置，该装置包括： 获取模块，用于获取待设计户型的房间信息； 设计模块，用于将所述房间信息输入至预先训练好的transformer网络模型，得到 针对所述待设计户型的家装设计信息； 其中，所述家装设计信息包括至少一种家具的类别信息以及所述家具的摆放位置 信息。 第三方面，本发明实施例还提供了一种设备，所述设备包括： 一个或多个处理器； 5 CN 111553012 A 说　明　书 3/12 页 存储装置，用于存储一个或多个程序， 当所述一个或多个程序被所述一个或多个处理器执行，使得所述一个或多个处理 器实现如本发明任一实施例所述的家装设计方法。 第四方面，本发明实施例还提供了一种包含计算机可执行指令的存储介质，所述 计算机可执行指令在由计算机处理器执行时用于执行如本发明任一实施例所述的家装设 计方法。 本发明实施例的技术方案，通过获取待设计户型的房间信息；将所述房间信息输 入至预先训练好的transformer网络模型，得到针对所述待设计户型的家装设计信息；其 中，所述家装设计信息包括至少一种家具的类别信息以及所述家具的摆放位置信息的技术 手段，实现了家装的智能设计，且进一步实现了对家具进行自动排布的目的，提高了家装设 计的智能化程度。 附图说明 结合附图并参考以下