HorizonNet
Room layout predictor
A PyTorch implementation of a deep learning model for learning room layouts from 360 photos and providing tools for layout analysis and data augmentation.
Pytorch implementation of HorizonNet: Learning Room Layout with 1D Representation and Pano Stretch Data Augmentation.
325 stars
21 watching
88 forks
Language: Python
last commit: over 2 years ago360-photocomputer-visioncvpr2019horizonnetpano-stretch-augmentationroom-layout
Related projects:
| Repository | Description | Stars |
|---|---|---|
| sunset1995/hohonet | A PyTorch implementation of a 360-degree indoor depth estimation and room layout analysis system using latent horizontal features. | 108 |
 zillow/psmnet-layout | A deep learning framework for estimating 3D room layouts from multi-view panoramas using stereo vision and fusion techniques. | 3 |
 zhijieshen-bjtu/dopnet | Estimates room layouts from indoor panoramic images by disentangling orthogonal planes and accounting for cross-scale distortion | 48 |
 mrgemy95/visual-interaction-networks-pytorch | An implementation of Deepmind's Visual Interaction Networks using PyTorch to predict future events in physical scenes. | 166 |
 zouchuhang/layoutnetv2 | An implementation of a 3D layout reconstruction algorithm from a single 360-degree image using PyTorch and machine learning techniques. | 222 |
 harvardnlp/pytorch-struct | A PyTorch library implementing differentiable structured prediction algorithms for deep learning applications. | 1,109 |
 fxia22/pointnet.pytorch | This is an implementation of the PointNet algorithm in PyTorch for 3D point cloud classification and segmentation tasks. | 2,175 |
 4uiiurz1/pytorch-res2net | Implementations of deep learning architectures using PyTorch for image classification tasks on various datasets. | 112 |
 isht7/pytorch-deeplab-resnet | A deep learning model implementation of the DeepLab ResNet architecture for image segmentation tasks. | 602 |
| zouchuhang/layoutnet | A deep learning implementation of a 3D room layout reconstruction method from a single RGB image | 419 |
 locuslab/optnet | A PyTorch module that adds differentiable optimization as a layer to neural networks | 517 |
 benedekrozemberczki/pytorch_geometric_temporal | A PyTorch extension for building temporal graph neural networks with support for recurrent and attention-based models | 2,694 |
 dyhan0920/pyramidnet-pytorch | An implementation of a deep neural network architecture for image classification tasks | 273 |
 damitkwr/esrnn-gpu | A PyTorch implementation of an optimized deep learning model for time series forecasting on GPUs. | 319 |
 xternalz/wideresnet-pytorch | An implementation of Wide Residual Networks in PyTorch for efficient deep learning on CIFAR10/100 datasets. | 334 |