├── 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 │ ├── london_bridge_78916675_4568141288.jpg │ ├── london_bridge_94185272_3874562886.jpg │ ├── piazza_san_marco_06795901_3725050516.jpg │ ├── piazza_san_marco_15148634_5228701572.jpg │ ├── piazza_san_marco_18627786_5929294590.jpg │ ├── piazza_san_marco_43351518_2659980686.jpg │ ├── piazza_san_marco_58751010_4849458397.jpg │ ├── st_pauls_cathedral_30776973_2635313996.jpg │ ├── st_pauls_cathedral_37347628_10902811376.jpg │ ├── united_states_capitol_26757027_6717084061.jpg │ └── united_states_capitol_98169888_3347710852.jpg ├── scannet_sample_images │ ├── scene0711_00_frame-001680.jpg │ ├── scene0711_00_frame-001995.jpg │ ├── scene0713_00_frame-001320.jpg │ ├── scene0713_00_frame-002025.jpg │ ├── scene0721_00_frame-000375.jpg │ ├── scene0721_00_frame-002745.jpg │ ├── scene0722_00_frame-000045.jpg │ ├── scene0722_00_frame-000735.jpg │ ├── scene0726_00_frame-000135.jpg │ ├── scene0726_00_frame-000210.jpg │ ├── scene0737_00_frame-000930.jpg │ ├── scene0737_00_frame-001095.jpg │ ├── scene0738_00_frame-000885.jpg │ ├── scene0738_00_frame-001065.jpg │ ├── scene0743_00_frame-000000.jpg │ ├── scene0743_00_frame-001275.jpg │ ├── scene0744_00_frame-000585.jpg │ ├── scene0744_00_frame-002310.jpg │ ├── scene0747_00_frame-000000.jpg │ ├── scene0747_00_frame-001530.jpg │ ├── scene0752_00_frame-000075.jpg │ ├── scene0752_00_frame-001440.jpg │ ├── scene0755_00_frame-000120.jpg │ ├── scene0755_00_frame-002055.jpg │ ├── scene0758_00_frame-000165.jpg │ ├── scene0758_00_frame-000510.jpg │ ├── scene0768_00_frame-001095.jpg │ ├── scene0768_00_frame-003435.jpg │ ├── scene0806_00_frame-000225.jpg │ └── scene0806_00_frame-001095.jpg └── scannet_test_1500 │ ├── intrinsics.npz │ ├── scannet_test.txt │ ├── statistics.json │ └── test.npz ├── configs └── data │ ├── __init__.py │ ├── base.py │ ├── debug │ └── .gitignore │ ├── megadepth_test_1500.py │ ├── megadepth_trainval_544.py │ ├── megadepth_trainval_832.py │ ├── scannet_test_1500.py │ └── scannet_trainval.py ├── docs └── TRAINING.md ├── environment.yaml ├── requirements.txt ├── test.py └── train.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/README.md -------------------------------------------------------------------------------- /assets/megadepth_test_1500_scene_info/0015_0.1_0.3.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/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/ASTR2023/ASTR/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/ASTR2023/ASTR/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/ASTR2023/ASTR/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/ASTR2023/ASTR/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/ASTR2023/ASTR/HEAD/assets/megadepth_test_1500_scene_info/megadepth_test_1500.txt -------------------------------------------------------------------------------- /assets/phototourism_sample_images/london_bridge_19481797_2295892421.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/phototourism_sample_images/london_bridge_19481797_2295892421.jpg -------------------------------------------------------------------------------- /assets/phototourism_sample_images/london_bridge_49190386_5209386933.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/phototourism_sample_images/london_bridge_49190386_5209386933.jpg -------------------------------------------------------------------------------- /assets/phototourism_sample_images/london_bridge_78916675_4568141288.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/phototourism_sample_images/london_bridge_78916675_4568141288.jpg -------------------------------------------------------------------------------- /assets/phototourism_sample_images/london_bridge_94185272_3874562886.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/phototourism_sample_images/london_bridge_94185272_3874562886.jpg -------------------------------------------------------------------------------- /assets/phototourism_sample_images/piazza_san_marco_06795901_3725050516.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/phototourism_sample_images/piazza_san_marco_06795901_3725050516.jpg -------------------------------------------------------------------------------- /assets/phototourism_sample_images/piazza_san_marco_15148634_5228701572.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/phototourism_sample_images/piazza_san_marco_15148634_5228701572.jpg -------------------------------------------------------------------------------- /assets/phototourism_sample_images/piazza_san_marco_18627786_5929294590.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/phototourism_sample_images/piazza_san_marco_18627786_5929294590.jpg -------------------------------------------------------------------------------- /assets/phototourism_sample_images/piazza_san_marco_43351518_2659980686.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/phototourism_sample_images/piazza_san_marco_43351518_2659980686.jpg -------------------------------------------------------------------------------- /assets/phototourism_sample_images/piazza_san_marco_58751010_4849458397.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/phototourism_sample_images/piazza_san_marco_58751010_4849458397.jpg -------------------------------------------------------------------------------- /assets/phototourism_sample_images/st_pauls_cathedral_30776973_2635313996.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/phototourism_sample_images/st_pauls_cathedral_30776973_2635313996.jpg -------------------------------------------------------------------------------- /assets/phototourism_sample_images/st_pauls_cathedral_37347628_10902811376.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/phototourism_sample_images/st_pauls_cathedral_37347628_10902811376.jpg -------------------------------------------------------------------------------- /assets/phototourism_sample_images/united_states_capitol_26757027_6717084061.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/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/ASTR2023/ASTR/HEAD/assets/phototourism_sample_images/united_states_capitol_98169888_3347710852.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0711_00_frame-001680.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0711_00_frame-001680.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0711_00_frame-001995.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0711_00_frame-001995.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0713_00_frame-001320.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0713_00_frame-001320.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0713_00_frame-002025.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0713_00_frame-002025.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0721_00_frame-000375.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0721_00_frame-000375.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0721_00_frame-002745.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0721_00_frame-002745.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0722_00_frame-000045.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0722_00_frame-000045.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0722_00_frame-000735.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0722_00_frame-000735.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0726_00_frame-000135.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0726_00_frame-000135.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0726_00_frame-000210.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0726_00_frame-000210.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0737_00_frame-000930.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0737_00_frame-000930.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0737_00_frame-001095.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0737_00_frame-001095.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0738_00_frame-000885.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0738_00_frame-000885.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0738_00_frame-001065.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0738_00_frame-001065.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0743_00_frame-000000.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0743_00_frame-000000.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0743_00_frame-001275.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0743_00_frame-001275.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0744_00_frame-000585.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0744_00_frame-000585.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0744_00_frame-002310.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0744_00_frame-002310.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0747_00_frame-000000.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0747_00_frame-000000.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0747_00_frame-001530.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0747_00_frame-001530.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0752_00_frame-000075.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0752_00_frame-000075.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0752_00_frame-001440.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0752_00_frame-001440.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0755_00_frame-000120.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0755_00_frame-000120.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0755_00_frame-002055.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0755_00_frame-002055.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0758_00_frame-000165.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0758_00_frame-000165.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0758_00_frame-000510.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0758_00_frame-000510.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0768_00_frame-001095.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0768_00_frame-001095.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0768_00_frame-003435.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0768_00_frame-003435.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0806_00_frame-000225.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0806_00_frame-000225.jpg -------------------------------------------------------------------------------- /assets/scannet_sample_images/scene0806_00_frame-001095.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_sample_images/scene0806_00_frame-001095.jpg -------------------------------------------------------------------------------- /assets/scannet_test_1500/intrinsics.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/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/ASTR2023/ASTR/HEAD/assets/scannet_test_1500/statistics.json -------------------------------------------------------------------------------- /assets/scannet_test_1500/test.npz: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/assets/scannet_test_1500/test.npz -------------------------------------------------------------------------------- /configs/data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /configs/data/base.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/configs/data/base.py -------------------------------------------------------------------------------- /configs/data/debug/.gitignore: -------------------------------------------------------------------------------- 1 | * 2 | */ 3 | !.gitignore 4 | -------------------------------------------------------------------------------- /configs/data/megadepth_test_1500.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/configs/data/megadepth_test_1500.py -------------------------------------------------------------------------------- /configs/data/megadepth_trainval_544.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/configs/data/megadepth_trainval_544.py -------------------------------------------------------------------------------- /configs/data/megadepth_trainval_832.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/configs/data/megadepth_trainval_832.py -------------------------------------------------------------------------------- /configs/data/scannet_test_1500.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/configs/data/scannet_test_1500.py -------------------------------------------------------------------------------- /configs/data/scannet_trainval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/configs/data/scannet_trainval.py -------------------------------------------------------------------------------- /docs/TRAINING.md: -------------------------------------------------------------------------------- 1 | TODO 2 | -------------------------------------------------------------------------------- /environment.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/environment.yaml -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/requirements.txt -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/test.py -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/ASTR2023/ASTR/HEAD/train.py --------------------------------------------------------------------------------