arXivDaily arXiv每日学术速递 周一至周五更新
arXiv 2008.06286cs.CV

GeoLayout: Geometry Driven Room Layout Estimation Based on Depth Maps of Planes

School of Control Science and Engineering, Shandong University(山东大学控制科学与工程学院);Google Research(谷歌研究院)

Weidong Zhang, Wei Zhang, Yinda Zhang

英文摘要:

The task of room layout estimation is to locate the wall-floor, wall-ceiling, and wall-wall boundaries. Most recent methods solve this problem based on edge/keypoint detection or semantic segmentation. However, these approaches have shown limited attention on the geometry of the dominant planes and the intersection between them, which has significant impact on room layout. In this work, we propose to incorporate geometric reasoning to deep learning for layout estimation. Our approach learns to infer the depth maps of the dominant planes in the scene by predicting the pixel-level surface parameters, and the layout can be generated by the intersection of the depth maps. Moreover, we present a new dataset with pixel-level depth annotation of dominant planes. It is larger than the existing datasets and contains both cuboid and non-cuboid rooms. Experimental results show that our approach produces considerable performance gains on both 2D and 3D datasets.

机构、总结、中文摘要均由AI分析而来,可能存在错误,请以论文正文为准。

↑