基于角点深度预测的室内场景3D布局估计方法和装置

本公开的实施例提供了基于角点深度预测的室内场景3D布局估计方法和装置，应用于室内场景的三维布局技术领域。所述方法包括获取目标室内场景图像；将目标室内场景图像输入深度学习模型，输出预测的角点热力图、角点深度值和嵌入向量；根据角点热力图生成角点坐标；对嵌入向量进行聚类得到粗略分割图，并使用粗略分割图的每个平面区域进行膨胀和腐蚀操作，得到目标区域；根据所述角点坐标与所述目标区域确定目标角点；根据所述目标角点的角点深度值进行平面拟合，得到每个平面对应的目标深度图；对目标深度图进行深度图相交计算，得到布局深度图。在不受因裁剪导致位置变动的影响下，得到更为准确的室内布局深度图。