├── .gitignore ├── LICENSE ├── README.md ├── assets ├── 1min-oral.png ├── comparison-3dmatch.png ├── demo_inputs.png ├── demo_outputs.png ├── dgr.gif ├── frontier.png └── results.npz ├── config.py ├── core ├── __init__.py ├── correspondence.py ├── deep_global_registration.py ├── knn.py ├── loss.py ├── metrics.py ├── registration.py └── trainer.py ├── dataloader ├── base_loader.py ├── data_loaders.py ├── inf_sampler.py ├── kitti_loader.py ├── split │ ├── test_3dmatch.txt │ ├── test_kitti.txt │ ├── test_modelnet40.txt │ ├── test_scan2cad.txt │ ├── train_3dmatch.txt │ ├── train_kitti.txt │ ├── train_modelnet40.txt │ ├── train_scan2cad.txt │ ├── val_3dmatch.txt │ ├── val_kitti.txt │ ├── val_modelnet40.txt │ └── val_scan2cad.txt ├── threedmatch_loader.py └── transforms.py ├── demo.py ├── model ├── __init__.py ├── common.py ├── pyramidnet.py ├── residual_block.py ├── resunet.py └── simpleunet.py ├── requirements.txt ├── scripts ├── analyze_stats.py ├── download_3dmatch.sh ├── test_3dmatch.py ├── test_kitti.py ├── train_3dmatch.sh └── train_kitti.sh ├── train.py └── util ├── __init__.py ├── file.py ├── integration.py ├── pointcloud.py └── timer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/README.md -------------------------------------------------------------------------------- /assets/1min-oral.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/assets/1min-oral.png -------------------------------------------------------------------------------- /assets/comparison-3dmatch.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/assets/comparison-3dmatch.png -------------------------------------------------------------------------------- /assets/demo_inputs.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/assets/demo_inputs.png -------------------------------------------------------------------------------- /assets/demo_outputs.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/assets/demo_outputs.png -------------------------------------------------------------------------------- /assets/dgr.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/assets/dgr.gif -------------------------------------------------------------------------------- /assets/frontier.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/assets/frontier.png -------------------------------------------------------------------------------- /assets/results.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/assets/results.npz -------------------------------------------------------------------------------- /config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/config.py -------------------------------------------------------------------------------- /core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/core/__init__.py -------------------------------------------------------------------------------- /core/correspondence.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/core/correspondence.py -------------------------------------------------------------------------------- /core/deep_global_registration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/core/deep_global_registration.py -------------------------------------------------------------------------------- /core/knn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/core/knn.py -------------------------------------------------------------------------------- /core/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/core/loss.py -------------------------------------------------------------------------------- /core/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/core/metrics.py -------------------------------------------------------------------------------- /core/registration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/core/registration.py -------------------------------------------------------------------------------- /core/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/core/trainer.py -------------------------------------------------------------------------------- /dataloader/base_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/dataloader/base_loader.py -------------------------------------------------------------------------------- /dataloader/data_loaders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/dataloader/data_loaders.py -------------------------------------------------------------------------------- /dataloader/inf_sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/dataloader/inf_sampler.py -------------------------------------------------------------------------------- /dataloader/kitti_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/dataloader/kitti_loader.py -------------------------------------------------------------------------------- /dataloader/split/test_3dmatch.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/dataloader/split/test_3dmatch.txt -------------------------------------------------------------------------------- /dataloader/split/test_kitti.txt: -------------------------------------------------------------------------------- 1 | 8 2 | 9 3 | 10 4 | -------------------------------------------------------------------------------- /dataloader/split/test_modelnet40.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/dataloader/split/test_modelnet40.txt -------------------------------------------------------------------------------- /dataloader/split/test_scan2cad.txt: -------------------------------------------------------------------------------- 1 | full_annotations_clean_test.json 2 | -------------------------------------------------------------------------------- /dataloader/split/train_3dmatch.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/dataloader/split/train_3dmatch.txt -------------------------------------------------------------------------------- /dataloader/split/train_kitti.txt: -------------------------------------------------------------------------------- 1 | 0 2 | 1 3 | 2 4 | 3 5 | 4 6 | 5 7 | -------------------------------------------------------------------------------- /dataloader/split/train_modelnet40.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/dataloader/split/train_modelnet40.txt -------------------------------------------------------------------------------- /dataloader/split/train_scan2cad.txt: -------------------------------------------------------------------------------- 1 | full_annotations_clean_train.json 2 | -------------------------------------------------------------------------------- /dataloader/split/val_3dmatch.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/dataloader/split/val_3dmatch.txt -------------------------------------------------------------------------------- /dataloader/split/val_kitti.txt: -------------------------------------------------------------------------------- 1 | 6 2 | 7 3 | -------------------------------------------------------------------------------- /dataloader/split/val_modelnet40.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/dataloader/split/val_modelnet40.txt -------------------------------------------------------------------------------- /dataloader/split/val_scan2cad.txt: -------------------------------------------------------------------------------- 1 | full_annotations_clean_val.json 2 | -------------------------------------------------------------------------------- /dataloader/threedmatch_loader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/dataloader/threedmatch_loader.py -------------------------------------------------------------------------------- /dataloader/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/dataloader/transforms.py -------------------------------------------------------------------------------- /demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/demo.py -------------------------------------------------------------------------------- /model/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/model/__init__.py -------------------------------------------------------------------------------- /model/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/model/common.py -------------------------------------------------------------------------------- /model/pyramidnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/model/pyramidnet.py -------------------------------------------------------------------------------- /model/residual_block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/model/residual_block.py -------------------------------------------------------------------------------- /model/resunet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/model/resunet.py -------------------------------------------------------------------------------- /model/simpleunet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/model/simpleunet.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/analyze_stats.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/scripts/analyze_stats.py -------------------------------------------------------------------------------- /scripts/download_3dmatch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/scripts/download_3dmatch.sh -------------------------------------------------------------------------------- /scripts/test_3dmatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/scripts/test_3dmatch.py -------------------------------------------------------------------------------- /scripts/test_kitti.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/scripts/test_kitti.py -------------------------------------------------------------------------------- /scripts/train_3dmatch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/scripts/train_3dmatch.sh -------------------------------------------------------------------------------- /scripts/train_kitti.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/scripts/train_kitti.sh -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/train.py -------------------------------------------------------------------------------- /util/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/util/__init__.py -------------------------------------------------------------------------------- /util/file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/util/file.py -------------------------------------------------------------------------------- /util/integration.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/util/integration.py -------------------------------------------------------------------------------- /util/pointcloud.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/util/pointcloud.py -------------------------------------------------------------------------------- /util/timer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chrischoy/DeepGlobalRegistration/HEAD/util/timer.py --------------------------------------------------------------------------------