HorizonNet
by sunset1995
Pytorch implementation of HorizonNet: Learning Room Layout with 1D Representation and Pano Stretch Data Augmentation.
AI summary
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.
- stars
- 325
- forks
- 88
- watching
- 21
Similar projects
Found by comparing what the projects do, not just their names.
360 analyzer
A PyTorch implementation of a 360-degree indoor depth estimation and room layout analysis system using latent horizontal features.
Layout estimator
A deep learning framework for estimating 3D room layouts from multi-view panoramas using stereo vision and fusion techniques.
Room layout estimator
Estimates room layouts from indoor panoramic images by disentangling orthogonal planes and accounting for cross-scale distortion
Future prediction model
An implementation of Deepmind's Visual Interaction Networks using PyTorch to predict future events in physical scenes.
Layout model
An implementation of a 3D layout reconstruction algorithm from a single 360-degree image using PyTorch and machine learning techniques.
Structured Prediction Library
A PyTorch library implementing differentiable structured prediction algorithms for deep learning applications.
Point Cloud Model
This is an implementation of the PointNet algorithm in PyTorch for 3D point cloud classification and segmentation tasks.
Deep learning models
Implementations of deep learning architectures using PyTorch for image classification tasks on various datasets.
Deep Learning Model
A deep learning model implementation of the DeepLab ResNet architecture for image segmentation tasks.
Layout reconstruction
A deep learning implementation of a 3D room layout reconstruction method from a single RGB image
Optimization layer
A PyTorch module that adds differentiable optimization as a layer to neural networks
Temporal graph NN library
A PyTorch extension for building temporal graph neural networks with support for recurrent and attention-based models
Neural net architecture
An implementation of a deep neural network architecture for image classification tasks
Time series forecasting model
A PyTorch implementation of an optimized deep learning model for time series forecasting on GPUs.
Deep network architecture
An implementation of Wide Residual Networks in PyTorch for efficient deep learning on CIFAR10/100 datasets.
