├── .gitignore ├── .gitmodules ├── LICENSE ├── README.md ├── docs ├── data_conventions.md └── module.png ├── pyproject.toml └── scrstudio ├── configs ├── __init__.py ├── base_config.py ├── config_utils.py ├── experiment_config.py └── method_configs.py ├── data ├── __init__.py ├── datamanagers │ ├── __init__.py │ ├── base_datamanager.py │ └── glbuffer_datamanager.py ├── datasets │ ├── __init__.py │ └── camloc_dataset.py ├── samplers.py └── utils │ ├── __init__.py │ └── readers.py ├── encoders ├── ace_encoder.py ├── base_encoder.py ├── dedode_encoder.py ├── loftr_encoder.py ├── netvlad.py └── pca_encoder.py ├── engine ├── __init__.py ├── callbacks.py ├── optimizers.py ├── schedulers.py └── trainer.py ├── model_components ├── __init__.py ├── losses.py └── mlp.py ├── models ├── __init__.py ├── base_model.py └── scrfacto.py ├── pipelines ├── __init__.py ├── base_pipeline.py └── node2vec_pipeline.py ├── scripts ├── completions │ ├── .gitignore │ ├── __init__.py │ ├── install.py │ ├── setup.bash │ └── setup.zsh ├── downloads │ ├── __init__.py │ ├── aachen.py │ ├── download_data.py │ ├── naver_lab.py │ └── utils.py ├── encoding_pca.py ├── eval.py ├── overlap_score.py ├── process_data.py ├── retrieval_feat.py └── train.py └── utils ├── __init__.py ├── comms.py ├── decorators.py ├── install_checks.py ├── io.py ├── misc.py ├── poses.py ├── printing.py ├── profiler.py ├── rich_utils.py ├── scripts.py └── writer.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/.gitignore -------------------------------------------------------------------------------- /.gitmodules: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/.gitmodules -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/README.md -------------------------------------------------------------------------------- /docs/data_conventions.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/docs/data_conventions.md -------------------------------------------------------------------------------- /docs/module.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/docs/module.png -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/pyproject.toml -------------------------------------------------------------------------------- /scrstudio/configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/configs/__init__.py -------------------------------------------------------------------------------- /scrstudio/configs/base_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/configs/base_config.py -------------------------------------------------------------------------------- /scrstudio/configs/config_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/configs/config_utils.py -------------------------------------------------------------------------------- /scrstudio/configs/experiment_config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/configs/experiment_config.py -------------------------------------------------------------------------------- /scrstudio/configs/method_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/configs/method_configs.py -------------------------------------------------------------------------------- /scrstudio/data/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/data/__init__.py -------------------------------------------------------------------------------- /scrstudio/data/datamanagers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/data/datamanagers/__init__.py -------------------------------------------------------------------------------- /scrstudio/data/datamanagers/base_datamanager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/data/datamanagers/base_datamanager.py -------------------------------------------------------------------------------- /scrstudio/data/datamanagers/glbuffer_datamanager.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/data/datamanagers/glbuffer_datamanager.py -------------------------------------------------------------------------------- /scrstudio/data/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/data/datasets/__init__.py -------------------------------------------------------------------------------- /scrstudio/data/datasets/camloc_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/data/datasets/camloc_dataset.py -------------------------------------------------------------------------------- /scrstudio/data/samplers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/data/samplers.py -------------------------------------------------------------------------------- /scrstudio/data/utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scrstudio/data/utils/readers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/data/utils/readers.py -------------------------------------------------------------------------------- /scrstudio/encoders/ace_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/encoders/ace_encoder.py -------------------------------------------------------------------------------- /scrstudio/encoders/base_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/encoders/base_encoder.py -------------------------------------------------------------------------------- /scrstudio/encoders/dedode_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/encoders/dedode_encoder.py -------------------------------------------------------------------------------- /scrstudio/encoders/loftr_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/encoders/loftr_encoder.py -------------------------------------------------------------------------------- /scrstudio/encoders/netvlad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/encoders/netvlad.py -------------------------------------------------------------------------------- /scrstudio/encoders/pca_encoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/encoders/pca_encoder.py -------------------------------------------------------------------------------- /scrstudio/engine/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/engine/__init__.py -------------------------------------------------------------------------------- /scrstudio/engine/callbacks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/engine/callbacks.py -------------------------------------------------------------------------------- /scrstudio/engine/optimizers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/engine/optimizers.py -------------------------------------------------------------------------------- /scrstudio/engine/schedulers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/engine/schedulers.py -------------------------------------------------------------------------------- /scrstudio/engine/trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/engine/trainer.py -------------------------------------------------------------------------------- /scrstudio/model_components/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/model_components/__init__.py -------------------------------------------------------------------------------- /scrstudio/model_components/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/model_components/losses.py -------------------------------------------------------------------------------- /scrstudio/model_components/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/model_components/mlp.py -------------------------------------------------------------------------------- /scrstudio/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/models/__init__.py -------------------------------------------------------------------------------- /scrstudio/models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/models/base_model.py -------------------------------------------------------------------------------- /scrstudio/models/scrfacto.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/models/scrfacto.py -------------------------------------------------------------------------------- /scrstudio/pipelines/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/pipelines/__init__.py -------------------------------------------------------------------------------- /scrstudio/pipelines/base_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/pipelines/base_pipeline.py -------------------------------------------------------------------------------- /scrstudio/pipelines/node2vec_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/pipelines/node2vec_pipeline.py -------------------------------------------------------------------------------- /scrstudio/scripts/completions/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/scripts/completions/.gitignore -------------------------------------------------------------------------------- /scrstudio/scripts/completions/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/scripts/completions/__init__.py -------------------------------------------------------------------------------- /scrstudio/scripts/completions/install.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/scripts/completions/install.py -------------------------------------------------------------------------------- /scrstudio/scripts/completions/setup.bash: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/scripts/completions/setup.bash -------------------------------------------------------------------------------- /scrstudio/scripts/completions/setup.zsh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/scripts/completions/setup.zsh -------------------------------------------------------------------------------- /scrstudio/scripts/downloads/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/scripts/downloads/__init__.py -------------------------------------------------------------------------------- /scrstudio/scripts/downloads/aachen.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/scripts/downloads/aachen.py -------------------------------------------------------------------------------- /scrstudio/scripts/downloads/download_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/scripts/downloads/download_data.py -------------------------------------------------------------------------------- /scrstudio/scripts/downloads/naver_lab.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/scripts/downloads/naver_lab.py -------------------------------------------------------------------------------- /scrstudio/scripts/downloads/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/scripts/downloads/utils.py -------------------------------------------------------------------------------- /scrstudio/scripts/encoding_pca.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/scripts/encoding_pca.py -------------------------------------------------------------------------------- /scrstudio/scripts/eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/scripts/eval.py -------------------------------------------------------------------------------- /scrstudio/scripts/overlap_score.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/scripts/overlap_score.py -------------------------------------------------------------------------------- /scrstudio/scripts/process_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/scripts/process_data.py -------------------------------------------------------------------------------- /scrstudio/scripts/retrieval_feat.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/scripts/retrieval_feat.py -------------------------------------------------------------------------------- /scrstudio/scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/scripts/train.py -------------------------------------------------------------------------------- /scrstudio/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/utils/__init__.py -------------------------------------------------------------------------------- /scrstudio/utils/comms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/utils/comms.py -------------------------------------------------------------------------------- /scrstudio/utils/decorators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/utils/decorators.py -------------------------------------------------------------------------------- /scrstudio/utils/install_checks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/utils/install_checks.py -------------------------------------------------------------------------------- /scrstudio/utils/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/utils/io.py -------------------------------------------------------------------------------- /scrstudio/utils/misc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/utils/misc.py -------------------------------------------------------------------------------- /scrstudio/utils/poses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/utils/poses.py -------------------------------------------------------------------------------- /scrstudio/utils/printing.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/utils/printing.py -------------------------------------------------------------------------------- /scrstudio/utils/profiler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/utils/profiler.py -------------------------------------------------------------------------------- /scrstudio/utils/rich_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/utils/rich_utils.py -------------------------------------------------------------------------------- /scrstudio/utils/scripts.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/utils/scripts.py -------------------------------------------------------------------------------- /scrstudio/utils/writer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/cvg/scrstudio/HEAD/scrstudio/utils/writer.py --------------------------------------------------------------------------------