一种预测室内空间布局的方法、装置及电子设备制造方法及图纸

本申请实施例涉及图像处理技术领域，特别是涉及一种预测室内空间布局的方法、装置及电子设备。在预测室内空间布局的方法中，通过将获取的待预测室内空间的环境图像输入空间布局预测模型，得到待预测室内空间的目标分割图像，该目标分割图像包括用于表征待预测室内空间的空间布局的至少两个分割区域。其中，空间布局预测模型的损失函数包括分割损失函数、边界损失函数和角点损失函数中的至少两种，使得空间布局预测模型考虑了分割区域、边界和边界角点等特征来预测室内环境的空间布局，提高了预测室内空间布局的准确性。预测室内空间布局的准确性。预测室内空间布局的准确性。

【技术实现步骤摘要】
一种预测室内空间布局的方法、装置及电子设备


[0001]本申请实施例涉及图像处理
，特别是涉及一种预测室内空间布局的方法、装置及电子设备。

技术介绍

[0002]预测室内空间布局的基本目标是根据给定的室内空间的环境图像恢复出室内空间的布局结构。目前，用于预测室内空间布局的空间布局预测模型在各个分割区域的边界处会出现锯齿和突变的情况，导致空间布局预测模型所预测的室内空间布局在各个边界处的预测不准确。

技术实现思路

[0003]本申请实施例提供一种预测室内空间布局的方法、装置及电子设备，能够提高预测室内空间布局的准确性。
[0004]为解决上述技术问题，本申请实施例采用的一个技术方案是：提供一种预测室内空间布局的方法，包括：
[0005]获取待预测室内空间的环境图像；
[0006]将所述环境图像输入空间布局预测模型，以获取所述待预测室内空间的目标分割图像，所述目标分割图像包括至少两个分割区域，所述至少两个分割区域用于表征所述待预测室内空间的空间布局。
[0007]为解决上述技术问题，本申请实施例采用的另一个技术方案是：提供一种预测室内空间布局的装置，包括：
[0008]第一获取模块，用于获取待预测室内空间的环境图像；
[0009]第一输入模块，用于将所述环境图像输入空间布局预测模型，以获取所述待预测室内空间的目标分割图像，所述目标分割图像包括至少两个分割区域，所述至少两个分割区域用于表征所述待预测室内空间的空间布局。
[0010]为解决上述技术问题，本申请实施例采用的另一个技术方案是：提供一种电子设备，包括：
[0011]至少一个处理器；以及
[0012]与所述至少一个处理器通信连接的存储器；其中，所述存储器存储有可被所述至少一个处理器执行的指令，所述指令被所述至少一个处理器执行，以使所述至少一个处理器能够用于执行上述预测室内空间布局的方法。
[0013]为解决上述技术问题，本申请实施例采用的另一个技术方案是：提供一种计算机可读存储介质，所述计算机可读存储介质存储有计算机可执行指令，所述计算机可执行指令用于被处理器执行上述预测室内空间布局的方法。
[0014]本申请实施例提供的一种预测室内空间布局的方法、装置及电子设备，通过将获取的待预测室内空间的环境图像输入空间布局预测模型，得到待预测室内空间的目标分割
图像，该目标分割图像包括用于表征待预测室内空间的空间布局的至少两个分割区域。其中，空间布局预测模型的损失函数包括分割损失函数、边界损失函数和角点损失函数中的至少两种，使得空间布局预测模型考虑了分割区域、边界和边界角点等特征来预测室内环境的空间布局，能够准确预测各个分割区域的边界，并能够准确确定各个分割区域的角点，提高了预测室内空间布局的准确性。
附图说明
[0015]一个或多个实施例通过与之对应的附图中的图片进行示例性说明，这些示例性说明并不构成对实施例的限定，附图中具有相同参考数字标号的元件表示为类似的元件，除非有特别申明，附图中的图不构成比例限制。
[0016]图1是本申请实施例提供的一种电子设备的硬件结构示意图；
[0017]图2是本申请实施例提供的一种预测室内空间布局的方法的流程示意图；
[0018]图3是标准分割图像与标准边界图像的对应关系图；
[0019]图4是本申请实施例提供的一种预测室内空间布局的装置的结构示意图；
[0020]图5是本申请另一实施例提供的一种预测室内空间布局的装置的结构示意图；
[0021]图6是本申请又一实施例提供的一种预测室内空间布局的装置的结构示意图。
具体实施方式
[0022]为使本申请实施例的技术方案更加清楚，下面将结合本申请实施例中的附图，对本申请实施例中的技术方案进行清楚、完整的描述，显然，所描述的实施例是本申请一部分实施例，而不是全部的实施例。应当理解，此处所描述的具体实施例仅用以解释本申请，并不用于限定本申请。基于本申请中的实施例，本领域普通技术人员在没有做出创造性劳动前提下所获得的所有其他实施例，都属于本申请保护的范围。
[0023]需要说明的是，当一个元件被表述“连接”另一个元件，它可以是直接连接到另一个元件、或者其间可以存在一个或多个居中的元件。此外，下面所描述的本申请各个实施例中所涉及到的技术特征只要彼此之间未构成冲突就可以相互组合。
[0024]本申请提供了一种预测室内空间布局的方法及装置，该方法及装置应用于电子设备，从而使得该电子设备能够通过将所获取的待预测室内空间的环境图像输入空间布局预测模型，得到待预测室内空间的空间布局。其中，由于空间布局预测模型的损失函数包括分割损失函数、边界损失函数和角点损失函数中的至少两种，因此，通过空间布局预测模型能够准确预测各个分割区域的边界，并能够准确确定各个分割区域的角点，进而使得电子设备能够提高预测室内空间布局的准确性。
[0025]可以理解的是，在本申请中，电子设备可以为手机、平板电脑、PC等具备图像采集功能的设备。
[0026]下面，将通过具体实施例对本申请进行阐述。
[0027]请参阅图1，是本申请实施例提供的一种电子设备的硬件结构示意图，该电子设备包括：存储器10、处理器20、拍摄模组30和显示模组40，处理器20分别连接存储器10、拍摄模组30和显示模组40。本领域技术人员应当理解的是，图1所示的结构并不构成对电子设备的限定，电子设备可以包括比图示更多或者更少的部件，或者组合某些部件，或者拆分某些部
件，或者采用不同的部件布置。
[0028]其中，存储器10作为一种非易失性计算机可读存储介质，可用于存储非易失性软件程序、指令以及模块。
[0029]存储器10可以包括存储程序区和存储数据区。其中，存储程序区可存储操作系统、至少一个功能所需要的应用程序；存储数据区可存储根据电子设备的使用所创建的数据等，该存储数据区还能够存储预设的数据。
[0030]此外，存储器10可以包括高速随机存取存储器，还可以包括非易失性存储器，例如至少一个磁盘存储器件、闪存器件、或其他非易失性固态存储器件。
[0031]在一些实施例中，存储器10可选包括相对于处理器20远程设置的存储器，这些远程设置的存储器可以通过网络连接至电子设备。上述网络包括但不限于互联网、企业内部网、局域网、移动通信网及其组合。
[0032]处理器20则为电子设备的控制中心，能够利用各种接口和线路连接整个电子设备的各个部分，通过运行或执行存储在存储器10内的非易失性软件程序、指令和模块，以及调用存储在存储器10内的数据，执行电子设备的各种功能并处理数据，从而对电子设备进行整体控制，例如实现本申请任一实施例所述的预测室内空间布局的方法。
[0033]处理器20可以为一个或多个，图1中以一个处理器20为例。
[0034]处理器20与存储器10、拍摄模组30以及显示模组40可以通过总线或者其他方式连接，图1本文档来自技高网...

【技术保护点】

【技术特征摘要】
1.一种预测室内空间布局的方法，其特征在于，包括：获取待预测室内空间的环境图像；将所述环境图像输入空间布局预测模型，以获取所述待预测室内空间的目标分割图像，所述目标分割图像包括至少两个分割区域，所述至少两个分割区域用于表征所述待预测室内空间的空间布局。2.根据权利要求1所述的方法，其特征在于，在将所述环境图像输入空间布局预测模型的步骤之前，所述方法还包括：获取训练样本，所述训练样本包括环境样本图像以及所述环境样本图像对应的标准分割图像；将所述环境样本图像和所述标准分割图像输入预设神经网络模型进行训练，训练时使用损失函数计算损失代价；将所述损失代价满足预设代价条件时的预设神经网络模型确定为所述空间布局预测模型。3.根据权利要求2所述的方法，其特征在于，所述方法还包括：若所述损失代价不满足所述预设代价条件时，利用所述损失代价反向调整所述预设神经网络模型。4.根据权利要求2所述的方法，其特征在于，所述空间布局预测模型的损失函数由分割损失函数、边界损失函数和角点损失函数中至少两种加权求和得到。5.根据权利要求4所述的方法，其特征在于，所述环境样本图像输入所述预设神经网络模型后得到所述环境样本图像的预测分割图像；则，所述方法还包括：基于所述预测分割图像和所述标准分割图像，确定所述分割损失函数、所述边界损失函数和所述角点损失函数。6.根据权利要求5所述的方法，其特征在于，所述基于所述预测分割图像和所述标准分割图像，确定所述分割损失函数，包括：比较所述预测分割图像和所述标准分割图像对应像素点的灰度值；将所述预测分割图像中灰度值与所述标准分割图像中对应像素点的灰度值相同的像素点确定为所述预测分割图像中预测正确的像素点；统计所述预测正确的像素点的第一数量和所述预测分割图像的像素点总数；根据所述第一数量和所述预测分割图像的像素点总数，确定所述分割损失函数。7.根据权利要求5所述的方法，其特征在于，所述预测分割图像和所述标准分割图像均包括至少两个分割区域；则，所述基于所述预测分割图像和所述标准分割图像，确定所述边界损失函数，包括：根据所述预测分割图像，确定预测边界图像，所述预测边界图像用于表征所述预测分割图像中各个分割区域的边界；根据所述标准分割图像，确定标准边界图像，所述标准边界图像用于表征所述标准分割图像中各个分割区域的边界；基于所述预测边界图像和所述标准边界图像，确定所述边界损失函数。8.根据权利要求7所述的方法，其特征在于，所述基于所述预测边界图像和所述标准边
...

【专利技术属性】
技术研发人员：赵超，
申请(专利权)人：深圳绿米联创科技有限公司，
类型：发明
国别省市：