├── .gitignore ├── LICENSE ├── README.md ├── assets ├── megadepth_test_1500_scene_info │ ├── 0015_0.1_0.3.npz │ ├── 0015_0.3_0.5.npz │ ├── 0022_0.1_0.3.npz │ ├── 0022_0.3_0.5.npz │ ├── 0022_0.5_0.7.npz │ └── megadepth_test_1500.txt ├── phototourism_sample_images │ ├── london_bridge_19481797_2295892421.jpg │ ├── london_bridge_49190386_5209386933.jpg │ ├── united_states_capitol_26757027_6717084061.jpg │ └── united_states_capitol_98169888_3347710852.jpg ├── scannet_sample_images │ ├── scene0707_00_15.jpg │ ├── scene0707_00_45.jpg │ ├── scene0758_00_165.jpg │ └── scene0758_00_510.jpg └── scannet_test_1500 │ ├── intrinsics.npz │ ├── scannet_test.txt │ ├── statistics.json │ └── test.npz ├── configs ├── data │ ├── __init__.py │ ├── base.py │ ├── megadepth_test_1500.py │ ├── megadepth_trainval_832.py │ ├── scannet_test_1500.py │ └── scannet_trainval.py └── edm │ ├── indoor │ └── edm_base.py │ └── outdoor │ └── edm_base.py ├── data ├── megadepth │ ├── index │ │ └── .gitignore │ ├── test │ │ └── .gitignore │ └── train │ │ └── .gitignore └── scannet │ ├── index │ └── .gitignore │ ├── test │ └── .gitignore │ └── train │ └── .gitignore ├── demo_single_pair.ipynb ├── deploy ├── edm_onnx_cpp │ ├── CMakeLists.txt │ ├── README.md │ ├── demo.cpp │ ├── demo.sh │ ├── edm │ │ ├── edm.cpp │ │ └── edm.h │ ├── scene0707_00_15.jpg │ └── scene0707_00_45.jpg ├── export_onnx.py ├── readme.md ├── requirements_deploy.txt └── run_onnx.py ├── environment.yaml ├── requirements.txt ├── runtime_single_pair.py ├── scripts ├── reproduce_test │ ├── indoor.sh │ ├── outdoor.sh │ └── outdoor_loransac.sh └── reproduce_train │ └── outdoor.sh ├── src ├── __init__.py ├── config │ └── default.py ├── datasets │ ├── megadepth.py │ ├── sampler.py │ └── scannet.py ├── edm │ ├── __init__.py │ ├── backbone │ │ └── resnet.py │ ├── edm.py │ ├── head │ │ ├── coarse_matching.py │ │ └── fine_matching.py │ ├── neck │ │ ├── __init__.py │ │ ├── loftr_module │ │ │ ├── __init__.py │ │ │ └── transformer.py │ │ └── neck.py │ └── utils │ │ ├── geometry.py │ │ └── supervision.py ├── lightning │ ├── data.py │ └── lightning_edm.py ├── losses │ └── edm_loss.py ├── optimizers │ └── __init__.py └── utils │ ├── augment.py │ ├── comm.py │ ├── dataloader.py │ ├── dataset.py │ ├── metrics.py │ ├── misc.py │ ├── plotting.py │ ├── profiler.py │ ├── warppers.py │ └── warppers_utils.py ├── test.py ├── train.py └── weights └── .gitignore /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/README.md -------------------------------------------------------------------------------- /assets/megadepth_test_1500_scene_info/0015_0.1_0.3.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/assets/megadepth_test_1500_scene_info/0015_0.1_0.3.npz -------------------------------------------------------------------------------- /assets/megadepth_test_1500_scene_info/0015_0.3_0.5.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/assets/megadepth_test_1500_scene_info/0015_0.3_0.5.npz -------------------------------------------------------------------------------- /assets/megadepth_test_1500_scene_info/0022_0.1_0.3.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/assets/megadepth_test_1500_scene_info/0022_0.1_0.3.npz -------------------------------------------------------------------------------- /assets/megadepth_test_1500_scene_info/0022_0.3_0.5.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/assets/megadepth_test_1500_scene_info/0022_0.3_0.5.npz -------------------------------------------------------------------------------- /assets/megadepth_test_1500_scene_info/0022_0.5_0.7.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/assets/megadepth_test_1500_scene_info/0022_0.5_0.7.npz -------------------------------------------------------------------------------- /assets/megadepth_test_1500_scene_info/megadepth_test_1500.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/assets/megadepth_test_1500_scene_info/megadepth_test_1500.txt -------------------------------------------------------------------------------- /assets/phototourism_sample_images/london_bridge_19481797_2295892421.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/assets/phototourism_sample_images/london_bridge_19481797_2295892421.jpg -------------------------------------------------------------------------------- /assets/phototourism_sample_images/london_bridge_49190386_5209386933.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/assets/phototourism_sample_images/london_bridge_49190386_5209386933.jpg -------------------------------------------------------------------------------- /assets/phototourism_sample_images/united_states_capitol_26757027_6717084061.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/assets/phototourism_sample_images/united_states_capitol_26757027_6717084061.jpg -------------------------------------------------------------------------------- /assets/phototourism_sample_images/united_states_capitol_98169888_3347710852.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/assets/phototourism_sample_images/united_states_capitol_98169888_3347710852.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0707_00_15.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/assets/scannet_sample_images/scene0707_00_15.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0707_00_45.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/assets/scannet_sample_images/scene0707_00_45.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0758_00_165.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/assets/scannet_sample_images/scene0758_00_165.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0758_00_510.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/assets/scannet_sample_images/scene0758_00_510.jpg -------------------------------------------------------------------------------- /assets/scannet_test_1500/intrinsics.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/assets/scannet_test_1500/intrinsics.npz -------------------------------------------------------------------------------- /assets/scannet_test_1500/scannet_test.txt: -------------------------------------------------------------------------------- 1 | test.npz -------------------------------------------------------------------------------- /assets/scannet_test_1500/statistics.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/assets/scannet_test_1500/statistics.json -------------------------------------------------------------------------------- /assets/scannet_test_1500/test.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/assets/scannet_test_1500/test.npz -------------------------------------------------------------------------------- /configs/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/configs/data/base.py -------------------------------------------------------------------------------- /configs/data/megadepth_test_1500.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/configs/data/megadepth_test_1500.py -------------------------------------------------------------------------------- /configs/data/megadepth_trainval_832.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/configs/data/megadepth_trainval_832.py -------------------------------------------------------------------------------- /configs/data/scannet_test_1500.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/configs/data/scannet_test_1500.py -------------------------------------------------------------------------------- /configs/data/scannet_trainval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/configs/data/scannet_trainval.py -------------------------------------------------------------------------------- /configs/edm/indoor/edm_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/configs/edm/indoor/edm_base.py -------------------------------------------------------------------------------- /configs/edm/outdoor/edm_base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/configs/edm/outdoor/edm_base.py -------------------------------------------------------------------------------- /data/megadepth/index/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/data/megadepth/index/.gitignore -------------------------------------------------------------------------------- /data/megadepth/test/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/data/megadepth/test/.gitignore -------------------------------------------------------------------------------- /data/megadepth/train/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/data/megadepth/train/.gitignore -------------------------------------------------------------------------------- /data/scannet/index/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/data/scannet/index/.gitignore -------------------------------------------------------------------------------- /data/scannet/test/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/data/scannet/test/.gitignore -------------------------------------------------------------------------------- /data/scannet/train/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/data/scannet/train/.gitignore -------------------------------------------------------------------------------- /demo_single_pair.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/demo_single_pair.ipynb -------------------------------------------------------------------------------- /deploy/edm_onnx_cpp/CMakeLists.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/deploy/edm_onnx_cpp/CMakeLists.txt -------------------------------------------------------------------------------- /deploy/edm_onnx_cpp/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/deploy/edm_onnx_cpp/README.md -------------------------------------------------------------------------------- /deploy/edm_onnx_cpp/demo.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/deploy/edm_onnx_cpp/demo.cpp -------------------------------------------------------------------------------- /deploy/edm_onnx_cpp/demo.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/deploy/edm_onnx_cpp/demo.sh -------------------------------------------------------------------------------- /deploy/edm_onnx_cpp/edm/edm.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/deploy/edm_onnx_cpp/edm/edm.cpp -------------------------------------------------------------------------------- /deploy/edm_onnx_cpp/edm/edm.h: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/deploy/edm_onnx_cpp/edm/edm.h -------------------------------------------------------------------------------- /deploy/edm_onnx_cpp/scene0707_00_15.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/deploy/edm_onnx_cpp/scene0707_00_15.jpg -------------------------------------------------------------------------------- /deploy/edm_onnx_cpp/scene0707_00_45.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/deploy/edm_onnx_cpp/scene0707_00_45.jpg -------------------------------------------------------------------------------- /deploy/export_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/deploy/export_onnx.py -------------------------------------------------------------------------------- /deploy/readme.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/deploy/readme.md -------------------------------------------------------------------------------- /deploy/requirements_deploy.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/deploy/requirements_deploy.txt -------------------------------------------------------------------------------- /deploy/run_onnx.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/deploy/run_onnx.py -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/environment.yaml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/requirements.txt -------------------------------------------------------------------------------- /runtime_single_pair.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/runtime_single_pair.py -------------------------------------------------------------------------------- /scripts/reproduce_test/indoor.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/scripts/reproduce_test/indoor.sh -------------------------------------------------------------------------------- /scripts/reproduce_test/outdoor.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/scripts/reproduce_test/outdoor.sh -------------------------------------------------------------------------------- /scripts/reproduce_test/outdoor_loransac.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/scripts/reproduce_test/outdoor_loransac.sh -------------------------------------------------------------------------------- /scripts/reproduce_train/outdoor.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/scripts/reproduce_train/outdoor.sh -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/config/default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/config/default.py -------------------------------------------------------------------------------- /src/datasets/megadepth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/datasets/megadepth.py -------------------------------------------------------------------------------- /src/datasets/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/datasets/sampler.py -------------------------------------------------------------------------------- /src/datasets/scannet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/datasets/scannet.py -------------------------------------------------------------------------------- /src/edm/__init__.py: -------------------------------------------------------------------------------- 1 | from .edm import EDM 2 | 3 | -------------------------------------------------------------------------------- /src/edm/backbone/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/edm/backbone/resnet.py -------------------------------------------------------------------------------- /src/edm/edm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/edm/edm.py -------------------------------------------------------------------------------- /src/edm/head/coarse_matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/edm/head/coarse_matching.py -------------------------------------------------------------------------------- /src/edm/head/fine_matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/edm/head/fine_matching.py -------------------------------------------------------------------------------- /src/edm/neck/__init__.py: -------------------------------------------------------------------------------- 1 | from .loftr_module import LocalFeatureTransformer 2 | -------------------------------------------------------------------------------- /src/edm/neck/loftr_module/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/edm/neck/loftr_module/__init__.py -------------------------------------------------------------------------------- /src/edm/neck/loftr_module/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/edm/neck/loftr_module/transformer.py -------------------------------------------------------------------------------- /src/edm/neck/neck.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/edm/neck/neck.py -------------------------------------------------------------------------------- /src/edm/utils/geometry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/edm/utils/geometry.py -------------------------------------------------------------------------------- /src/edm/utils/supervision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/edm/utils/supervision.py -------------------------------------------------------------------------------- /src/lightning/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/lightning/data.py -------------------------------------------------------------------------------- /src/lightning/lightning_edm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/lightning/lightning_edm.py -------------------------------------------------------------------------------- /src/losses/edm_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/losses/edm_loss.py -------------------------------------------------------------------------------- /src/optimizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/optimizers/__init__.py -------------------------------------------------------------------------------- /src/utils/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/utils/augment.py -------------------------------------------------------------------------------- /src/utils/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/utils/comm.py -------------------------------------------------------------------------------- /src/utils/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/utils/dataloader.py -------------------------------------------------------------------------------- /src/utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/utils/dataset.py -------------------------------------------------------------------------------- /src/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/utils/metrics.py -------------------------------------------------------------------------------- /src/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/utils/misc.py -------------------------------------------------------------------------------- /src/utils/plotting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/utils/plotting.py -------------------------------------------------------------------------------- /src/utils/profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/utils/profiler.py -------------------------------------------------------------------------------- /src/utils/warppers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/utils/warppers.py -------------------------------------------------------------------------------- /src/utils/warppers_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/src/utils/warppers_utils.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/train.py -------------------------------------------------------------------------------- /weights/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/chicleee/EDM/HEAD/weights/.gitignore --------------------------------------------------------------------------------