├── .gitignore ├── LICENSE ├── README.md ├── assets ├── cloud_bin_21.pth ├── cloud_bin_34.pth ├── demo.png ├── dist_thresh.txt ├── fix_knn_feats.png ├── inlier_thresh.txt ├── results.png └── teaser_predator.jpg ├── common ├── colors.py ├── math │ ├── random.py │ ├── se3.py │ └── so3.py ├── math_torch │ └── se3.py ├── misc.py └── torch.py ├── configs ├── benchmarks │ ├── 3DLoMatch │ │ ├── 7-scenes-redkitchen │ │ │ ├── gt.info │ │ │ ├── gt.log │ │ │ └── gt_overlap.log │ │ ├── sun3d-home_at-home_at_scan1_2013_jan_1 │ │ │ ├── gt.info │ │ │ ├── gt.log │ │ │ └── gt_overlap.log │ │ ├── sun3d-home_md-home_md_scan9_2012_sep_30 │ │ │ ├── gt.info │ │ │ ├── gt.log │ │ │ └── gt_overlap.log │ │ ├── sun3d-hotel_uc-scan3 │ │ │ ├── gt.info │ │ │ ├── gt.log │ │ │ └── gt_overlap.log │ │ ├── sun3d-hotel_umd-maryland_hotel1 │ │ │ ├── gt.info │ │ │ ├── gt.log │ │ │ └── gt_overlap.log │ │ ├── sun3d-hotel_umd-maryland_hotel3 │ │ │ ├── gt.info │ │ │ ├── gt.log │ │ │ └── gt_overlap.log │ │ ├── sun3d-mit_76_studyroom-76-1studyroom2 │ │ │ ├── gt.info │ │ │ ├── gt.log │ │ │ └── gt_overlap.log │ │ └── sun3d-mit_lab_hj-lab_hj_tea_nov_2_2012_scan1_erika │ │ │ ├── gt.info │ │ │ ├── gt.log │ │ │ └── gt_overlap.log │ └── 3DMatch │ │ ├── 7-scenes-redkitchen │ │ ├── gt.info │ │ ├── gt.log │ │ └── gt_overlap.log │ │ ├── sun3d-home_at-home_at_scan1_2013_jan_1 │ │ ├── gt.info │ │ ├── gt.log │ │ └── gt_overlap.log │ │ ├── sun3d-home_md-home_md_scan9_2012_sep_30 │ │ ├── gt.info │ │ ├── gt.log │ │ └── gt_overlap.log │ │ ├── sun3d-hotel_uc-scan3 │ │ ├── gt.info │ │ ├── gt.log │ │ └── gt_overlap.log │ │ ├── sun3d-hotel_umd-maryland_hotel1 │ │ ├── gt.info │ │ ├── gt.log │ │ └── gt_overlap.log │ │ ├── sun3d-hotel_umd-maryland_hotel3 │ │ ├── gt.info │ │ ├── gt.log │ │ └── gt_overlap.log │ │ ├── sun3d-mit_76_studyroom-76-1studyroom2 │ │ ├── gt.info │ │ ├── gt.log │ │ └── gt_overlap.log │ │ └── sun3d-mit_lab_hj-lab_hj_tea_nov_2_2012_scan1_erika │ │ ├── gt.info │ │ ├── gt.log │ │ └── gt_overlap.log ├── indoor │ ├── 3DLoMatch.pkl │ ├── 3DMatch.pkl │ ├── train_3dmatch.txt │ ├── train_info.pkl │ ├── val_3dmatch.txt │ └── val_info.pkl ├── kitti │ ├── test_kitti.txt │ ├── train_kitti.txt │ └── val_kitti.txt ├── modelnet │ ├── modelnet40_all.txt │ ├── modelnet40_half1.txt │ └── modelnet40_half2.txt ├── models.py ├── test │ ├── indoor.yaml │ ├── kitti.yaml │ └── modelnet.yaml └── train │ ├── indoor.yaml │ ├── kitti.yaml │ └── modelnet.yaml ├── cpp_wrappers ├── compile_wrappers.sh ├── cpp_neighbors │ ├── build.bat │ ├── neighbors │ │ ├── neighbors.cpp │ │ └── neighbors.h │ ├── setup.py │ └── wrapper.cpp ├── cpp_subsampling │ ├── build.bat │ ├── grid_subsampling │ │ ├── grid_subsampling.cpp │ │ └── grid_subsampling.h │ ├── setup.py │ └── wrapper.cpp └── cpp_utils │ ├── cloud │ ├── cloud.cpp │ └── cloud.h │ └── nanoflann │ └── nanoflann.hpp ├── datasets ├── __init__.py ├── dataloader.py ├── indoor.py ├── kitti.py ├── modelnet.py └── transforms.py ├── kernels ├── dispositions │ └── k_015_center_3D.ply └── kernel_points.py ├── lib ├── __init__.py ├── benchmark.py ├── benchmark_utils.py ├── loss.py ├── ply.py ├── tester.py ├── timer.py ├── trainer.py └── utils.py ├── main.py ├── models ├── __init__.py ├── architectures.py ├── blocks.py └── gcn.py ├── requirements.txt └── scripts ├── cal_overlap.py ├── demo.py ├── download_data_weight.sh └── evaluate_predator.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/README.md -------------------------------------------------------------------------------- /assets/cloud_bin_21.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/assets/cloud_bin_21.pth -------------------------------------------------------------------------------- /assets/cloud_bin_34.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/assets/cloud_bin_34.pth -------------------------------------------------------------------------------- /assets/demo.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/assets/demo.png -------------------------------------------------------------------------------- /assets/dist_thresh.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/assets/dist_thresh.txt -------------------------------------------------------------------------------- /assets/fix_knn_feats.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/assets/fix_knn_feats.png -------------------------------------------------------------------------------- /assets/inlier_thresh.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/assets/inlier_thresh.txt -------------------------------------------------------------------------------- /assets/results.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/assets/results.png -------------------------------------------------------------------------------- /assets/teaser_predator.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/assets/teaser_predator.jpg -------------------------------------------------------------------------------- /common/colors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/common/colors.py -------------------------------------------------------------------------------- /common/math/random.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/common/math/random.py -------------------------------------------------------------------------------- /common/math/se3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/common/math/se3.py -------------------------------------------------------------------------------- /common/math/so3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/common/math/so3.py -------------------------------------------------------------------------------- /common/math_torch/se3.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/common/math_torch/se3.py -------------------------------------------------------------------------------- /common/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/common/misc.py -------------------------------------------------------------------------------- /common/torch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/common/torch.py -------------------------------------------------------------------------------- /configs/benchmarks/3DLoMatch/7-scenes-redkitchen/gt.info: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DLoMatch/7-scenes-redkitchen/gt.info -------------------------------------------------------------------------------- /configs/benchmarks/3DLoMatch/7-scenes-redkitchen/gt.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DLoMatch/7-scenes-redkitchen/gt.log -------------------------------------------------------------------------------- /configs/benchmarks/3DLoMatch/7-scenes-redkitchen/gt_overlap.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DLoMatch/7-scenes-redkitchen/gt_overlap.log -------------------------------------------------------------------------------- /configs/benchmarks/3DLoMatch/sun3d-home_at-home_at_scan1_2013_jan_1/gt.info: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DLoMatch/sun3d-home_at-home_at_scan1_2013_jan_1/gt.info -------------------------------------------------------------------------------- /configs/benchmarks/3DLoMatch/sun3d-home_at-home_at_scan1_2013_jan_1/gt.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DLoMatch/sun3d-home_at-home_at_scan1_2013_jan_1/gt.log -------------------------------------------------------------------------------- /configs/benchmarks/3DLoMatch/sun3d-home_at-home_at_scan1_2013_jan_1/gt_overlap.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DLoMatch/sun3d-home_at-home_at_scan1_2013_jan_1/gt_overlap.log -------------------------------------------------------------------------------- /configs/benchmarks/3DLoMatch/sun3d-home_md-home_md_scan9_2012_sep_30/gt.info: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DLoMatch/sun3d-home_md-home_md_scan9_2012_sep_30/gt.info -------------------------------------------------------------------------------- /configs/benchmarks/3DLoMatch/sun3d-home_md-home_md_scan9_2012_sep_30/gt.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DLoMatch/sun3d-home_md-home_md_scan9_2012_sep_30/gt.log -------------------------------------------------------------------------------- /configs/benchmarks/3DLoMatch/sun3d-home_md-home_md_scan9_2012_sep_30/gt_overlap.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DLoMatch/sun3d-home_md-home_md_scan9_2012_sep_30/gt_overlap.log -------------------------------------------------------------------------------- /configs/benchmarks/3DLoMatch/sun3d-hotel_uc-scan3/gt.info: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DLoMatch/sun3d-hotel_uc-scan3/gt.info -------------------------------------------------------------------------------- /configs/benchmarks/3DLoMatch/sun3d-hotel_uc-scan3/gt.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DLoMatch/sun3d-hotel_uc-scan3/gt.log -------------------------------------------------------------------------------- /configs/benchmarks/3DLoMatch/sun3d-hotel_uc-scan3/gt_overlap.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DLoMatch/sun3d-hotel_uc-scan3/gt_overlap.log -------------------------------------------------------------------------------- /configs/benchmarks/3DLoMatch/sun3d-hotel_umd-maryland_hotel1/gt.info: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DLoMatch/sun3d-hotel_umd-maryland_hotel1/gt.info -------------------------------------------------------------------------------- /configs/benchmarks/3DLoMatch/sun3d-hotel_umd-maryland_hotel1/gt.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DLoMatch/sun3d-hotel_umd-maryland_hotel1/gt.log -------------------------------------------------------------------------------- /configs/benchmarks/3DLoMatch/sun3d-hotel_umd-maryland_hotel1/gt_overlap.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DLoMatch/sun3d-hotel_umd-maryland_hotel1/gt_overlap.log -------------------------------------------------------------------------------- /configs/benchmarks/3DLoMatch/sun3d-hotel_umd-maryland_hotel3/gt.info: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DLoMatch/sun3d-hotel_umd-maryland_hotel3/gt.info -------------------------------------------------------------------------------- /configs/benchmarks/3DLoMatch/sun3d-hotel_umd-maryland_hotel3/gt.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DLoMatch/sun3d-hotel_umd-maryland_hotel3/gt.log -------------------------------------------------------------------------------- /configs/benchmarks/3DLoMatch/sun3d-hotel_umd-maryland_hotel3/gt_overlap.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DLoMatch/sun3d-hotel_umd-maryland_hotel3/gt_overlap.log -------------------------------------------------------------------------------- /configs/benchmarks/3DLoMatch/sun3d-mit_76_studyroom-76-1studyroom2/gt.info: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DLoMatch/sun3d-mit_76_studyroom-76-1studyroom2/gt.info -------------------------------------------------------------------------------- /configs/benchmarks/3DLoMatch/sun3d-mit_76_studyroom-76-1studyroom2/gt.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DLoMatch/sun3d-mit_76_studyroom-76-1studyroom2/gt.log -------------------------------------------------------------------------------- /configs/benchmarks/3DLoMatch/sun3d-mit_76_studyroom-76-1studyroom2/gt_overlap.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DLoMatch/sun3d-mit_76_studyroom-76-1studyroom2/gt_overlap.log -------------------------------------------------------------------------------- /configs/benchmarks/3DLoMatch/sun3d-mit_lab_hj-lab_hj_tea_nov_2_2012_scan1_erika/gt.info: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DLoMatch/sun3d-mit_lab_hj-lab_hj_tea_nov_2_2012_scan1_erika/gt.info -------------------------------------------------------------------------------- /configs/benchmarks/3DLoMatch/sun3d-mit_lab_hj-lab_hj_tea_nov_2_2012_scan1_erika/gt.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DLoMatch/sun3d-mit_lab_hj-lab_hj_tea_nov_2_2012_scan1_erika/gt.log -------------------------------------------------------------------------------- /configs/benchmarks/3DLoMatch/sun3d-mit_lab_hj-lab_hj_tea_nov_2_2012_scan1_erika/gt_overlap.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DLoMatch/sun3d-mit_lab_hj-lab_hj_tea_nov_2_2012_scan1_erika/gt_overlap.log -------------------------------------------------------------------------------- /configs/benchmarks/3DMatch/7-scenes-redkitchen/gt.info: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DMatch/7-scenes-redkitchen/gt.info -------------------------------------------------------------------------------- /configs/benchmarks/3DMatch/7-scenes-redkitchen/gt.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DMatch/7-scenes-redkitchen/gt.log -------------------------------------------------------------------------------- /configs/benchmarks/3DMatch/7-scenes-redkitchen/gt_overlap.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DMatch/7-scenes-redkitchen/gt_overlap.log -------------------------------------------------------------------------------- /configs/benchmarks/3DMatch/sun3d-home_at-home_at_scan1_2013_jan_1/gt.info: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DMatch/sun3d-home_at-home_at_scan1_2013_jan_1/gt.info -------------------------------------------------------------------------------- /configs/benchmarks/3DMatch/sun3d-home_at-home_at_scan1_2013_jan_1/gt.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DMatch/sun3d-home_at-home_at_scan1_2013_jan_1/gt.log -------------------------------------------------------------------------------- /configs/benchmarks/3DMatch/sun3d-home_at-home_at_scan1_2013_jan_1/gt_overlap.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DMatch/sun3d-home_at-home_at_scan1_2013_jan_1/gt_overlap.log -------------------------------------------------------------------------------- /configs/benchmarks/3DMatch/sun3d-home_md-home_md_scan9_2012_sep_30/gt.info: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DMatch/sun3d-home_md-home_md_scan9_2012_sep_30/gt.info -------------------------------------------------------------------------------- /configs/benchmarks/3DMatch/sun3d-home_md-home_md_scan9_2012_sep_30/gt.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DMatch/sun3d-home_md-home_md_scan9_2012_sep_30/gt.log -------------------------------------------------------------------------------- /configs/benchmarks/3DMatch/sun3d-home_md-home_md_scan9_2012_sep_30/gt_overlap.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DMatch/sun3d-home_md-home_md_scan9_2012_sep_30/gt_overlap.log -------------------------------------------------------------------------------- /configs/benchmarks/3DMatch/sun3d-hotel_uc-scan3/gt.info: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DMatch/sun3d-hotel_uc-scan3/gt.info -------------------------------------------------------------------------------- /configs/benchmarks/3DMatch/sun3d-hotel_uc-scan3/gt.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DMatch/sun3d-hotel_uc-scan3/gt.log -------------------------------------------------------------------------------- /configs/benchmarks/3DMatch/sun3d-hotel_uc-scan3/gt_overlap.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DMatch/sun3d-hotel_uc-scan3/gt_overlap.log -------------------------------------------------------------------------------- /configs/benchmarks/3DMatch/sun3d-hotel_umd-maryland_hotel1/gt.info: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DMatch/sun3d-hotel_umd-maryland_hotel1/gt.info -------------------------------------------------------------------------------- /configs/benchmarks/3DMatch/sun3d-hotel_umd-maryland_hotel1/gt.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DMatch/sun3d-hotel_umd-maryland_hotel1/gt.log -------------------------------------------------------------------------------- /configs/benchmarks/3DMatch/sun3d-hotel_umd-maryland_hotel1/gt_overlap.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DMatch/sun3d-hotel_umd-maryland_hotel1/gt_overlap.log -------------------------------------------------------------------------------- /configs/benchmarks/3DMatch/sun3d-hotel_umd-maryland_hotel3/gt.info: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DMatch/sun3d-hotel_umd-maryland_hotel3/gt.info -------------------------------------------------------------------------------- /configs/benchmarks/3DMatch/sun3d-hotel_umd-maryland_hotel3/gt.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DMatch/sun3d-hotel_umd-maryland_hotel3/gt.log -------------------------------------------------------------------------------- /configs/benchmarks/3DMatch/sun3d-hotel_umd-maryland_hotel3/gt_overlap.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DMatch/sun3d-hotel_umd-maryland_hotel3/gt_overlap.log -------------------------------------------------------------------------------- /configs/benchmarks/3DMatch/sun3d-mit_76_studyroom-76-1studyroom2/gt.info: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DMatch/sun3d-mit_76_studyroom-76-1studyroom2/gt.info -------------------------------------------------------------------------------- /configs/benchmarks/3DMatch/sun3d-mit_76_studyroom-76-1studyroom2/gt.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DMatch/sun3d-mit_76_studyroom-76-1studyroom2/gt.log -------------------------------------------------------------------------------- /configs/benchmarks/3DMatch/sun3d-mit_76_studyroom-76-1studyroom2/gt_overlap.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DMatch/sun3d-mit_76_studyroom-76-1studyroom2/gt_overlap.log -------------------------------------------------------------------------------- /configs/benchmarks/3DMatch/sun3d-mit_lab_hj-lab_hj_tea_nov_2_2012_scan1_erika/gt.info: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DMatch/sun3d-mit_lab_hj-lab_hj_tea_nov_2_2012_scan1_erika/gt.info -------------------------------------------------------------------------------- /configs/benchmarks/3DMatch/sun3d-mit_lab_hj-lab_hj_tea_nov_2_2012_scan1_erika/gt.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DMatch/sun3d-mit_lab_hj-lab_hj_tea_nov_2_2012_scan1_erika/gt.log -------------------------------------------------------------------------------- /configs/benchmarks/3DMatch/sun3d-mit_lab_hj-lab_hj_tea_nov_2_2012_scan1_erika/gt_overlap.log: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/benchmarks/3DMatch/sun3d-mit_lab_hj-lab_hj_tea_nov_2_2012_scan1_erika/gt_overlap.log -------------------------------------------------------------------------------- /configs/indoor/3DLoMatch.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/indoor/3DLoMatch.pkl -------------------------------------------------------------------------------- /configs/indoor/3DMatch.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/indoor/3DMatch.pkl -------------------------------------------------------------------------------- /configs/indoor/train_3dmatch.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/indoor/train_3dmatch.txt -------------------------------------------------------------------------------- /configs/indoor/train_info.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/indoor/train_info.pkl -------------------------------------------------------------------------------- /configs/indoor/val_3dmatch.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/indoor/val_3dmatch.txt -------------------------------------------------------------------------------- /configs/indoor/val_info.pkl: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/indoor/val_info.pkl -------------------------------------------------------------------------------- /configs/kitti/test_kitti.txt: -------------------------------------------------------------------------------- 1 | 8 2 | 9 3 | 10 -------------------------------------------------------------------------------- /configs/kitti/train_kitti.txt: -------------------------------------------------------------------------------- 1 | 0 2 | 1 3 | 2 4 | 3 5 | 4 6 | 5 -------------------------------------------------------------------------------- /configs/kitti/val_kitti.txt: -------------------------------------------------------------------------------- 1 | 6 2 | 7 -------------------------------------------------------------------------------- /configs/modelnet/modelnet40_all.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/modelnet/modelnet40_all.txt -------------------------------------------------------------------------------- /configs/modelnet/modelnet40_half1.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/modelnet/modelnet40_half1.txt -------------------------------------------------------------------------------- /configs/modelnet/modelnet40_half2.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/modelnet/modelnet40_half2.txt -------------------------------------------------------------------------------- /configs/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/models.py -------------------------------------------------------------------------------- /configs/test/indoor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/test/indoor.yaml -------------------------------------------------------------------------------- /configs/test/kitti.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/test/kitti.yaml -------------------------------------------------------------------------------- /configs/test/modelnet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/test/modelnet.yaml -------------------------------------------------------------------------------- /configs/train/indoor.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/train/indoor.yaml -------------------------------------------------------------------------------- /configs/train/kitti.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/train/kitti.yaml -------------------------------------------------------------------------------- /configs/train/modelnet.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/configs/train/modelnet.yaml -------------------------------------------------------------------------------- /cpp_wrappers/compile_wrappers.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/cpp_wrappers/compile_wrappers.sh -------------------------------------------------------------------------------- /cpp_wrappers/cpp_neighbors/build.bat: -------------------------------------------------------------------------------- 1 | @echo off 2 | py setup.py build_ext --inplace 3 | 4 | 5 | pause -------------------------------------------------------------------------------- /cpp_wrappers/cpp_neighbors/neighbors/neighbors.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/cpp_wrappers/cpp_neighbors/neighbors/neighbors.cpp -------------------------------------------------------------------------------- /cpp_wrappers/cpp_neighbors/neighbors/neighbors.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/cpp_wrappers/cpp_neighbors/neighbors/neighbors.h -------------------------------------------------------------------------------- /cpp_wrappers/cpp_neighbors/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/cpp_wrappers/cpp_neighbors/setup.py -------------------------------------------------------------------------------- /cpp_wrappers/cpp_neighbors/wrapper.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/cpp_wrappers/cpp_neighbors/wrapper.cpp -------------------------------------------------------------------------------- /cpp_wrappers/cpp_subsampling/build.bat: -------------------------------------------------------------------------------- 1 | @echo off 2 | py setup.py build_ext --inplace 3 | 4 | 5 | pause -------------------------------------------------------------------------------- /cpp_wrappers/cpp_subsampling/grid_subsampling/grid_subsampling.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/cpp_wrappers/cpp_subsampling/grid_subsampling/grid_subsampling.cpp -------------------------------------------------------------------------------- /cpp_wrappers/cpp_subsampling/grid_subsampling/grid_subsampling.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/cpp_wrappers/cpp_subsampling/grid_subsampling/grid_subsampling.h -------------------------------------------------------------------------------- /cpp_wrappers/cpp_subsampling/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/cpp_wrappers/cpp_subsampling/setup.py -------------------------------------------------------------------------------- /cpp_wrappers/cpp_subsampling/wrapper.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/cpp_wrappers/cpp_subsampling/wrapper.cpp -------------------------------------------------------------------------------- /cpp_wrappers/cpp_utils/cloud/cloud.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/cpp_wrappers/cpp_utils/cloud/cloud.cpp -------------------------------------------------------------------------------- /cpp_wrappers/cpp_utils/cloud/cloud.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/cpp_wrappers/cpp_utils/cloud/cloud.h -------------------------------------------------------------------------------- /cpp_wrappers/cpp_utils/nanoflann/nanoflann.hpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/cpp_wrappers/cpp_utils/nanoflann/nanoflann.hpp -------------------------------------------------------------------------------- /datasets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /datasets/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/datasets/dataloader.py -------------------------------------------------------------------------------- /datasets/indoor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/datasets/indoor.py -------------------------------------------------------------------------------- /datasets/kitti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/datasets/kitti.py -------------------------------------------------------------------------------- /datasets/modelnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/datasets/modelnet.py -------------------------------------------------------------------------------- /datasets/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/datasets/transforms.py -------------------------------------------------------------------------------- /kernels/dispositions/k_015_center_3D.ply: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/kernels/dispositions/k_015_center_3D.ply -------------------------------------------------------------------------------- /kernels/kernel_points.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/kernels/kernel_points.py -------------------------------------------------------------------------------- /lib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /lib/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/lib/benchmark.py -------------------------------------------------------------------------------- /lib/benchmark_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/lib/benchmark_utils.py -------------------------------------------------------------------------------- /lib/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/lib/loss.py -------------------------------------------------------------------------------- /lib/ply.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/lib/ply.py -------------------------------------------------------------------------------- /lib/tester.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/lib/tester.py -------------------------------------------------------------------------------- /lib/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/lib/timer.py -------------------------------------------------------------------------------- /lib/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/lib/trainer.py -------------------------------------------------------------------------------- /lib/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/lib/utils.py -------------------------------------------------------------------------------- /main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/main.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/architectures.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/models/architectures.py -------------------------------------------------------------------------------- /models/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/models/blocks.py -------------------------------------------------------------------------------- /models/gcn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/models/gcn.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/cal_overlap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/scripts/cal_overlap.py -------------------------------------------------------------------------------- /scripts/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/scripts/demo.py -------------------------------------------------------------------------------- /scripts/download_data_weight.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/scripts/download_data_weight.sh -------------------------------------------------------------------------------- /scripts/evaluate_predator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/prs-eth/OverlapPredator/HEAD/scripts/evaluate_predator.py --------------------------------------------------------------------------------