1. Undirected graph representing strategy for general room layout estimation.
- Author
-
Yao, Hui, Miao, Jun, Zheng, Yilin, Zhang, Guoxiang, and Chu, Jun
- Subjects
- *
IMAGE processing , *IMAGE representation , *SIGNAL processing , *ALGORITHMS , *VISUAL communication - Abstract
• A layout plane is a closed loop surrounded by layout keypoints. • The wall structure can be parametrization as an undirected graph. • Layout keypoint prediction is relative to human pose estimation. • Cuboid room assumption is limited in real life. Room layout estimation aims to predict the spatial structure of a room from a single image. Most existing methods relying on 2D cues are suitable for cuboid rooms, but non-cuboid rooms. 3D layout estimation methods can reconstruct 3D models of general rooms, but they are trained with depth information on high collection costs, consume large computation resources, and run slowly. This paper considers an undirected graph representation method of the general room layouts, which includes cuboid and non-cuboid rooms consisting of a single ceiling, a single floor, and multiple walls. To this end, we first predict the positions of the layout vertices and then use the network automatically learn the connection relationship between the vertices. The final layout is obtained through a simple layout inference post-processing algorithm. The experimental results both on cuboid and non-cuboid datasets validate the effectiveness and efficiency of our method. The code is available at https://github.com/Hui-Yao/2D-graph-layout-estimation. [ABSTRACT FROM AUTHOR]
- Published
- 2023
- Full Text
- View/download PDF