[发明专利]基于深度对抗网络模型的室内功能区划分方法

说明书

本发明公开一种基于深度对抗网络模型的室内功能区划分方法。本发明首先把一个空的房间根据具体使用划分成若干矩形的功能区域。进一步，对传统的生成式条件对抗网络进行改进，并应用实际工程数据集进行训练用以学习这些功能区之间的相对关系，以及功能区位置大小和过道，门窗，户型形状的内在联系，获得生成式功能区分割模型。最后，在房间功能区域划分基础之上，选择各自区域的一个主家具，用层次关联结构对家具集群在各自区域内部进行位置，朝向的描述。最终完成室内家具的自动布局，本发明的布局有效的利用了空间且美观，并且能够满足人们使用、通行等方面的要求。

技术领域

本发明属于计算机视觉图像处理领域，具体涉及一种基于深度对抗网络（DCGAN）模型对室内空间进行功能区域划分的方法。

背景技术

室内场景的布局及合成是当前计算机图形学、虚拟现实等领域研究的重要问题之一，随着数字娱乐、室内导航以及室内设计等方面需求的不断提高，其中，自动、智能的家具布局方法越来越受到研究和应用的重视。

近年来，家具自动布局问题的大致方法可以归为三类，一类是基于规则的方法（Rule based），通过在一个户型中对不同的物件，定义一系列不同的位置约束，如物件不能穿越墙体；过道上不能备物件阻挡等，进而通过求解约束的最优化问题来确定物体出现的位置。其二是基于草图的方法（Sketch based），从一个草图，或者一个粗略的3D扫描样版，丰富完善形成一个完善的3D场景。另一类是基于数据驱动的方法，或者基于学习的方法，该类方法从海量的数据中学习到一种层次的概率模型，根据每一物件出现在空间位置的概率确定布局位置。本发明方法属于第三类方法范畴。

发明内容

本发明针对当前计算机辅助室内设计方面，智能化不足提出了一种基于深度对抗网络对室内空间进行功能区划分的方法。该方法能够对一个空户型，进行最合理的功能区划分。比如在客厅合理的位置划分出，用餐区，视听区，休闲区。

本发明包括以下步骤：

第一步：数据集制作

从家具布局完善的户型图中提取卧室，客厅，书房布局图，并对这些房间标注门窗；按照家具功能的类似程度和集中情况，圈出功能区；这些处理过的布局图，被用作模型的训练，所述的功能区是指一个户型中的家具按照其功能属性集中成群出现在户型中的若干个区域。

第二步：把数据集输入DCGAN神经网络模型进行训练，并对模型进行参数调整和结构改善；进过训练、参数调整和结构改善后的DCGAN神经网络模型，其输入为空的户型图；输出即为功能区划分完毕的户型图。

所述的DCGAN神经网络模型经过了以下的迭代训练：

S1.随机噪声联合空户型信息输入DCGAN神经网络模型中的生成器，生成器产生一组初代假样本。

S2.使用第一步得到的数据集，训练DCGAN神经网络模型中的判别器，判别器返回一个概率阈值参数，用于反映此轮生成器产生样本接近真实样本的程度。

S3.利用判别网络的返回的概率阈值参数指导生成器调整自身模型参数，生成一组比上次迭代更为标准的样本。

S4.返回S2，迭代调整到判别器无法判断生成器生成样本是否为真实样本时停止迭代。

本发明的有益效果：本发明创造性的使用深度对抗网络来解决户型区域划分这一类问题。这在以往的工作中是没有过的。具体对比背景技术提到的三个方面说明其有益效果。一，相比传统的基于规则的方法而言，本发明不用指定复杂的规则，时间代价最小。二，相比于基于样例的方法，本发明方法更为灵活，能够适应各种复杂多变的户型，三，相对于一般的数据驱动方法，本发明中的模型更加容易训练——传统的深度学习达到相同的效果需要数万个样本进行训练，而本发明中的模型是一种半监督学习，只使用400例样本训练就达到了很好的效果，稳定性也更好。

附图说明