• Laser & Optoelectronics Progress
  • Vol. 59, Issue 18, 1810007 (2022)
Youjun Yue1, Jie Zhang1、*, Hui Zhao1、2, and Hongjun Wang1
Author Affiliations
  • 1Tianjin Key Laboratory of Control Theory & Applications in Complicated Systems, School of Electrical and Electronic Engineering, Tianjin University of Technology, Tianjin 300384, China
  • 2College of Engineering and Technology, Tianjin Agricultural University, Tianjin 300392, China
  • show less
    DOI: 10.3788/LOP202259.1810007 Cite this Article Set citation alerts
    Youjun Yue, Jie Zhang, Hui Zhao, Hongjun Wang. Real-Time Indoor Scene Layout Estimation Based on Improved Lightweight Network[J]. Laser & Optoelectronics Progress, 2022, 59(18): 1810007 Copy Citation Text show less

    Abstract

    This study proposes a real-time layout estimation method based on an improved lightweight network to simplify the network structure of layout estimation and improve the use of output features. A lightweight coding and decoding network was used to obtain the main plane segmentation images of indoor scenes directly end-to-end and realize real-time layout estimation. Aiming at the problem of low feature usage in previous joint learning methods, a simplified joint learning module was introduced and the gradient of the output segmentation graph was used as the output edge. Additionally, the loss of the edge was directly integrated into the output loss of the entire network to improve feature utilization and simplify the joint learning network. Aiming at the imbalance of positive and negative labels of dataset and the imbalance of layout type distribution, to improve the stability of network training, segmentation semantic transfer was used to initialize the network parameters in this paper using the semantic segmentation network parameters trained on the LSUN dataset. The performance of the proposed method was evaluated using two benchmark datasets. The results show that the average pixel error of the proposed method is 7.35% and 8.32% on the LSUN and Hedau datasets, respectively. Ablation experiments prove the effectiveness of hierarchical supervision, simplified joint learning, and semantic transfer mechanism for improving accuracy. Finally, the experimental results show that the proposed method can estimate accurate indoor scene layout in real-time.
    Youjun Yue, Jie Zhang, Hui Zhao, Hongjun Wang. Real-Time Indoor Scene Layout Estimation Based on Improved Lightweight Network[J]. Laser & Optoelectronics Progress, 2022, 59(18): 1810007
    Download Citation