[发明专利]基于二维图像生成房间结构模型的方法及装置

摘要

本发明实施例公开了一种基于二维图像生成房间结构模型的方法及装置，其中，获取房间的单张二维图像；将所述二维图像输入预先训练的三维图像神经网络，经所述三维图像神经网络输出所述二维图像对应的深度信息及结构分割信息、物品分割信息；获取所述深度信息对应点云数据；根据所述结构分割信息、物品分割信息，去除所述点云数据中的物品信息，保留结构信息；获取所述结构信息中的水平面信息和高度信息；根据所述水平面信息和高度信息对所述房间建立所述二维图像对应的三维房间结构模型。本发明方案将房间二维图像自动转换为三维模型，通过简单的普通相机采集房间图像，就能够推断出该房间的一个三维模型，能够满足快速简洁的全屋重建的需求。