├── .gitignore ├── LICENSE ├── README.md ├── data ├── scannetv2.txt ├── scannetv2_enet.pth ├── scannetv2_test.txt ├── scannetv2_train.txt └── scannetv2_val.txt ├── img ├── snapshot.png └── snapshot_pred.png ├── lib ├── __init__.py ├── config.py ├── dataset.py ├── enet.py ├── loss.py ├── pc_util.py ├── projection.py ├── scene_util.py ├── solver.py └── utils.py ├── pointnet2 ├── _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 ├── _version.py ├── pointnet2_modules.py ├── pointnet2_semseg.py ├── pointnet2_test.py ├── pointnet2_utils.py ├── pytorch_utils.py ├── setup.py └── src │ ├── ball_query.cpp │ ├── ball_query_gpu.cu │ ├── ball_query_gpu.h │ ├── cuda_utils.h │ ├── group_points.cpp │ ├── group_points_gpu.cu │ ├── group_points_gpu.h │ ├── interpolate.cpp │ ├── interpolate_gpu.cu │ ├── interpolate_gpu.h │ ├── pointnet2_api.cpp │ ├── sampling.cpp │ ├── sampling_gpu.cu │ └── sampling_gpu.h ├── preprocessing ├── collect_scannet_scenes.py ├── scannet_util.py ├── scannetv2-labels.combined.tsv └── visualize_prep_scene.py ├── requirements.txt ├── scripts ├── compute_multiview_features.py ├── compute_multiview_projection.py ├── eval.py ├── project_multiview_features.py ├── train.py └── visualize.py └── slurm ├── compute_multiview_features.job ├── compute_multiview_projection.job ├── eval.job ├── prep.job ├── project_multiview_features.job ├── train.job └── visualize.job /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/README.md -------------------------------------------------------------------------------- /data/scannetv2.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/data/scannetv2.txt -------------------------------------------------------------------------------- /data/scannetv2_enet.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/data/scannetv2_enet.pth -------------------------------------------------------------------------------- /data/scannetv2_test.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/data/scannetv2_test.txt -------------------------------------------------------------------------------- /data/scannetv2_train.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/data/scannetv2_train.txt -------------------------------------------------------------------------------- /data/scannetv2_val.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/data/scannetv2_val.txt -------------------------------------------------------------------------------- /img/snapshot.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/img/snapshot.png -------------------------------------------------------------------------------- /img/snapshot_pred.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/img/snapshot_pred.png -------------------------------------------------------------------------------- /lib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/lib/config.py -------------------------------------------------------------------------------- /lib/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/lib/dataset.py -------------------------------------------------------------------------------- /lib/enet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/lib/enet.py -------------------------------------------------------------------------------- /lib/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/lib/loss.py -------------------------------------------------------------------------------- /lib/pc_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/lib/pc_util.py -------------------------------------------------------------------------------- /lib/projection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/lib/projection.py -------------------------------------------------------------------------------- /lib/scene_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/lib/scene_util.py -------------------------------------------------------------------------------- /lib/solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/lib/solver.py -------------------------------------------------------------------------------- /lib/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/lib/utils.py -------------------------------------------------------------------------------- /pointnet2/_ext_src/include/ball_query.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/_ext_src/include/ball_query.h -------------------------------------------------------------------------------- /pointnet2/_ext_src/include/cuda_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/_ext_src/include/cuda_utils.h -------------------------------------------------------------------------------- /pointnet2/_ext_src/include/group_points.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/_ext_src/include/group_points.h -------------------------------------------------------------------------------- /pointnet2/_ext_src/include/interpolate.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/_ext_src/include/interpolate.h -------------------------------------------------------------------------------- /pointnet2/_ext_src/include/sampling.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/_ext_src/include/sampling.h -------------------------------------------------------------------------------- /pointnet2/_ext_src/include/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/_ext_src/include/utils.h -------------------------------------------------------------------------------- /pointnet2/_ext_src/src/ball_query.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/_ext_src/src/ball_query.cpp -------------------------------------------------------------------------------- /pointnet2/_ext_src/src/ball_query_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/_ext_src/src/ball_query_gpu.cu -------------------------------------------------------------------------------- /pointnet2/_ext_src/src/bindings.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/_ext_src/src/bindings.cpp -------------------------------------------------------------------------------- /pointnet2/_ext_src/src/group_points.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/_ext_src/src/group_points.cpp -------------------------------------------------------------------------------- /pointnet2/_ext_src/src/group_points_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/_ext_src/src/group_points_gpu.cu -------------------------------------------------------------------------------- /pointnet2/_ext_src/src/interpolate.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/_ext_src/src/interpolate.cpp -------------------------------------------------------------------------------- /pointnet2/_ext_src/src/interpolate_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/_ext_src/src/interpolate_gpu.cu -------------------------------------------------------------------------------- /pointnet2/_ext_src/src/sampling.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/_ext_src/src/sampling.cpp -------------------------------------------------------------------------------- /pointnet2/_ext_src/src/sampling_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/_ext_src/src/sampling_gpu.cu -------------------------------------------------------------------------------- /pointnet2/_version.py: -------------------------------------------------------------------------------- 1 | __version__ = "3.0.0" 2 | -------------------------------------------------------------------------------- /pointnet2/pointnet2_modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/pointnet2_modules.py -------------------------------------------------------------------------------- /pointnet2/pointnet2_semseg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/pointnet2_semseg.py -------------------------------------------------------------------------------- /pointnet2/pointnet2_test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/pointnet2_test.py -------------------------------------------------------------------------------- /pointnet2/pointnet2_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/pointnet2_utils.py -------------------------------------------------------------------------------- /pointnet2/pytorch_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/pytorch_utils.py -------------------------------------------------------------------------------- /pointnet2/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/setup.py -------------------------------------------------------------------------------- /pointnet2/src/ball_query.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/src/ball_query.cpp -------------------------------------------------------------------------------- /pointnet2/src/ball_query_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/src/ball_query_gpu.cu -------------------------------------------------------------------------------- /pointnet2/src/ball_query_gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/src/ball_query_gpu.h -------------------------------------------------------------------------------- /pointnet2/src/cuda_utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/src/cuda_utils.h -------------------------------------------------------------------------------- /pointnet2/src/group_points.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/src/group_points.cpp -------------------------------------------------------------------------------- /pointnet2/src/group_points_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/src/group_points_gpu.cu -------------------------------------------------------------------------------- /pointnet2/src/group_points_gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/src/group_points_gpu.h -------------------------------------------------------------------------------- /pointnet2/src/interpolate.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/src/interpolate.cpp -------------------------------------------------------------------------------- /pointnet2/src/interpolate_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/src/interpolate_gpu.cu -------------------------------------------------------------------------------- /pointnet2/src/interpolate_gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/src/interpolate_gpu.h -------------------------------------------------------------------------------- /pointnet2/src/pointnet2_api.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/src/pointnet2_api.cpp -------------------------------------------------------------------------------- /pointnet2/src/sampling.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/src/sampling.cpp -------------------------------------------------------------------------------- /pointnet2/src/sampling_gpu.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/src/sampling_gpu.cu -------------------------------------------------------------------------------- /pointnet2/src/sampling_gpu.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/pointnet2/src/sampling_gpu.h -------------------------------------------------------------------------------- /preprocessing/collect_scannet_scenes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/preprocessing/collect_scannet_scenes.py -------------------------------------------------------------------------------- /preprocessing/scannet_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/preprocessing/scannet_util.py -------------------------------------------------------------------------------- /preprocessing/scannetv2-labels.combined.tsv: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/preprocessing/scannetv2-labels.combined.tsv -------------------------------------------------------------------------------- /preprocessing/visualize_prep_scene.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/preprocessing/visualize_prep_scene.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- 1 | prefetch_generator -------------------------------------------------------------------------------- /scripts/compute_multiview_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/scripts/compute_multiview_features.py -------------------------------------------------------------------------------- /scripts/compute_multiview_projection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/scripts/compute_multiview_projection.py -------------------------------------------------------------------------------- /scripts/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/scripts/eval.py -------------------------------------------------------------------------------- /scripts/project_multiview_features.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/scripts/project_multiview_features.py -------------------------------------------------------------------------------- /scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/scripts/train.py -------------------------------------------------------------------------------- /scripts/visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/scripts/visualize.py -------------------------------------------------------------------------------- /slurm/compute_multiview_features.job: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/slurm/compute_multiview_features.job -------------------------------------------------------------------------------- /slurm/compute_multiview_projection.job: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/slurm/compute_multiview_projection.job -------------------------------------------------------------------------------- /slurm/eval.job: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/slurm/eval.job -------------------------------------------------------------------------------- /slurm/prep.job: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/slurm/prep.job -------------------------------------------------------------------------------- /slurm/project_multiview_features.job: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/slurm/project_multiview_features.job -------------------------------------------------------------------------------- /slurm/train.job: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/slurm/train.job -------------------------------------------------------------------------------- /slurm/visualize.job: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/daveredrum/Pointnet2.ScanNet/HEAD/slurm/visualize.job --------------------------------------------------------------------------------