├── .gitignore ├── .style.yapf ├── README.md ├── data_preprocess ├── points100 ├── points20 ├── points200 ├── points50 ├── scannet.py ├── scannet_eff.py └── utils.py ├── demo.py ├── doc ├── result1.png ├── result2.png ├── result3.png ├── result4.png ├── result5.png ├── result6.png └── result7.png ├── finetune ├── LICENSE ├── config.py ├── datasets │ └── README.md ├── ddp_main.py ├── ddp_mian.sh ├── ddp_mian_20.sh ├── lib │ ├── .test_mean.py │ ├── __init__.py │ ├── dataloader.py │ ├── dataset.py │ ├── datasets │ │ ├── __init__.py │ │ ├── preprocessing │ │ │ └── scannet.py │ │ ├── scannet.py │ │ └── scannet_test.py │ ├── distributed_utils.py │ ├── layers.py │ ├── math_functions.py │ ├── pc_utils.py │ ├── solvers.py │ ├── test.py │ ├── train.py │ ├── transforms.py │ ├── utils.py │ └── voxelizer.py ├── main.py ├── models │ ├── __init__.py │ ├── conditional_random_fields.py │ ├── model.py │ ├── modules │ │ ├── __init__.py │ │ ├── common.py │ │ ├── resnet_block.py │ │ └── senet_block.py │ ├── res16unet.py │ ├── resnet.py │ ├── resunet.py │ └── wrapper.py ├── run.py ├── run.sh ├── scannet.py ├── scripts │ ├── ddp_mian.sh │ └── train_scannet.sh └── splits │ └── scannet │ ├── scannetv2_test.txt │ ├── scannetv2_train.txt │ ├── scannetv2_trainval.txt │ └── scannetv2_val.txt ├── pointnet2 ├── .gitignore ├── _ext_src │ ├── include │ │ ├── ball_query.h │ │ ├── cuda_utils.h │ │ ├── group_points.h │ │ ├── interpolate.h │ │ ├── sampling.h │ │ └── utils.h │ └── src │ │ ├── ball_query.cpp │ │ ├── ball_query_gpu.cu │ │ ├── bindings.cpp │ │ ├── group_points.cpp │ │ ├── group_points_gpu.cu │ │ ├── interpolate.cpp │ │ ├── interpolate_gpu.cu │ │ ├── sampling.cpp │ │ └── sampling_gpu.cu ├── pointnet2_modules.py ├── pointnet2_test.py ├── pointnet2_utils.py ├── pytorch_utils.py └── setup.py ├── pretrain ├── .gitignore ├── .style.yapf ├── README.md ├── config.py ├── ddp_main.py ├── ddp_main_save.py ├── ddp_mian_bt.sh ├── demo │ ├── scannet.py │ └── stanford.py ├── imgs │ ├── scannet.png │ └── stanford.png ├── lib │ ├── __init__.py │ ├── dataloader.py │ ├── dataset.py │ ├── datasets │ │ ├── __init__.py │ │ ├── preprocessing │ │ │ ├── scannet.py │ │ │ └── stanford.py │ │ ├── scannet.py │ │ └── scannet_test.py │ ├── distributed_utils.py │ ├── hack.py │ ├── layers.py │ ├── math_functions.py │ ├── pc_utils.py │ ├── solvers.py │ ├── test.py │ ├── train.py │ ├── transforms.py │ ├── utils.py │ └── voxelizer.py ├── main.py ├── models │ ├── __init__.py │ ├── conditional_random_fields.py │ ├── model.py │ ├── modules │ │ ├── __init__.py │ │ ├── common.py │ │ ├── resnet_block.py │ │ └── senet_block.py │ ├── res16unet.py │ ├── resnet.py │ ├── resunet.py │ └── wrapper.py ├── run.py ├── run.sh ├── sampling.py ├── scannet_bt.py ├── scripts │ ├── ddp_mian.sh │ ├── tmp_dir_scannet │ │ └── 2021-05-16_19-45-28.txt │ ├── train_scannet.sh │ ├── train_stanford.sh │ ├── train_synthia4d.sh │ └── train_synthia4d_temporal.sh └── splits │ └── scannet │ ├── scannetv2_test.txt │ ├── scannetv2_test_oo.txt │ ├── scannetv2_train.txt │ ├── scannetv2_train_2.txt │ ├── scannetv2_trainval.txt │ └── scannetv2_val.txt └── requirements.sh /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/.gitignore -------------------------------------------------------------------------------- /.style.yapf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/.style.yapf -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/README.md -------------------------------------------------------------------------------- /data_preprocess/points100: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/data_preprocess/points100 -------------------------------------------------------------------------------- /data_preprocess/points20: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/data_preprocess/points20 -------------------------------------------------------------------------------- /data_preprocess/points200: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/data_preprocess/points200 -------------------------------------------------------------------------------- /data_preprocess/points50: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/data_preprocess/points50 -------------------------------------------------------------------------------- /data_preprocess/scannet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/data_preprocess/scannet.py -------------------------------------------------------------------------------- /data_preprocess/scannet_eff.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/data_preprocess/scannet_eff.py -------------------------------------------------------------------------------- /data_preprocess/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/data_preprocess/utils.py -------------------------------------------------------------------------------- /demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/demo.py -------------------------------------------------------------------------------- /doc/result1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/doc/result1.png -------------------------------------------------------------------------------- /doc/result2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/doc/result2.png -------------------------------------------------------------------------------- /doc/result3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/doc/result3.png -------------------------------------------------------------------------------- /doc/result4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/doc/result4.png -------------------------------------------------------------------------------- /doc/result5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/doc/result5.png -------------------------------------------------------------------------------- /doc/result6.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/doc/result6.png -------------------------------------------------------------------------------- /doc/result7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/doc/result7.png -------------------------------------------------------------------------------- /finetune/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/LICENSE -------------------------------------------------------------------------------- /finetune/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/config.py -------------------------------------------------------------------------------- /finetune/datasets/README.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /finetune/ddp_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/ddp_main.py -------------------------------------------------------------------------------- /finetune/ddp_mian.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/ddp_mian.sh -------------------------------------------------------------------------------- /finetune/ddp_mian_20.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/ddp_mian_20.sh -------------------------------------------------------------------------------- /finetune/lib/.test_mean.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/lib/.test_mean.py -------------------------------------------------------------------------------- /finetune/lib/__init__.py: -------------------------------------------------------------------------------- 1 | import open3d as o3d 2 | -------------------------------------------------------------------------------- /finetune/lib/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/lib/dataloader.py -------------------------------------------------------------------------------- /finetune/lib/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/lib/dataset.py -------------------------------------------------------------------------------- /finetune/lib/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/lib/datasets/__init__.py -------------------------------------------------------------------------------- /finetune/lib/datasets/preprocessing/scannet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/lib/datasets/preprocessing/scannet.py -------------------------------------------------------------------------------- /finetune/lib/datasets/scannet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/lib/datasets/scannet.py -------------------------------------------------------------------------------- /finetune/lib/datasets/scannet_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/lib/datasets/scannet_test.py -------------------------------------------------------------------------------- /finetune/lib/distributed_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/lib/distributed_utils.py -------------------------------------------------------------------------------- /finetune/lib/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/lib/layers.py -------------------------------------------------------------------------------- /finetune/lib/math_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/lib/math_functions.py -------------------------------------------------------------------------------- /finetune/lib/pc_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/lib/pc_utils.py -------------------------------------------------------------------------------- /finetune/lib/solvers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/lib/solvers.py -------------------------------------------------------------------------------- /finetune/lib/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/lib/test.py -------------------------------------------------------------------------------- /finetune/lib/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/lib/train.py -------------------------------------------------------------------------------- /finetune/lib/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/lib/transforms.py -------------------------------------------------------------------------------- /finetune/lib/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/lib/utils.py -------------------------------------------------------------------------------- /finetune/lib/voxelizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/lib/voxelizer.py -------------------------------------------------------------------------------- /finetune/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/main.py -------------------------------------------------------------------------------- /finetune/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/models/__init__.py -------------------------------------------------------------------------------- /finetune/models/conditional_random_fields.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/models/conditional_random_fields.py -------------------------------------------------------------------------------- /finetune/models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/models/model.py -------------------------------------------------------------------------------- /finetune/models/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /finetune/models/modules/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/models/modules/common.py -------------------------------------------------------------------------------- /finetune/models/modules/resnet_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/models/modules/resnet_block.py -------------------------------------------------------------------------------- /finetune/models/modules/senet_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/models/modules/senet_block.py -------------------------------------------------------------------------------- /finetune/models/res16unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/models/res16unet.py -------------------------------------------------------------------------------- /finetune/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/models/resnet.py -------------------------------------------------------------------------------- /finetune/models/resunet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/models/resunet.py -------------------------------------------------------------------------------- /finetune/models/wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/models/wrapper.py -------------------------------------------------------------------------------- /finetune/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/run.py -------------------------------------------------------------------------------- /finetune/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/run.sh -------------------------------------------------------------------------------- /finetune/scannet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/scannet.py -------------------------------------------------------------------------------- /finetune/scripts/ddp_mian.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/scripts/ddp_mian.sh -------------------------------------------------------------------------------- /finetune/scripts/train_scannet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/scripts/train_scannet.sh -------------------------------------------------------------------------------- /finetune/splits/scannet/scannetv2_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/splits/scannet/scannetv2_test.txt -------------------------------------------------------------------------------- /finetune/splits/scannet/scannetv2_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/splits/scannet/scannetv2_train.txt -------------------------------------------------------------------------------- /finetune/splits/scannet/scannetv2_trainval.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/splits/scannet/scannetv2_trainval.txt -------------------------------------------------------------------------------- /finetune/splits/scannet/scannetv2_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/finetune/splits/scannet/scannetv2_val.txt -------------------------------------------------------------------------------- /pointnet2/.gitignore: -------------------------------------------------------------------------------- 1 | build/ 2 | dist/ 3 | *.egg-info -------------------------------------------------------------------------------- /pointnet2/_ext_src/include/ball_query.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pointnet2/_ext_src/include/ball_query.h -------------------------------------------------------------------------------- /pointnet2/_ext_src/include/cuda_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pointnet2/_ext_src/include/cuda_utils.h -------------------------------------------------------------------------------- /pointnet2/_ext_src/include/group_points.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pointnet2/_ext_src/include/group_points.h -------------------------------------------------------------------------------- /pointnet2/_ext_src/include/interpolate.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pointnet2/_ext_src/include/interpolate.h -------------------------------------------------------------------------------- /pointnet2/_ext_src/include/sampling.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pointnet2/_ext_src/include/sampling.h -------------------------------------------------------------------------------- /pointnet2/_ext_src/include/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pointnet2/_ext_src/include/utils.h -------------------------------------------------------------------------------- /pointnet2/_ext_src/src/ball_query.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pointnet2/_ext_src/src/ball_query.cpp -------------------------------------------------------------------------------- /pointnet2/_ext_src/src/ball_query_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pointnet2/_ext_src/src/ball_query_gpu.cu -------------------------------------------------------------------------------- /pointnet2/_ext_src/src/bindings.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pointnet2/_ext_src/src/bindings.cpp -------------------------------------------------------------------------------- /pointnet2/_ext_src/src/group_points.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pointnet2/_ext_src/src/group_points.cpp -------------------------------------------------------------------------------- /pointnet2/_ext_src/src/group_points_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pointnet2/_ext_src/src/group_points_gpu.cu -------------------------------------------------------------------------------- /pointnet2/_ext_src/src/interpolate.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pointnet2/_ext_src/src/interpolate.cpp -------------------------------------------------------------------------------- /pointnet2/_ext_src/src/interpolate_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pointnet2/_ext_src/src/interpolate_gpu.cu -------------------------------------------------------------------------------- /pointnet2/_ext_src/src/sampling.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pointnet2/_ext_src/src/sampling.cpp -------------------------------------------------------------------------------- /pointnet2/_ext_src/src/sampling_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pointnet2/_ext_src/src/sampling_gpu.cu -------------------------------------------------------------------------------- /pointnet2/pointnet2_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pointnet2/pointnet2_modules.py -------------------------------------------------------------------------------- /pointnet2/pointnet2_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pointnet2/pointnet2_test.py -------------------------------------------------------------------------------- /pointnet2/pointnet2_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pointnet2/pointnet2_utils.py -------------------------------------------------------------------------------- /pointnet2/pytorch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pointnet2/pytorch_utils.py -------------------------------------------------------------------------------- /pointnet2/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pointnet2/setup.py -------------------------------------------------------------------------------- /pretrain/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/.gitignore -------------------------------------------------------------------------------- /pretrain/.style.yapf: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/.style.yapf -------------------------------------------------------------------------------- /pretrain/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/README.md -------------------------------------------------------------------------------- /pretrain/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/config.py -------------------------------------------------------------------------------- /pretrain/ddp_main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/ddp_main.py -------------------------------------------------------------------------------- /pretrain/ddp_main_save.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/ddp_main_save.py -------------------------------------------------------------------------------- /pretrain/ddp_mian_bt.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/ddp_mian_bt.sh -------------------------------------------------------------------------------- /pretrain/demo/scannet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/demo/scannet.py -------------------------------------------------------------------------------- /pretrain/demo/stanford.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/demo/stanford.py -------------------------------------------------------------------------------- /pretrain/imgs/scannet.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/imgs/scannet.png -------------------------------------------------------------------------------- /pretrain/imgs/stanford.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/imgs/stanford.png -------------------------------------------------------------------------------- /pretrain/lib/__init__.py: -------------------------------------------------------------------------------- 1 | import open3d as o3d 2 | -------------------------------------------------------------------------------- /pretrain/lib/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/lib/dataloader.py -------------------------------------------------------------------------------- /pretrain/lib/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/lib/dataset.py -------------------------------------------------------------------------------- /pretrain/lib/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/lib/datasets/__init__.py -------------------------------------------------------------------------------- /pretrain/lib/datasets/preprocessing/scannet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/lib/datasets/preprocessing/scannet.py -------------------------------------------------------------------------------- /pretrain/lib/datasets/preprocessing/stanford.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/lib/datasets/preprocessing/stanford.py -------------------------------------------------------------------------------- /pretrain/lib/datasets/scannet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/lib/datasets/scannet.py -------------------------------------------------------------------------------- /pretrain/lib/datasets/scannet_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/lib/datasets/scannet_test.py -------------------------------------------------------------------------------- /pretrain/lib/distributed_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/lib/distributed_utils.py -------------------------------------------------------------------------------- /pretrain/lib/hack.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/lib/hack.py -------------------------------------------------------------------------------- /pretrain/lib/layers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/lib/layers.py -------------------------------------------------------------------------------- /pretrain/lib/math_functions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/lib/math_functions.py -------------------------------------------------------------------------------- /pretrain/lib/pc_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/lib/pc_utils.py -------------------------------------------------------------------------------- /pretrain/lib/solvers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/lib/solvers.py -------------------------------------------------------------------------------- /pretrain/lib/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/lib/test.py -------------------------------------------------------------------------------- /pretrain/lib/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/lib/train.py -------------------------------------------------------------------------------- /pretrain/lib/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/lib/transforms.py -------------------------------------------------------------------------------- /pretrain/lib/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/lib/utils.py -------------------------------------------------------------------------------- /pretrain/lib/voxelizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/lib/voxelizer.py -------------------------------------------------------------------------------- /pretrain/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/main.py -------------------------------------------------------------------------------- /pretrain/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/models/__init__.py -------------------------------------------------------------------------------- /pretrain/models/conditional_random_fields.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/models/conditional_random_fields.py -------------------------------------------------------------------------------- /pretrain/models/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/models/model.py -------------------------------------------------------------------------------- /pretrain/models/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /pretrain/models/modules/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/models/modules/common.py -------------------------------------------------------------------------------- /pretrain/models/modules/resnet_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/models/modules/resnet_block.py -------------------------------------------------------------------------------- /pretrain/models/modules/senet_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/models/modules/senet_block.py -------------------------------------------------------------------------------- /pretrain/models/res16unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/models/res16unet.py -------------------------------------------------------------------------------- /pretrain/models/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/models/resnet.py -------------------------------------------------------------------------------- /pretrain/models/resunet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/models/resunet.py -------------------------------------------------------------------------------- /pretrain/models/wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/models/wrapper.py -------------------------------------------------------------------------------- /pretrain/run.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/run.py -------------------------------------------------------------------------------- /pretrain/run.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/run.sh -------------------------------------------------------------------------------- /pretrain/sampling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/sampling.py -------------------------------------------------------------------------------- /pretrain/scannet_bt.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/scannet_bt.py -------------------------------------------------------------------------------- /pretrain/scripts/ddp_mian.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/scripts/ddp_mian.sh -------------------------------------------------------------------------------- /pretrain/scripts/tmp_dir_scannet/2021-05-16_19-45-28.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/scripts/tmp_dir_scannet/2021-05-16_19-45-28.txt -------------------------------------------------------------------------------- /pretrain/scripts/train_scannet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/scripts/train_scannet.sh -------------------------------------------------------------------------------- /pretrain/scripts/train_stanford.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/scripts/train_stanford.sh -------------------------------------------------------------------------------- /pretrain/scripts/train_synthia4d.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/scripts/train_synthia4d.sh -------------------------------------------------------------------------------- /pretrain/scripts/train_synthia4d_temporal.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/scripts/train_synthia4d_temporal.sh -------------------------------------------------------------------------------- /pretrain/splits/scannet/scannetv2_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/splits/scannet/scannetv2_test.txt -------------------------------------------------------------------------------- /pretrain/splits/scannet/scannetv2_test_oo.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/splits/scannet/scannetv2_test_oo.txt -------------------------------------------------------------------------------- /pretrain/splits/scannet/scannetv2_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/splits/scannet/scannetv2_train.txt -------------------------------------------------------------------------------- /pretrain/splits/scannet/scannetv2_train_2.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/splits/scannet/scannetv2_train_2.txt -------------------------------------------------------------------------------- /pretrain/splits/scannet/scannetv2_trainval.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/splits/scannet/scannetv2_trainval.txt -------------------------------------------------------------------------------- /pretrain/splits/scannet/scannetv2_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/pretrain/splits/scannet/scannetv2_val.txt -------------------------------------------------------------------------------- /requirements.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/OPEN-AIR-SUN/Viewpoint-Bottleneck/HEAD/requirements.sh --------------------------------------------------------------------------------