├── .gitignore ├── LICENSE ├── PlaneRecTR ├── PlaneRecTR_model.py ├── __init__.py ├── config.py ├── data │ ├── __init__.py │ ├── dataset_mappers │ │ ├── __init__.py │ │ ├── nyuv2_plane_dataset_mapper.py │ │ └── scannetv1_plane_dataset_mapper.py │ └── datasets │ │ ├── __init__.py │ │ ├── register_nyuv2_plane.py │ │ └── register_scannetv1_plane.py ├── evaluation │ ├── __init__.py │ └── planeSeg_evaluation.py ├── modeling │ ├── __init__.py │ ├── backbone │ │ ├── __init__.py │ │ ├── hrnet.py │ │ └── swin.py │ ├── criterion.py │ ├── matcher.py │ ├── meta_arch │ │ ├── __init__.py │ │ └── mask_former_head.py │ ├── pixel_decoder │ │ ├── __init__.py │ │ ├── fpn.py │ │ ├── msdeformattn.py │ │ └── ops │ │ │ ├── functions │ │ │ ├── __init__.py │ │ │ └── ms_deform_attn_func.py │ │ │ ├── make.sh │ │ │ ├── modules │ │ │ ├── __init__.py │ │ │ └── ms_deform_attn.py │ │ │ ├── setup.py │ │ │ ├── src │ │ │ ├── cpu │ │ │ │ ├── ms_deform_attn_cpu.cpp │ │ │ │ └── ms_deform_attn_cpu.h │ │ │ ├── cuda │ │ │ │ ├── ms_deform_attn_cuda.cu │ │ │ │ ├── ms_deform_attn_cuda.h │ │ │ │ └── ms_deform_im2col_cuda.cuh │ │ │ ├── ms_deform_attn.h │ │ │ └── vision.cpp │ │ │ └── test.py │ └── transformer_decoder │ │ ├── __init__.py │ │ ├── masked_transformer_decoder.py │ │ ├── position_encoding.py │ │ └── transformer.py ├── test_time_augmentation.py └── utils │ ├── __init__.py │ ├── disp.py │ ├── metrics.py │ ├── metrics_de.py │ ├── metrics_onlyparams.py │ └── misc.py ├── README.md ├── configs ├── PlaneRecTRNYUV2 │ ├── Base-Segmentation.yaml │ ├── PlaneRecTR_R50_bs16_50ep.yaml │ ├── hrnet │ │ └── PlaneRecTR_hrnet_w32_imagenet_pretrained.yaml │ └── swin │ │ └── PlaneRecTR_swin_base_384_bs16_50ep.yaml └── PlaneRecTRScanNetV1 │ ├── Base-Segmentation.yaml │ ├── PlaneRecTR_R50_bs16_50ep.yaml │ ├── PlaneRecTR_R50_demo.yaml │ ├── hrnet │ └── PlaneRecTR_hrnet_w32_imagenet_pretrained.yaml │ └── swin │ └── PlaneRecTR_swin_base_384_bs16_50ep.yaml ├── demo ├── 359_d2_image.png ├── demo.py └── predictor.py ├── figs └── overview.png ├── requirements.txt ├── tools ├── generate_nyuv2_plane_dataset.py └── generate_scannetv1_plane_dataset.py └── train_net.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/LICENSE -------------------------------------------------------------------------------- /PlaneRecTR/PlaneRecTR_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/PlaneRecTR_model.py -------------------------------------------------------------------------------- /PlaneRecTR/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/__init__.py -------------------------------------------------------------------------------- /PlaneRecTR/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/config.py -------------------------------------------------------------------------------- /PlaneRecTR/data/__init__.py: -------------------------------------------------------------------------------- 1 | from . import datasets 2 | -------------------------------------------------------------------------------- /PlaneRecTR/data/dataset_mappers/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PlaneRecTR/data/dataset_mappers/nyuv2_plane_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/data/dataset_mappers/nyuv2_plane_dataset_mapper.py -------------------------------------------------------------------------------- /PlaneRecTR/data/dataset_mappers/scannetv1_plane_dataset_mapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/data/dataset_mappers/scannetv1_plane_dataset_mapper.py -------------------------------------------------------------------------------- /PlaneRecTR/data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/data/datasets/__init__.py -------------------------------------------------------------------------------- /PlaneRecTR/data/datasets/register_nyuv2_plane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/data/datasets/register_nyuv2_plane.py -------------------------------------------------------------------------------- /PlaneRecTR/data/datasets/register_scannetv1_plane.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/data/datasets/register_scannetv1_plane.py -------------------------------------------------------------------------------- /PlaneRecTR/evaluation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PlaneRecTR/evaluation/planeSeg_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/evaluation/planeSeg_evaluation.py -------------------------------------------------------------------------------- /PlaneRecTR/modeling/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/__init__.py -------------------------------------------------------------------------------- /PlaneRecTR/modeling/backbone/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PlaneRecTR/modeling/backbone/hrnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/backbone/hrnet.py -------------------------------------------------------------------------------- /PlaneRecTR/modeling/backbone/swin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/backbone/swin.py -------------------------------------------------------------------------------- /PlaneRecTR/modeling/criterion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/criterion.py -------------------------------------------------------------------------------- /PlaneRecTR/modeling/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/matcher.py -------------------------------------------------------------------------------- /PlaneRecTR/modeling/meta_arch/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. 2 | -------------------------------------------------------------------------------- /PlaneRecTR/modeling/meta_arch/mask_former_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/meta_arch/mask_former_head.py -------------------------------------------------------------------------------- /PlaneRecTR/modeling/pixel_decoder/__init__.py: -------------------------------------------------------------------------------- 1 | # Copyright (c) Facebook, Inc. and its affiliates. 2 | -------------------------------------------------------------------------------- /PlaneRecTR/modeling/pixel_decoder/fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/pixel_decoder/fpn.py -------------------------------------------------------------------------------- /PlaneRecTR/modeling/pixel_decoder/msdeformattn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/pixel_decoder/msdeformattn.py -------------------------------------------------------------------------------- /PlaneRecTR/modeling/pixel_decoder/ops/functions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/pixel_decoder/ops/functions/__init__.py -------------------------------------------------------------------------------- /PlaneRecTR/modeling/pixel_decoder/ops/functions/ms_deform_attn_func.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/pixel_decoder/ops/functions/ms_deform_attn_func.py -------------------------------------------------------------------------------- /PlaneRecTR/modeling/pixel_decoder/ops/make.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/pixel_decoder/ops/make.sh -------------------------------------------------------------------------------- /PlaneRecTR/modeling/pixel_decoder/ops/modules/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/pixel_decoder/ops/modules/__init__.py -------------------------------------------------------------------------------- /PlaneRecTR/modeling/pixel_decoder/ops/modules/ms_deform_attn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/pixel_decoder/ops/modules/ms_deform_attn.py -------------------------------------------------------------------------------- /PlaneRecTR/modeling/pixel_decoder/ops/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/pixel_decoder/ops/setup.py -------------------------------------------------------------------------------- /PlaneRecTR/modeling/pixel_decoder/ops/src/cpu/ms_deform_attn_cpu.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/pixel_decoder/ops/src/cpu/ms_deform_attn_cpu.cpp -------------------------------------------------------------------------------- /PlaneRecTR/modeling/pixel_decoder/ops/src/cpu/ms_deform_attn_cpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/pixel_decoder/ops/src/cpu/ms_deform_attn_cpu.h -------------------------------------------------------------------------------- /PlaneRecTR/modeling/pixel_decoder/ops/src/cuda/ms_deform_attn_cuda.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/pixel_decoder/ops/src/cuda/ms_deform_attn_cuda.cu -------------------------------------------------------------------------------- /PlaneRecTR/modeling/pixel_decoder/ops/src/cuda/ms_deform_attn_cuda.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/pixel_decoder/ops/src/cuda/ms_deform_attn_cuda.h -------------------------------------------------------------------------------- /PlaneRecTR/modeling/pixel_decoder/ops/src/cuda/ms_deform_im2col_cuda.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/pixel_decoder/ops/src/cuda/ms_deform_im2col_cuda.cuh -------------------------------------------------------------------------------- /PlaneRecTR/modeling/pixel_decoder/ops/src/ms_deform_attn.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/pixel_decoder/ops/src/ms_deform_attn.h -------------------------------------------------------------------------------- /PlaneRecTR/modeling/pixel_decoder/ops/src/vision.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/pixel_decoder/ops/src/vision.cpp -------------------------------------------------------------------------------- /PlaneRecTR/modeling/pixel_decoder/ops/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/pixel_decoder/ops/test.py -------------------------------------------------------------------------------- /PlaneRecTR/modeling/transformer_decoder/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/transformer_decoder/__init__.py -------------------------------------------------------------------------------- /PlaneRecTR/modeling/transformer_decoder/masked_transformer_decoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/transformer_decoder/masked_transformer_decoder.py -------------------------------------------------------------------------------- /PlaneRecTR/modeling/transformer_decoder/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/transformer_decoder/position_encoding.py -------------------------------------------------------------------------------- /PlaneRecTR/modeling/transformer_decoder/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/modeling/transformer_decoder/transformer.py -------------------------------------------------------------------------------- /PlaneRecTR/test_time_augmentation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/test_time_augmentation.py -------------------------------------------------------------------------------- /PlaneRecTR/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /PlaneRecTR/utils/disp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/utils/disp.py -------------------------------------------------------------------------------- /PlaneRecTR/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/utils/metrics.py -------------------------------------------------------------------------------- /PlaneRecTR/utils/metrics_de.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/utils/metrics_de.py -------------------------------------------------------------------------------- /PlaneRecTR/utils/metrics_onlyparams.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/utils/metrics_onlyparams.py -------------------------------------------------------------------------------- /PlaneRecTR/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/PlaneRecTR/utils/misc.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/README.md -------------------------------------------------------------------------------- /configs/PlaneRecTRNYUV2/Base-Segmentation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/configs/PlaneRecTRNYUV2/Base-Segmentation.yaml -------------------------------------------------------------------------------- /configs/PlaneRecTRNYUV2/PlaneRecTR_R50_bs16_50ep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/configs/PlaneRecTRNYUV2/PlaneRecTR_R50_bs16_50ep.yaml -------------------------------------------------------------------------------- /configs/PlaneRecTRNYUV2/hrnet/PlaneRecTR_hrnet_w32_imagenet_pretrained.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/configs/PlaneRecTRNYUV2/hrnet/PlaneRecTR_hrnet_w32_imagenet_pretrained.yaml -------------------------------------------------------------------------------- /configs/PlaneRecTRNYUV2/swin/PlaneRecTR_swin_base_384_bs16_50ep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/configs/PlaneRecTRNYUV2/swin/PlaneRecTR_swin_base_384_bs16_50ep.yaml -------------------------------------------------------------------------------- /configs/PlaneRecTRScanNetV1/Base-Segmentation.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/configs/PlaneRecTRScanNetV1/Base-Segmentation.yaml -------------------------------------------------------------------------------- /configs/PlaneRecTRScanNetV1/PlaneRecTR_R50_bs16_50ep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/configs/PlaneRecTRScanNetV1/PlaneRecTR_R50_bs16_50ep.yaml -------------------------------------------------------------------------------- /configs/PlaneRecTRScanNetV1/PlaneRecTR_R50_demo.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/configs/PlaneRecTRScanNetV1/PlaneRecTR_R50_demo.yaml -------------------------------------------------------------------------------- /configs/PlaneRecTRScanNetV1/hrnet/PlaneRecTR_hrnet_w32_imagenet_pretrained.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/configs/PlaneRecTRScanNetV1/hrnet/PlaneRecTR_hrnet_w32_imagenet_pretrained.yaml -------------------------------------------------------------------------------- /configs/PlaneRecTRScanNetV1/swin/PlaneRecTR_swin_base_384_bs16_50ep.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/configs/PlaneRecTRScanNetV1/swin/PlaneRecTR_swin_base_384_bs16_50ep.yaml -------------------------------------------------------------------------------- /demo/359_d2_image.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/demo/359_d2_image.png -------------------------------------------------------------------------------- /demo/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/demo/demo.py -------------------------------------------------------------------------------- /demo/predictor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/demo/predictor.py -------------------------------------------------------------------------------- /figs/overview.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/figs/overview.png -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/requirements.txt -------------------------------------------------------------------------------- /tools/generate_nyuv2_plane_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/tools/generate_nyuv2_plane_dataset.py -------------------------------------------------------------------------------- /tools/generate_scannetv1_plane_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/tools/generate_scannetv1_plane_dataset.py -------------------------------------------------------------------------------- /train_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/SJingjia/PlaneRecTR/HEAD/train_net.py --------------------------------------------------------------------------------