[发明专利]一种基于深度卷积网络的室内家居自动布局方法

说明书

本发明涉及一种基于深度卷积网络的室内家居自动布局方法，采用全新设计策略，应用创新设计的检测模型，引入像素点预测家具框模式，并加入家具框类别、对应家具框类别概率、以及相对该家具框的Centerness值的获得，通过与已布局各二维家具图的二维家居户型图进行比较，实现对所设计检测模型的训练，应用训练后的检测模型，针对实际目标户型图进行应用，实现目标户型图中各二维家具图的布局，帮助设计师进行家装设计，极大的节省了设计师手动布局的时间，提高工作效率。

技术领域

本发明涉及一种基于深度卷积网络的室内家居自动布局方法，属于智能家装设计技术领域。

背景技术

在家装设计中，布局的家具大小、位置和类别，都需要按实际户型图形状和大小来设计，如此家装设计师在手动布局过程中将耗费大量的时间，严重影响到工作效率。

发明内容

本发明所要解决的技术问题是提供一种基于深度卷积网络的室内家居自动布局方法，能够帮助设计师进行家装设计，极大的节省了设计师手动布局的时间，提高工作效率。

本发明为了解决上述技术问题采用以下技术方案：本发明设计了一种基于深度卷积网络的室内家居自动布局方法，以YOLOV3模型为基础，在其HEAD模块中加入FCOS模型中的Centerness分支，构成检测模型，并基于该检测模型执行如下步骤A至步骤E，实现目标户型室内家居自动布局；

步骤A.基于预设各二维家具图、以及各二维家具图所对应的尺寸标签信息，采集预设数量分别已布局各二维家具图的二维家居户型图、以及该各二维家居户型图分别所对应未布局二维家具图的二维户型图，构成样本库，然后进入步骤B；

步骤B.分别针对样本库中的各幅二维户型图，将二维户型图先经过检测模型里的DarkNet53网络提取出三个不同尺度的特征图，接着将小尺度的特征图向上取样，并与大尺度的特征图融合，从而构成FPN网络，由该FPN网络中提取出三个不同尺度大小的特征图，作为该二维户型图所对应不同尺度大小的各特征图，然后进入步骤C；

步骤C.分别针对样本库中的各幅二维户型图，采用检测模型，针对二维户型图所对应不同尺度大小的各特征图进行处理，获得二维户型图中各像素点分别预测对应的一个满足预设各二维家具图的家具框、该家具框类别、对应该家具框类别的概率、以及相对该家具框的Centerness值；进而通过相对该家具框的Centerness值、与对应该家具框类别的概率之间的乘积，构成该家具框的得分，即获得二维户型图中各预测家具框的得分，进一步删除二维户型图中得分低于预设分数阈值的各个预测家具框，更新该二维户型图中的各预测家具框，然后进入步骤D；

步骤D.根据样本库中各二维户型图中各预测家具框的得分，结合二维户型图所对应二维家居户型图中已布局的各二维家具图，针对检测模型进行训练，获得训练后的检测模型，然后进入步骤E；

步骤E.采用步骤C的方法针对目标户型进行处理，获得目标户型图中各预测家具框的得分，然后采用训练后的检测模型进行处理，获得目标户型图中各家具框、以及对应家具框类别，实现目标户型图中各二维家具图的布局。

作为本发明的一种优选技术方案：所述步骤C中，分别针对样本库中的各幅二维户型图，执行如下操作，获得二维户型图中各像素点分别预测对应的一个满足预设各二维家具图的家具框、该家具框类别、对应该家具框类别的概率、以及相对该家具框的Centerness值；

将二维户型图所对应不同尺度大小的各特征图、分别输入检测模型中的HEAD模块，其中，HEAD模块包括Regression支路和Classification支路，各支路分别包括预设层数、预设卷积核大小的各卷积层，且Regression支路中经过最后一个卷积层输出W*H*4的特征图，W为特征图宽，H为特征图高，即Regression支路输出特征图中各像素点分别预测对应的一个满足预设各二维家具图的家具框；