├── LICENSE ├── README.md ├── assets ├── CTR.png ├── LGFCTR.png ├── 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 └── sample9.png-sample10.png ├── configs ├── data │ ├── __init__.py │ ├── base.py │ ├── megadepth_test_1500.py │ └── megadepth_trainval_800.py └── lgfctr │ └── outdoor │ ├── lgfctr_ds.py │ └── lgfctr_ds_eval.py ├── environment.yaml ├── requirements.txt ├── scripts ├── reproduce_test │ └── outdoor_ds.sh └── reproduce_train │ └── outdoor_ds.sh ├── src ├── __init__.py ├── config │ └── default.py ├── datasets │ ├── megadepth.py │ ├── sampler.py │ └── scannet.py ├── lgfctr │ ├── __init__.py │ ├── coarse_matching.py │ ├── fine_matching.py │ ├── fine_preprocess.py │ ├── geometry.py │ ├── lgfctr.py │ ├── linear_attention.py │ ├── position_encoding.py │ ├── stem.py │ ├── supervision.py │ └── transformer.py ├── lightning │ ├── data.py │ └── lightning_lgfctr.py ├── losses │ └── lgfctr_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 └── vis ├── vis_attention.py └── vis_single_pair.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/README.md -------------------------------------------------------------------------------- /assets/CTR.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/assets/CTR.png -------------------------------------------------------------------------------- /assets/LGFCTR.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/assets/LGFCTR.png -------------------------------------------------------------------------------- /assets/megadepth_test_1500_scene_info/0015_0.1_0.3.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/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/zwh0527/LGFCTR/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/zwh0527/LGFCTR/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/zwh0527/LGFCTR/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/zwh0527/LGFCTR/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/zwh0527/LGFCTR/HEAD/assets/megadepth_test_1500_scene_info/megadepth_test_1500.txt -------------------------------------------------------------------------------- /assets/sample9.png-sample10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/assets/sample9.png-sample10.png -------------------------------------------------------------------------------- /configs/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/configs/data/base.py -------------------------------------------------------------------------------- /configs/data/megadepth_test_1500.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/configs/data/megadepth_test_1500.py -------------------------------------------------------------------------------- /configs/data/megadepth_trainval_800.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/configs/data/megadepth_trainval_800.py -------------------------------------------------------------------------------- /configs/lgfctr/outdoor/lgfctr_ds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/configs/lgfctr/outdoor/lgfctr_ds.py -------------------------------------------------------------------------------- /configs/lgfctr/outdoor/lgfctr_ds_eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/configs/lgfctr/outdoor/lgfctr_ds_eval.py -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/environment.yaml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/reproduce_test/outdoor_ds.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/scripts/reproduce_test/outdoor_ds.sh -------------------------------------------------------------------------------- /scripts/reproduce_train/outdoor_ds.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/scripts/reproduce_train/outdoor_ds.sh -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/config/default.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/config/default.py -------------------------------------------------------------------------------- /src/datasets/megadepth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/datasets/megadepth.py -------------------------------------------------------------------------------- /src/datasets/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/datasets/sampler.py -------------------------------------------------------------------------------- /src/datasets/scannet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/datasets/scannet.py -------------------------------------------------------------------------------- /src/lgfctr/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/lgfctr/__init__.py -------------------------------------------------------------------------------- /src/lgfctr/coarse_matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/lgfctr/coarse_matching.py -------------------------------------------------------------------------------- /src/lgfctr/fine_matching.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/lgfctr/fine_matching.py -------------------------------------------------------------------------------- /src/lgfctr/fine_preprocess.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/lgfctr/fine_preprocess.py -------------------------------------------------------------------------------- /src/lgfctr/geometry.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/lgfctr/geometry.py -------------------------------------------------------------------------------- /src/lgfctr/lgfctr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/lgfctr/lgfctr.py -------------------------------------------------------------------------------- /src/lgfctr/linear_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/lgfctr/linear_attention.py -------------------------------------------------------------------------------- /src/lgfctr/position_encoding.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/lgfctr/position_encoding.py -------------------------------------------------------------------------------- /src/lgfctr/stem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/lgfctr/stem.py -------------------------------------------------------------------------------- /src/lgfctr/supervision.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/lgfctr/supervision.py -------------------------------------------------------------------------------- /src/lgfctr/transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/lgfctr/transformer.py -------------------------------------------------------------------------------- /src/lightning/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/lightning/data.py -------------------------------------------------------------------------------- /src/lightning/lightning_lgfctr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/lightning/lightning_lgfctr.py -------------------------------------------------------------------------------- /src/losses/lgfctr_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/losses/lgfctr_loss.py -------------------------------------------------------------------------------- /src/optimizers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/optimizers/__init__.py -------------------------------------------------------------------------------- /src/utils/augment.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/utils/augment.py -------------------------------------------------------------------------------- /src/utils/comm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/utils/comm.py -------------------------------------------------------------------------------- /src/utils/dataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/utils/dataloader.py -------------------------------------------------------------------------------- /src/utils/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/utils/dataset.py -------------------------------------------------------------------------------- /src/utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/utils/metrics.py -------------------------------------------------------------------------------- /src/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/utils/misc.py -------------------------------------------------------------------------------- /src/utils/plotting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/utils/plotting.py -------------------------------------------------------------------------------- /src/utils/profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/src/utils/profiler.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/train.py -------------------------------------------------------------------------------- /vis/vis_attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/vis/vis_attention.py -------------------------------------------------------------------------------- /vis/vis_single_pair.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/zwh0527/LGFCTR/HEAD/vis/vis_single_pair.py --------------------------------------------------------------------------------