Siyuan Huang, Siyuan Qi, Yixin Zhu, Yinxue Xiao, Yuanlu Xu, Song-Chun Zhu
We propose a computational framework to jointly parse a single RGB image and reconstruct a holistic 3D configuration composed by a set of CAD models using a stochastic grammar model. Specifically, we introduce a Holistic Scene Grammar (HSG) to represent the 3D scene structure, which characterizes a joint distribution over the functional and geometric space of indoor scenes. The proposed HSG captures three essential and often latent dimensions of the indoor scenes: i) latent human context, describing the affordance and the functionality of a room arrangement, ii) geometric constraints over the scene configurations, and iii) physical constraints that guarantee physically plausible parsing and reconstruction. We solve this joint parsing and reconstruction problem in an analysis-by-synthesis fashion, seeking to minimize the differences between the input image and the rendered images generated by our 3D representation, over the space of depth, surface normal, and object segmentation map. The optimal configuration, represented by a parse graph, is inferred using Markov chain Monte Carlo (MCMC), which efficiently traverses through the non-differentiable solution space, jointly optimizing object localization, 3D layout, and hidden human context. Experimental results demonstrate that the proposed algorithm improves the generalization ability and significantly outperforms prior methods on 3D layout estimation, 3D object detection, and holistic scene understanding.
| Task | Dataset | Metric | Value | Model |
|---|---|---|---|---|
| Object Detection | SUN RGB-D | AP@0.15 (10 / NYU-37) | 14.01 | Holistic |
| Object Detection | SUN RGB-D | AP@0.15 (10 / PNet-30) | 14.01 | Holistic |
| 3D | SUN RGB-D | AP@0.15 (10 / NYU-37) | 14.01 | Holistic |
| 3D | SUN RGB-D | AP@0.15 (10 / PNet-30) | 14.01 | Holistic |
| 3D Object Detection | SUN RGB-D | AP@0.15 (10 / NYU-37) | 14.01 | Holistic |
| 3D Object Detection | SUN RGB-D | AP@0.15 (10 / PNet-30) | 14.01 | Holistic |
| 2D Classification | SUN RGB-D | AP@0.15 (10 / NYU-37) | 14.01 | Holistic |
| 2D Classification | SUN RGB-D | AP@0.15 (10 / PNet-30) | 14.01 | Holistic |
| 2D Object Detection | SUN RGB-D | AP@0.15 (10 / NYU-37) | 14.01 | Holistic |
| 2D Object Detection | SUN RGB-D | AP@0.15 (10 / PNet-30) | 14.01 | Holistic |
| 16k | SUN RGB-D | AP@0.15 (10 / NYU-37) | 14.01 | Holistic |
| 16k | SUN RGB-D | AP@0.15 (10 / PNet-30) | 14.01 | Holistic |
| Room Layout Estimation | SUN RGB-D | Camera Pitch | 7.6 | Holistic |
| Room Layout Estimation | SUN RGB-D | Camera Roll | 3.12 | Holistic |
| Room Layout Estimation | SUN RGB-D | IoU | 54.9 | Holistic |