Title
Horizonnet: Learning Room Layout With 1d Representation And Pano Stretch Data Augmentation
Abstract
We present a new approach to the problem of estimating the 3D room layout from a single panoramic image. We represent room layout as three 1D vectors that encode, at each image column, the boundary positions of floor-wall and ceiling-wall, and the existence of wall-wall boundary. The proposed network, HorizonNet, trained for predicting 1D layout, outperforms previous state-of-the-art approaches. The designed post-processing procedure for recovering 3D room layouts from 1D predictions can automatically infer the room shape with low computation cost-it takes less than 20ms for a panorama image while prior works might need dozens of seconds. We also propose Pano Stretch Data Augmentation, which can diversify panorama data and be applied to other panorama-related learning tasks. Due to the limited data available for non-cuboid layout, we relabel 65 general layout from the current dataset for fine-tuning. Our approach shows good performance on general layouts by qualitative results and cross-validation.
Year
DOI
Venue
2019
10.1109/CVPR.2019.00114
2019 IEEE/CVF CONFERENCE ON COMPUTER VISION AND PATTERN RECOGNITION (CVPR 2019)
DocType
Volume
ISSN
Conference
abs/1901.03861
1063-6919
Citations 
PageRank 
References 
2
0.36
0
Authors
4
Name
Order
Citations
PageRank
Cheng Sun123.06
Chi-Wei Hsiao221.04
Min Sun3108359.15
Hwann-Tzong Chen482652.13