├── .gitignore ├── README.md ├── configs ├── data │ ├── 7scenes.py │ ├── aachen_v10.py │ ├── aachen_v8.py │ ├── base.py │ ├── cambridge.py │ ├── scannet.py │ ├── scannet_code_finetune.py │ ├── store1f.py │ ├── store4f.py │ └── storeb1.py └── neumap │ ├── 7scenes_stage1.py │ ├── 7scenes_stage2.py │ ├── aachen_c1.py │ ├── aachen_c100.py │ ├── aachen_scratch.py │ ├── aachen_stage1_c1.py │ ├── aachen_stage1_c100.py │ ├── aachen_stage2_c100.py │ ├── cambridge_stage1.py │ ├── cambridge_stage2.py │ ├── scannet.py │ ├── scannet_code_finetune.py │ └── store.py ├── requirement.txt ├── scripts ├── reproduce_test │ ├── 7scenes.sh │ ├── aachen_v10.sh │ ├── aachen_v8.sh │ ├── cambridge.sh │ ├── scannet.sh │ └── store.sh └── reproduce_train │ ├── 7scenes_stage1.sh │ ├── 7scenes_stage2.sh │ ├── aachen_v10_stage1.sh │ ├── aachen_v10_stage2.sh │ ├── aachen_v8.sh │ ├── cambridge_stage1.sh │ ├── cambridge_stage2.sh │ ├── scannet.sh │ ├── scannet_code_finetune.sh │ └── store.sh ├── src ├── NeuMap │ ├── __init__.py │ ├── backbone │ │ ├── __init__.py │ │ ├── configs.py │ │ ├── resnet_fpn.py │ │ └── utils.py │ ├── neumap.py │ └── transformer_module │ │ └── vit.py ├── config │ └── default.py ├── datasets │ ├── kapture.py │ ├── sampler.py │ └── scannet.py ├── lightning │ ├── data.py │ └── lightning_NeuMap.py ├── losses │ └── loss.py ├── optimizers │ └── __init__.py └── utils │ ├── augment.py │ ├── comm.py │ ├── dataloader.py │ ├── dataset.py │ ├── geometry.py │ ├── metrics.py │ ├── misc.py │ └── profiler.py ├── teaser.png ├── test.py ├── tools ├── get_label_file.py ├── get_params.py ├── get_params_cambridge.py └── get_query_file.py ├── train.py └── weights └── outdoor_ds.ckpt /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/.gitignore -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/README.md -------------------------------------------------------------------------------- /configs/data/7scenes.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/configs/data/7scenes.py -------------------------------------------------------------------------------- /configs/data/aachen_v10.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/configs/data/aachen_v10.py -------------------------------------------------------------------------------- /configs/data/aachen_v8.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/configs/data/aachen_v8.py -------------------------------------------------------------------------------- /configs/data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/configs/data/base.py -------------------------------------------------------------------------------- /configs/data/cambridge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/configs/data/cambridge.py -------------------------------------------------------------------------------- /configs/data/scannet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/configs/data/scannet.py -------------------------------------------------------------------------------- /configs/data/scannet_code_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/configs/data/scannet_code_finetune.py -------------------------------------------------------------------------------- /configs/data/store1f.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/configs/data/store1f.py -------------------------------------------------------------------------------- /configs/data/store4f.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/configs/data/store4f.py -------------------------------------------------------------------------------- /configs/data/storeb1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/configs/data/storeb1.py -------------------------------------------------------------------------------- /configs/neumap/7scenes_stage1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/configs/neumap/7scenes_stage1.py -------------------------------------------------------------------------------- /configs/neumap/7scenes_stage2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/configs/neumap/7scenes_stage2.py -------------------------------------------------------------------------------- /configs/neumap/aachen_c1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/configs/neumap/aachen_c1.py -------------------------------------------------------------------------------- /configs/neumap/aachen_c100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/configs/neumap/aachen_c100.py -------------------------------------------------------------------------------- /configs/neumap/aachen_scratch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/configs/neumap/aachen_scratch.py -------------------------------------------------------------------------------- /configs/neumap/aachen_stage1_c1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/configs/neumap/aachen_stage1_c1.py -------------------------------------------------------------------------------- /configs/neumap/aachen_stage1_c100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/configs/neumap/aachen_stage1_c100.py -------------------------------------------------------------------------------- /configs/neumap/aachen_stage2_c100.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/configs/neumap/aachen_stage2_c100.py -------------------------------------------------------------------------------- /configs/neumap/cambridge_stage1.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/configs/neumap/cambridge_stage1.py -------------------------------------------------------------------------------- /configs/neumap/cambridge_stage2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/configs/neumap/cambridge_stage2.py -------------------------------------------------------------------------------- /configs/neumap/scannet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/configs/neumap/scannet.py -------------------------------------------------------------------------------- /configs/neumap/scannet_code_finetune.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/configs/neumap/scannet_code_finetune.py -------------------------------------------------------------------------------- /configs/neumap/store.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/configs/neumap/store.py -------------------------------------------------------------------------------- /requirement.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/requirement.txt -------------------------------------------------------------------------------- /scripts/reproduce_test/7scenes.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/scripts/reproduce_test/7scenes.sh -------------------------------------------------------------------------------- /scripts/reproduce_test/aachen_v10.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/scripts/reproduce_test/aachen_v10.sh -------------------------------------------------------------------------------- /scripts/reproduce_test/aachen_v8.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/scripts/reproduce_test/aachen_v8.sh -------------------------------------------------------------------------------- /scripts/reproduce_test/cambridge.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/scripts/reproduce_test/cambridge.sh -------------------------------------------------------------------------------- /scripts/reproduce_test/scannet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/scripts/reproduce_test/scannet.sh -------------------------------------------------------------------------------- /scripts/reproduce_test/store.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/scripts/reproduce_test/store.sh -------------------------------------------------------------------------------- /scripts/reproduce_train/7scenes_stage1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/scripts/reproduce_train/7scenes_stage1.sh -------------------------------------------------------------------------------- /scripts/reproduce_train/7scenes_stage2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/scripts/reproduce_train/7scenes_stage2.sh -------------------------------------------------------------------------------- /scripts/reproduce_train/aachen_v10_stage1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/scripts/reproduce_train/aachen_v10_stage1.sh -------------------------------------------------------------------------------- /scripts/reproduce_train/aachen_v10_stage2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/scripts/reproduce_train/aachen_v10_stage2.sh -------------------------------------------------------------------------------- /scripts/reproduce_train/aachen_v8.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/scripts/reproduce_train/aachen_v8.sh -------------------------------------------------------------------------------- /scripts/reproduce_train/cambridge_stage1.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/scripts/reproduce_train/cambridge_stage1.sh -------------------------------------------------------------------------------- /scripts/reproduce_train/cambridge_stage2.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/scripts/reproduce_train/cambridge_stage2.sh -------------------------------------------------------------------------------- /scripts/reproduce_train/scannet.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/scripts/reproduce_train/scannet.sh -------------------------------------------------------------------------------- /scripts/reproduce_train/scannet_code_finetune.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/scripts/reproduce_train/scannet_code_finetune.sh -------------------------------------------------------------------------------- /scripts/reproduce_train/store.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/scripts/reproduce_train/store.sh -------------------------------------------------------------------------------- /src/NeuMap/__init__.py: -------------------------------------------------------------------------------- 1 | from .neumap import NeuMap 2 | -------------------------------------------------------------------------------- /src/NeuMap/backbone/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/src/NeuMap/backbone/__init__.py -------------------------------------------------------------------------------- /src/NeuMap/backbone/configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/src/NeuMap/backbone/configs.py -------------------------------------------------------------------------------- /src/NeuMap/backbone/resnet_fpn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/src/NeuMap/backbone/resnet_fpn.py -------------------------------------------------------------------------------- /src/NeuMap/backbone/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/src/NeuMap/backbone/utils.py -------------------------------------------------------------------------------- /src/NeuMap/neumap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/src/NeuMap/neumap.py -------------------------------------------------------------------------------- /src/NeuMap/transformer_module/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/src/NeuMap/transformer_module/vit.py -------------------------------------------------------------------------------- /src/config/default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/src/config/default.py -------------------------------------------------------------------------------- /src/datasets/kapture.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/src/datasets/kapture.py -------------------------------------------------------------------------------- /src/datasets/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/src/datasets/sampler.py -------------------------------------------------------------------------------- /src/datasets/scannet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/src/datasets/scannet.py -------------------------------------------------------------------------------- /src/lightning/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/src/lightning/data.py -------------------------------------------------------------------------------- /src/lightning/lightning_NeuMap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/src/lightning/lightning_NeuMap.py -------------------------------------------------------------------------------- /src/losses/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/src/losses/loss.py -------------------------------------------------------------------------------- /src/optimizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/src/optimizers/__init__.py -------------------------------------------------------------------------------- /src/utils/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/src/utils/augment.py -------------------------------------------------------------------------------- /src/utils/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/src/utils/comm.py -------------------------------------------------------------------------------- /src/utils/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/src/utils/dataloader.py -------------------------------------------------------------------------------- /src/utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/src/utils/dataset.py -------------------------------------------------------------------------------- /src/utils/geometry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/src/utils/geometry.py -------------------------------------------------------------------------------- /src/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/src/utils/metrics.py -------------------------------------------------------------------------------- /src/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/src/utils/misc.py -------------------------------------------------------------------------------- /src/utils/profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/src/utils/profiler.py -------------------------------------------------------------------------------- /teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/teaser.png -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/test.py -------------------------------------------------------------------------------- /tools/get_label_file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/tools/get_label_file.py -------------------------------------------------------------------------------- /tools/get_params.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/tools/get_params.py -------------------------------------------------------------------------------- /tools/get_params_cambridge.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/tools/get_params_cambridge.py -------------------------------------------------------------------------------- /tools/get_query_file.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/tools/get_query_file.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/train.py -------------------------------------------------------------------------------- /weights/outdoor_ds.ckpt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Tangshitao/NeuMap/HEAD/weights/outdoor_ds.ckpt --------------------------------------------------------------------------------