├── LICENSE ├── README.md └── code ├── .DS_Store ├── .gitignore ├── QuadTreeAttention ├── .DS_Store ├── QuadtreeAttention │ ├── .DS_Store │ ├── functions │ │ └── quadtree_attention.py │ ├── modules │ │ └── quadtree_attention.py │ └── src │ │ ├── score_computation.cpp │ │ ├── score_computation.h │ │ ├── score_computation_kernal.cu │ │ ├── utils.h │ │ ├── value_aggregation.cpp │ │ ├── value_aggregation.h │ │ └── value_aggregation_kernel.cu └── setup.py └── occ^2 ├── .gitignore ├── .gitmodules ├── LICENSE ├── assets └── scannet_test_1500 │ ├── intrinsics.npz │ ├── scannet_test.txt │ ├── statistics.json │ └── test.npz ├── configs ├── data │ ├── __init__.py │ ├── base.py │ ├── scannet_test_1500.py │ └── scannet_trainval.py └── loftr │ ├── indoor │ ├── buggy_pos_enc │ │ ├── loftr_ds.py │ │ ├── loftr_ds_dense.py │ │ ├── loftr_ot.py │ │ └── loftr_ot_dense.py │ ├── loftr_ds_quadtree.py │ └── loftr_ds_quadtree_eval.py │ └── tartanair │ ├── loftr_ds_quadtree.py │ └── loftr_ds_quadtree_eval.py ├── docs └── TRAINING.md ├── environment.yaml ├── makefile ├── notebooks ├── demo_single_pair.ipynb └── visualize_dump_results.ipynb ├── reader.py ├── requirements.txt ├── scripts ├── reproduce_test │ ├── indoor_ds_quadtree.sh │ └── tartanair_ds_quadtree.sh └── reproduce_train │ └── indoor_quadtree_ds.sh ├── src ├── __init__.py ├── config │ └── default.py ├── datasets │ ├── sampler.py │ └── scannet.py ├── lightning │ ├── data.py │ └── lightning_loftr.py ├── loftr │ ├── __init__.py │ ├── backbone │ │ ├── __init__.py │ │ ├── global_feature.py │ │ └── resnet_fpn.py │ ├── loftr.py │ ├── loftr_module │ │ ├── __init__.py │ │ ├── fine_preprocess.py │ │ ├── linear_attention.py │ │ ├── quadtree_attention.py │ │ └── transformer.py │ └── utils │ │ ├── coarse_matching.py │ │ ├── cvpr_ds_config.py │ │ ├── fine_matching.py │ │ ├── gemo.py │ │ ├── geometry.py │ │ ├── position_encoding.py │ │ └── supervision.py ├── losses │ └── loftr_loss.py ├── optimizers │ └── __init__.py └── utils │ ├── augment.py │ ├── comm.py │ ├── dataloader.py │ ├── dataset.py │ ├── metrics.py │ ├── misc.py │ ├── plotting.py │ └── profiler.py ├── test.py └── train.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/README.md -------------------------------------------------------------------------------- /code/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/.DS_Store -------------------------------------------------------------------------------- /code/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/.gitignore -------------------------------------------------------------------------------- /code/QuadTreeAttention/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/QuadTreeAttention/.DS_Store -------------------------------------------------------------------------------- /code/QuadTreeAttention/QuadtreeAttention/.DS_Store: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/QuadTreeAttention/QuadtreeAttention/.DS_Store -------------------------------------------------------------------------------- /code/QuadTreeAttention/QuadtreeAttention/functions/quadtree_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/QuadTreeAttention/QuadtreeAttention/functions/quadtree_attention.py -------------------------------------------------------------------------------- /code/QuadTreeAttention/QuadtreeAttention/modules/quadtree_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/QuadTreeAttention/QuadtreeAttention/modules/quadtree_attention.py -------------------------------------------------------------------------------- /code/QuadTreeAttention/QuadtreeAttention/src/score_computation.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/QuadTreeAttention/QuadtreeAttention/src/score_computation.cpp -------------------------------------------------------------------------------- /code/QuadTreeAttention/QuadtreeAttention/src/score_computation.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/QuadTreeAttention/QuadtreeAttention/src/score_computation.h -------------------------------------------------------------------------------- /code/QuadTreeAttention/QuadtreeAttention/src/score_computation_kernal.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/QuadTreeAttention/QuadtreeAttention/src/score_computation_kernal.cu -------------------------------------------------------------------------------- /code/QuadTreeAttention/QuadtreeAttention/src/utils.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/QuadTreeAttention/QuadtreeAttention/src/utils.h -------------------------------------------------------------------------------- /code/QuadTreeAttention/QuadtreeAttention/src/value_aggregation.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/QuadTreeAttention/QuadtreeAttention/src/value_aggregation.cpp -------------------------------------------------------------------------------- /code/QuadTreeAttention/QuadtreeAttention/src/value_aggregation.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/QuadTreeAttention/QuadtreeAttention/src/value_aggregation.h -------------------------------------------------------------------------------- /code/QuadTreeAttention/QuadtreeAttention/src/value_aggregation_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/QuadTreeAttention/QuadtreeAttention/src/value_aggregation_kernel.cu -------------------------------------------------------------------------------- /code/QuadTreeAttention/setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/QuadTreeAttention/setup.py -------------------------------------------------------------------------------- /code/occ^2/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/.gitignore -------------------------------------------------------------------------------- /code/occ^2/.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/.gitmodules -------------------------------------------------------------------------------- /code/occ^2/LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/LICENSE -------------------------------------------------------------------------------- /code/occ^2/assets/scannet_test_1500/intrinsics.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/assets/scannet_test_1500/intrinsics.npz -------------------------------------------------------------------------------- /code/occ^2/assets/scannet_test_1500/scannet_test.txt: -------------------------------------------------------------------------------- 1 | test.npz -------------------------------------------------------------------------------- /code/occ^2/assets/scannet_test_1500/statistics.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/assets/scannet_test_1500/statistics.json -------------------------------------------------------------------------------- /code/occ^2/assets/scannet_test_1500/test.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/assets/scannet_test_1500/test.npz -------------------------------------------------------------------------------- /code/occ^2/configs/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/occ^2/configs/data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/configs/data/base.py -------------------------------------------------------------------------------- /code/occ^2/configs/data/scannet_test_1500.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/configs/data/scannet_test_1500.py -------------------------------------------------------------------------------- /code/occ^2/configs/data/scannet_trainval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/configs/data/scannet_trainval.py -------------------------------------------------------------------------------- /code/occ^2/configs/loftr/indoor/buggy_pos_enc/loftr_ds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/configs/loftr/indoor/buggy_pos_enc/loftr_ds.py -------------------------------------------------------------------------------- /code/occ^2/configs/loftr/indoor/buggy_pos_enc/loftr_ds_dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/configs/loftr/indoor/buggy_pos_enc/loftr_ds_dense.py -------------------------------------------------------------------------------- /code/occ^2/configs/loftr/indoor/buggy_pos_enc/loftr_ot.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/configs/loftr/indoor/buggy_pos_enc/loftr_ot.py -------------------------------------------------------------------------------- /code/occ^2/configs/loftr/indoor/buggy_pos_enc/loftr_ot_dense.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/configs/loftr/indoor/buggy_pos_enc/loftr_ot_dense.py -------------------------------------------------------------------------------- /code/occ^2/configs/loftr/indoor/loftr_ds_quadtree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/configs/loftr/indoor/loftr_ds_quadtree.py -------------------------------------------------------------------------------- /code/occ^2/configs/loftr/indoor/loftr_ds_quadtree_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/configs/loftr/indoor/loftr_ds_quadtree_eval.py -------------------------------------------------------------------------------- /code/occ^2/configs/loftr/tartanair/loftr_ds_quadtree.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/configs/loftr/tartanair/loftr_ds_quadtree.py -------------------------------------------------------------------------------- /code/occ^2/configs/loftr/tartanair/loftr_ds_quadtree_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/configs/loftr/tartanair/loftr_ds_quadtree_eval.py -------------------------------------------------------------------------------- /code/occ^2/docs/TRAINING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/docs/TRAINING.md -------------------------------------------------------------------------------- /code/occ^2/environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/environment.yaml -------------------------------------------------------------------------------- /code/occ^2/makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/makefile -------------------------------------------------------------------------------- /code/occ^2/notebooks/demo_single_pair.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/notebooks/demo_single_pair.ipynb -------------------------------------------------------------------------------- /code/occ^2/notebooks/visualize_dump_results.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/notebooks/visualize_dump_results.ipynb -------------------------------------------------------------------------------- /code/occ^2/reader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/reader.py -------------------------------------------------------------------------------- /code/occ^2/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/requirements.txt -------------------------------------------------------------------------------- /code/occ^2/scripts/reproduce_test/indoor_ds_quadtree.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/scripts/reproduce_test/indoor_ds_quadtree.sh -------------------------------------------------------------------------------- /code/occ^2/scripts/reproduce_test/tartanair_ds_quadtree.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/scripts/reproduce_test/tartanair_ds_quadtree.sh -------------------------------------------------------------------------------- /code/occ^2/scripts/reproduce_train/indoor_quadtree_ds.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/scripts/reproduce_train/indoor_quadtree_ds.sh -------------------------------------------------------------------------------- /code/occ^2/src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /code/occ^2/src/config/default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/config/default.py -------------------------------------------------------------------------------- /code/occ^2/src/datasets/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/datasets/sampler.py -------------------------------------------------------------------------------- /code/occ^2/src/datasets/scannet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/datasets/scannet.py -------------------------------------------------------------------------------- /code/occ^2/src/lightning/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/lightning/data.py -------------------------------------------------------------------------------- /code/occ^2/src/lightning/lightning_loftr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/lightning/lightning_loftr.py -------------------------------------------------------------------------------- /code/occ^2/src/loftr/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/loftr/__init__.py -------------------------------------------------------------------------------- /code/occ^2/src/loftr/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/loftr/backbone/__init__.py -------------------------------------------------------------------------------- /code/occ^2/src/loftr/backbone/global_feature.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/loftr/backbone/global_feature.py -------------------------------------------------------------------------------- /code/occ^2/src/loftr/backbone/resnet_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/loftr/backbone/resnet_fpn.py -------------------------------------------------------------------------------- /code/occ^2/src/loftr/loftr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/loftr/loftr.py -------------------------------------------------------------------------------- /code/occ^2/src/loftr/loftr_module/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/loftr/loftr_module/__init__.py -------------------------------------------------------------------------------- /code/occ^2/src/loftr/loftr_module/fine_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/loftr/loftr_module/fine_preprocess.py -------------------------------------------------------------------------------- /code/occ^2/src/loftr/loftr_module/linear_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/loftr/loftr_module/linear_attention.py -------------------------------------------------------------------------------- /code/occ^2/src/loftr/loftr_module/quadtree_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/loftr/loftr_module/quadtree_attention.py -------------------------------------------------------------------------------- /code/occ^2/src/loftr/loftr_module/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/loftr/loftr_module/transformer.py -------------------------------------------------------------------------------- /code/occ^2/src/loftr/utils/coarse_matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/loftr/utils/coarse_matching.py -------------------------------------------------------------------------------- /code/occ^2/src/loftr/utils/cvpr_ds_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/loftr/utils/cvpr_ds_config.py -------------------------------------------------------------------------------- /code/occ^2/src/loftr/utils/fine_matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/loftr/utils/fine_matching.py -------------------------------------------------------------------------------- /code/occ^2/src/loftr/utils/gemo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/loftr/utils/gemo.py -------------------------------------------------------------------------------- /code/occ^2/src/loftr/utils/geometry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/loftr/utils/geometry.py -------------------------------------------------------------------------------- /code/occ^2/src/loftr/utils/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/loftr/utils/position_encoding.py -------------------------------------------------------------------------------- /code/occ^2/src/loftr/utils/supervision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/loftr/utils/supervision.py -------------------------------------------------------------------------------- /code/occ^2/src/losses/loftr_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/losses/loftr_loss.py -------------------------------------------------------------------------------- /code/occ^2/src/optimizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/optimizers/__init__.py -------------------------------------------------------------------------------- /code/occ^2/src/utils/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/utils/augment.py -------------------------------------------------------------------------------- /code/occ^2/src/utils/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/utils/comm.py -------------------------------------------------------------------------------- /code/occ^2/src/utils/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/utils/dataloader.py -------------------------------------------------------------------------------- /code/occ^2/src/utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/utils/dataset.py -------------------------------------------------------------------------------- /code/occ^2/src/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/utils/metrics.py -------------------------------------------------------------------------------- /code/occ^2/src/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/utils/misc.py -------------------------------------------------------------------------------- /code/occ^2/src/utils/plotting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/utils/plotting.py -------------------------------------------------------------------------------- /code/occ^2/src/utils/profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/src/utils/profiler.py -------------------------------------------------------------------------------- /code/occ^2/test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/test.py -------------------------------------------------------------------------------- /code/occ^2/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/megvii-research/Occ2net/HEAD/code/occ^2/train.py --------------------------------------------------------------------------------