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

摘要

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