├── .gitignore ├── LICENSE ├── README.md ├── cfg.yaml ├── datasets ├── __init__.py ├── custom.py ├── nyu303.py ├── structured3d.py └── sunrgbd.py ├── example └── NYU0969 │ ├── annotation │ └── index.json │ ├── annotation2D3D │ ├── index.json │ ├── index.json_20150709082104 │ ├── index.json_20150709082113 │ ├── index.json_20150709082133 │ └── index.json_20150709082141 │ ├── annotation2Dfinal │ └── index.json │ ├── annotation3D │ └── index.json │ ├── annotation3Dfinal │ └── index.json │ ├── annotation3Dlayout │ └── index.json │ ├── depth │ └── NYU0969.png │ ├── depth_bfx │ └── NYU0969.png │ ├── extrinsics │ └── 20150118235333.txt │ ├── fullres │ ├── NYU0969.jpg │ ├── NYU0969.png │ └── intrinsics.txt │ ├── image │ └── NYU0969.jpg │ ├── intrinsics.txt │ ├── scene.txt │ └── seg.mat ├── figs └── teaser.gif ├── models ├── __init__.py ├── detector.py ├── heads.py ├── hr_cfg.py ├── hrnet.py ├── loss.py ├── reconstruction.py ├── utils.py └── visualize.py ├── requirements.txt ├── results ├── 0.png └── 0_1.png ├── test.py └── train.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/README.md -------------------------------------------------------------------------------- /cfg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/cfg.yaml -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/datasets/__init__.py -------------------------------------------------------------------------------- /datasets/custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/datasets/custom.py -------------------------------------------------------------------------------- /datasets/nyu303.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/datasets/nyu303.py -------------------------------------------------------------------------------- /datasets/structured3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/datasets/structured3d.py -------------------------------------------------------------------------------- /datasets/sunrgbd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/datasets/sunrgbd.py -------------------------------------------------------------------------------- /example/NYU0969/annotation/index.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/example/NYU0969/annotation/index.json -------------------------------------------------------------------------------- /example/NYU0969/annotation2D3D/index.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/example/NYU0969/annotation2D3D/index.json -------------------------------------------------------------------------------- /example/NYU0969/annotation2D3D/index.json_20150709082104: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/example/NYU0969/annotation2D3D/index.json_20150709082104 -------------------------------------------------------------------------------- /example/NYU0969/annotation2D3D/index.json_20150709082113: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/example/NYU0969/annotation2D3D/index.json_20150709082113 -------------------------------------------------------------------------------- /example/NYU0969/annotation2D3D/index.json_20150709082133: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/example/NYU0969/annotation2D3D/index.json_20150709082133 -------------------------------------------------------------------------------- /example/NYU0969/annotation2D3D/index.json_20150709082141: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/example/NYU0969/annotation2D3D/index.json_20150709082141 -------------------------------------------------------------------------------- /example/NYU0969/annotation2Dfinal/index.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/example/NYU0969/annotation2Dfinal/index.json -------------------------------------------------------------------------------- /example/NYU0969/annotation3D/index.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/example/NYU0969/annotation3D/index.json -------------------------------------------------------------------------------- /example/NYU0969/annotation3Dfinal/index.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/example/NYU0969/annotation3Dfinal/index.json -------------------------------------------------------------------------------- /example/NYU0969/annotation3Dlayout/index.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/example/NYU0969/annotation3Dlayout/index.json -------------------------------------------------------------------------------- /example/NYU0969/depth/NYU0969.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/example/NYU0969/depth/NYU0969.png -------------------------------------------------------------------------------- /example/NYU0969/depth_bfx/NYU0969.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/example/NYU0969/depth_bfx/NYU0969.png -------------------------------------------------------------------------------- /example/NYU0969/extrinsics/20150118235333.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/example/NYU0969/extrinsics/20150118235333.txt -------------------------------------------------------------------------------- /example/NYU0969/fullres/NYU0969.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/example/NYU0969/fullres/NYU0969.jpg -------------------------------------------------------------------------------- /example/NYU0969/fullres/NYU0969.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/example/NYU0969/fullres/NYU0969.png -------------------------------------------------------------------------------- /example/NYU0969/fullres/intrinsics.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/example/NYU0969/fullres/intrinsics.txt -------------------------------------------------------------------------------- /example/NYU0969/image/NYU0969.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/example/NYU0969/image/NYU0969.jpg -------------------------------------------------------------------------------- /example/NYU0969/intrinsics.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/example/NYU0969/intrinsics.txt -------------------------------------------------------------------------------- /example/NYU0969/scene.txt: -------------------------------------------------------------------------------- 1 | bedroom -------------------------------------------------------------------------------- /example/NYU0969/seg.mat: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/example/NYU0969/seg.mat -------------------------------------------------------------------------------- /figs/teaser.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/figs/teaser.gif -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/models/__init__.py -------------------------------------------------------------------------------- /models/detector.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/models/detector.py -------------------------------------------------------------------------------- /models/heads.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/models/heads.py -------------------------------------------------------------------------------- /models/hr_cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/models/hr_cfg.py -------------------------------------------------------------------------------- /models/hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/models/hrnet.py -------------------------------------------------------------------------------- /models/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/models/loss.py -------------------------------------------------------------------------------- /models/reconstruction.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/models/reconstruction.py -------------------------------------------------------------------------------- /models/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/models/utils.py -------------------------------------------------------------------------------- /models/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/models/visualize.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/requirements.txt -------------------------------------------------------------------------------- /results/0.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/results/0.png -------------------------------------------------------------------------------- /results/0_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/results/0_1.png -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/CYang0515/NonCuboidRoom/HEAD/train.py --------------------------------------------------------------------------------