一种基于纯注意力网络的房间家具自动布局的方法技术

本发明专利技术公开了一种基于纯注意力网络的房间家具自动布局的方法，包括以下步骤：步骤1：获取与处理训练样本的数据；步骤2：对家具与房间分别进行embedding处理；步骤3：对家具布局的顺序进行位置postion embedding步骤4：使用房间信息与物品信息与已布局物品信息构建为编码器；步骤5：构建解码器；步骤6：输出目标物品位置方向。该基于纯注意力网络的房间家具自动布局的方法能够根据输入的户型信息，输入的待布局家具信息，快速将家具布局在指定户型中。本发明专利技术将深度学习在自然语言处理方面的使用技术应用到房间自动布局设计，可以节省大量时间与成功成本，提升生产效率。提升生产效率。提升生产效率。

【技术实现步骤摘要】
一种基于纯注意力网络的房间家具自动布局的方法


[0001]本专利技术属于家装设计
，具体为一种基于纯注意力网络的房间家具自动布局的方法。

技术介绍

[0002]家庭装修领域也就是家装领域的需求越来越高、需求量越来越大，是目前需求缺口最大的领域之一，业主装修自己的房子时，往往会去家装公司找设计师先设计自己的房子，现有家装设计中，主要通过设计师通过CAD等软件结合经验知识设计户主房间家具摆放，消耗大量时间与人工，并且现有的自动套用设计的网站上效果往往比较差。

技术实现思路

[0003]针对现有技术存在的上述问题，本专利技术所要解决的技术问题在于提供一种基于纯注意力网络的房间家具自动布局的方法，以解决目前的在家装设计中设计费时费力的问题。
[0004]为了解决上述技术问题，本专利技术采用的技术方案为：
[0005]一种基于纯注意力网络的房间家具自动布局的方法，包括以下步骤：
[0006]步骤1：获取与处理训练样本的数据；
[0007]步骤2：对家具与房间分别进行embedding处理；
[0008]步骤3：对家具布局的顺序进行位置embedding(postion embedding)；
[0009]步骤4：使用房间信息与物品信息与已布局物品信息构建为编码器；
[0010]步骤5：构建解码器；
[0011]步骤6：输出目标物品位置方向。
[0012]进一步的，所述步骤1中待布局家具的最多家具个数作为padding的长度。
[0013]进一步的，所述步骤2中通过用指定的数字编码房间信息，家具信息，表征家具、门窗墙的位置，与家具的方向，然后接入embedding层embedding的size设置为5。
[0014]进一步的，所述步骤3中postion embedding方法对待布局的物品进行先后顺序区分的方法为：
[0015][0016][0017]进一步的，pos指的是这个家具在整体待布局家具中的位置，2i指的是家具embedding信息的偶数维度，2i+1指的是家具embedding信息的奇数维度。
[0018]进一步的，所述步骤4中使用的多头注意力机制中的head的个数为4，编码器的个数为2。
[0019]进一步的，所述步骤5中使用的已布局家具信息与步骤3共用的embedding层。
[0020]有益效果：与现有技术相比，本申请具有以下优势：
[0021]该基于纯注意力网络的房间家具自动布局的方法能够根据输入的户型信息，输入的待布局家具信息，快速将家具布局在指定户型中。本专利技术将深度学习在自然语言处理方面的使用技术应用到房间自动布局设计，可以节省大量时间与成功成本，提升生产效率。
附图说明
[0022]图1是基于纯注意力网络的房间家具自动布局的方法数据输入示意图；
[0023]图2是网络框架结构图。
具体实施方式
[0024]下面结合附图对本专利技术做进一步的说明。
[0025]如图1、图2所示，本申请的基于纯注意力网络的房间家具自动布局的方法，包括以下步骤；
[0026]步骤1：获取与处理训练样本的数据，标注户型图中的门窗墙(房间信息)与家具方向位置(家具信息)，按照指定的家具顺序处理布局家具，padding成统一长度的序列数据；
[0027]步骤2：对家具与房间分别进行embedding处理，对家具与房间信息的编码进行embedding；
[0028]步骤3：对家具布局的顺序进行位置embeddin(postion embedding)，使用位置编码区分布局家具的先后顺序；
[0029]步骤4：使用房间信息与物品信息与已布局物品信息构建为编码器，房间信息的embedding与物品信息的embedding以及已布局物品的上下文信息embedding信息concate，然后加上多头自注意力层(multiheadself attention)、残差层、层归一化(norm)，与前馈网络层(FeedForward)构成编码器，该编码器的结构见网络结构图中左侧部分，该部分用于捕捉所有待布局物品的信息与当前布局物品的房间内上下文信息之间的相互关系(注意力)，设置的超参为编码器的个数：2，注意力头的个数为：4，num_units为：256；
[0030]步骤5：构建解码器，已布局物品的实际位置与方向，与待布局物品的信息(长宽默认方向等)经过embedding，concate计算自注意力，编解码器之间的注意力，残差，层归一化与前馈网络层，构成解码器，该编码器的结构见网络结构图中右侧部分，该部分用于捕捉所有已布局物品的位置与编码器之间的关系(注意力)，设置的超参为编码器的个数：2，注意力头的个数为：4，num_units为：256；
[0031]步骤6：输出目标物品位置方向，解码器特征输出后接入全连接层输出预测物品的位置与方向，位置为具体房间中的该物品中心点所在的位置，方向为相对于每个物品默认方向旋转0
°
，90
°
，180
°
，270
°
四个角度的方向。
[0032]步骤1中待布局家具的最多家具个数作为padding的长度。
[0033]步骤2中通过用指定的数字编码房间信息，家具信息，表征家具、门窗墙的位置，与家具的方向，然后接入embedding层embedding的size设置为5，如家具类型编码，家具的长宽信息编码，房间的门窗墙等信息进行编码，并对编码的信息进行embedding操作，该过程中使用的超参：embedding word size为5000，embedding size为5。
[0034]步骤3中postion embedding方法对待布局的物品进行先后顺序区分的方法为：
[0035][0036][0037]pos指的是这个家具在整体待布局家具中的位置，2i指的是家具embedding信息的偶数维度，2i+1指的是家具embedding信息的奇数维度。
[0038]步骤4中使用的多头注意力机制中的head的个数为4，编码器的个数为2。
[0039]步骤5中使用的已布局家具信息与步骤3共用的embedding层。
[0040]该模型在卧室中的预测效果表：
[0041]房间Top1_accTop4_accTop8_acc卧室0.6600.9350.971
[0042]本专利技术提供了一种基于纯注意力网络的房间家具自动布局的方法的思路及实施方法，具体应用途径很多，以上仅是本专利技术的优选实施方式，应当指出，对于本
的普通技术人员来说，在不脱离本专利技术原理的前提下，还可以做出若干改进，这些改进也应视为本专利技术的保护范围。
本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种基于纯注意力网络的房间家具自动布局的方法，其特征在于，包括以下步骤：步骤1：获取与处理训练样本的数据；步骤2：对家具与房间分别进行embedding处理；步骤3：对家具布局的顺序进行位置embedding、postion embedding；步骤4：使用房间信息与物品信息与已布局物品信息构建为编码器；步骤5：构建解码器；步骤6：输出目标物品位置方向。2.根据权利要求1所述的基于纯注意力网络的房间家具自动布局的方法，其特征在于：所述步骤1中待布局家具的最多家具个数作为padding的长度。3.根据权利要求1所述的基于纯注意力网络的房间家具自动布局的方法，其特征在于：所述步骤2中通过用指定的数字编码房间信息，家具信息，表征家具、门窗墙的位置，与家具的方向，然后接入embedding层embedding的size设置为5。4.根据权利要求...

【专利技术属性】
技术研发人员：陈旋，吕成云，朱召文，
申请(专利权)人：江苏艾佳家居用品有限公司，
类型：发明
国别省市：