├── README.md ├── figs ├── SMPointSeg.png ├── SMSR.png ├── SMStereo.png ├── SM_conv.png ├── SM_generation.png ├── brain.png ├── results_pc_1.png ├── results_pc_2.png ├── results_pc_3.png ├── results_pc_4.png ├── results_sr_1.png ├── results_sr_2.png ├── results_stereo_1.png ├── results_stereo_2.png └── sparsity.png └── point_cloud_semantic_segmentation ├── 6_fold_S3DIS.py ├── 6_fold_S3DIS.sh ├── README.md ├── compile_op.sh ├── data ├── S3DIS.py └── SemanticKitti.py ├── data_prepare_S3DIS.sh ├── data_prepare_SemanticKITTI.sh ├── inference_S3DIS.py ├── inference_S3DIS.sh ├── inference_SemanticKITTI.py ├── inference_SemanticKITTI.sh ├── model ├── SMPointSeg.py └── modules.py ├── runs └── S3DIS │ ├── Area1.pth │ ├── Area2.pth │ ├── Area3.pth │ ├── Area4.pth │ ├── Area5.pth │ └── Area6.pth ├── train_S3DIS.py ├── train_S3DIS.sh ├── train_SemanticKITTI.py ├── train_SemanticKITTI.sh └── utils ├── cpp_wrappers ├── compile_wrappers.sh ├── cpp_subsampling │ ├── build.bat │ ├── grid_subsampling │ │ ├── grid_subsampling.cpp │ │ └── grid_subsampling.h │ ├── setup.py │ └── wrapper.cpp └── cpp_utils │ └── cloud │ ├── cloud.cpp │ └── cloud.h ├── data_prepare_S3DIS.py ├── data_prepare_SemanticKITTI.py ├── helper_ply.py ├── helper_tool.py ├── meta ├── anno_paths.txt └── class_names.txt ├── metrics.py ├── nearest_neighbors ├── KDTreeTableAdaptor.h ├── knn.cpp ├── knn.pyx ├── knn_.cxx ├── knn_.h ├── nanoflann.hpp ├── setup.py └── test.py ├── option_S3DIS.py ├── option_SemanticKITTI.py └── semantic-kitti.yaml /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/README.md -------------------------------------------------------------------------------- /figs/SMPointSeg.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/figs/SMPointSeg.png -------------------------------------------------------------------------------- /figs/SMSR.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/figs/SMSR.png -------------------------------------------------------------------------------- /figs/SMStereo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/figs/SMStereo.png -------------------------------------------------------------------------------- /figs/SM_conv.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/figs/SM_conv.png -------------------------------------------------------------------------------- /figs/SM_generation.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/figs/SM_generation.png -------------------------------------------------------------------------------- /figs/brain.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/figs/brain.png -------------------------------------------------------------------------------- /figs/results_pc_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/figs/results_pc_1.png -------------------------------------------------------------------------------- /figs/results_pc_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/figs/results_pc_2.png -------------------------------------------------------------------------------- /figs/results_pc_3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/figs/results_pc_3.png -------------------------------------------------------------------------------- /figs/results_pc_4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/figs/results_pc_4.png -------------------------------------------------------------------------------- /figs/results_sr_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/figs/results_sr_1.png -------------------------------------------------------------------------------- /figs/results_sr_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/figs/results_sr_2.png -------------------------------------------------------------------------------- /figs/results_stereo_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/figs/results_stereo_1.png -------------------------------------------------------------------------------- /figs/results_stereo_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/figs/results_stereo_2.png -------------------------------------------------------------------------------- /figs/sparsity.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/figs/sparsity.png -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/6_fold_S3DIS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/6_fold_S3DIS.py -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/6_fold_S3DIS.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/6_fold_S3DIS.sh -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/README.md -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/compile_op.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/compile_op.sh -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/data/S3DIS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/data/S3DIS.py -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/data/SemanticKitti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/data/SemanticKitti.py -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/data_prepare_S3DIS.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/data_prepare_S3DIS.sh -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/data_prepare_SemanticKITTI.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/data_prepare_SemanticKITTI.sh -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/inference_S3DIS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/inference_S3DIS.py -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/inference_S3DIS.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/inference_S3DIS.sh -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/inference_SemanticKITTI.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/inference_SemanticKITTI.py -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/inference_SemanticKITTI.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/inference_SemanticKITTI.sh -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/model/SMPointSeg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/model/SMPointSeg.py -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/model/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/model/modules.py -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/runs/S3DIS/Area1.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/runs/S3DIS/Area1.pth -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/runs/S3DIS/Area2.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/runs/S3DIS/Area2.pth -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/runs/S3DIS/Area3.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/runs/S3DIS/Area3.pth -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/runs/S3DIS/Area4.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/runs/S3DIS/Area4.pth -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/runs/S3DIS/Area5.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/runs/S3DIS/Area5.pth -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/runs/S3DIS/Area6.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/runs/S3DIS/Area6.pth -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/train_S3DIS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/train_S3DIS.py -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/train_S3DIS.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/train_S3DIS.sh -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/train_SemanticKITTI.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/train_SemanticKITTI.py -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/train_SemanticKITTI.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/train_SemanticKITTI.sh -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/cpp_wrappers/compile_wrappers.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/utils/cpp_wrappers/compile_wrappers.sh -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/cpp_wrappers/cpp_subsampling/build.bat: -------------------------------------------------------------------------------- 1 | @echo off 2 | py setup.py build_ext --inplace 3 | 4 | 5 | pause -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/cpp_wrappers/cpp_subsampling/grid_subsampling/grid_subsampling.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/utils/cpp_wrappers/cpp_subsampling/grid_subsampling/grid_subsampling.cpp -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/cpp_wrappers/cpp_subsampling/grid_subsampling/grid_subsampling.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/utils/cpp_wrappers/cpp_subsampling/grid_subsampling/grid_subsampling.h -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/cpp_wrappers/cpp_subsampling/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/utils/cpp_wrappers/cpp_subsampling/setup.py -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/cpp_wrappers/cpp_subsampling/wrapper.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/utils/cpp_wrappers/cpp_subsampling/wrapper.cpp -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/cpp_wrappers/cpp_utils/cloud/cloud.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/utils/cpp_wrappers/cpp_utils/cloud/cloud.cpp -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/cpp_wrappers/cpp_utils/cloud/cloud.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/utils/cpp_wrappers/cpp_utils/cloud/cloud.h -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/data_prepare_S3DIS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/utils/data_prepare_S3DIS.py -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/data_prepare_SemanticKITTI.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/utils/data_prepare_SemanticKITTI.py -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/helper_ply.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/utils/helper_ply.py -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/helper_tool.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/utils/helper_tool.py -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/meta/anno_paths.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/utils/meta/anno_paths.txt -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/meta/class_names.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/utils/meta/class_names.txt -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/utils/metrics.py -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/nearest_neighbors/KDTreeTableAdaptor.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/utils/nearest_neighbors/KDTreeTableAdaptor.h -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/nearest_neighbors/knn.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/utils/nearest_neighbors/knn.cpp -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/nearest_neighbors/knn.pyx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/utils/nearest_neighbors/knn.pyx -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/nearest_neighbors/knn_.cxx: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/utils/nearest_neighbors/knn_.cxx -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/nearest_neighbors/knn_.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/utils/nearest_neighbors/knn_.h -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/nearest_neighbors/nanoflann.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/utils/nearest_neighbors/nanoflann.hpp -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/nearest_neighbors/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/utils/nearest_neighbors/setup.py -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/nearest_neighbors/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/utils/nearest_neighbors/test.py -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/option_S3DIS.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/utils/option_S3DIS.py -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/option_SemanticKITTI.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/utils/option_SemanticKITTI.py -------------------------------------------------------------------------------- /point_cloud_semantic_segmentation/utils/semantic-kitti.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/LongguangWang/SparseMask/HEAD/point_cloud_semantic_segmentation/utils/semantic-kitti.yaml --------------------------------------------------------------------------------