├── .gitignore ├── LICENSE ├── README.md ├── assets ├── sacre_coeur_A.jpg ├── sacre_coeur_B.jpg ├── toronto_A.jpg └── toronto_B.jpg ├── data └── .gitignore ├── demo ├── demo_3D_effect.py ├── demo_fundamental.py ├── demo_match.py ├── demo_match_opencv_sift.py ├── demo_match_tiny.py └── gif │ └── .gitignore ├── experiments ├── eval_roma_outdoor.py ├── eval_tiny_roma_v1_outdoor.py ├── roma_indoor.py ├── train_roma_outdoor.py └── train_tiny_roma_v1_outdoor.py ├── requirements.txt ├── romatch ├── __init__.py ├── benchmarks │ ├── __init__.py │ ├── hpatches_sequences_homog_benchmark.py │ ├── megadepth_dense_benchmark.py │ ├── megadepth_pose_estimation_benchmark.py │ ├── megadepth_pose_estimation_benchmark_poselib.py │ └── scannet_benchmark.py ├── checkpointing │ ├── __init__.py │ └── checkpoint.py ├── datasets │ ├── __init__.py │ ├── megadepth.py │ └── scannet.py ├── losses │ ├── __init__.py │ ├── robust_loss.py │ └── robust_loss_tiny_roma.py ├── models │ ├── __init__.py │ ├── encoders.py │ ├── matcher.py │ ├── model_zoo │ │ ├── __init__.py │ │ └── roma_models.py │ ├── tiny.py │ └── transformer │ │ ├── __init__.py │ │ ├── dinov2.py │ │ └── layers │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── block.py │ │ ├── dino_head.py │ │ ├── drop_path.py │ │ ├── layer_scale.py │ │ ├── mlp.py │ │ ├── patch_embed.py │ │ └── swiglu_ffn.py ├── train │ ├── __init__.py │ └── train.py └── utils │ ├── __init__.py │ ├── kde.py │ ├── local_correlation.py │ ├── transforms.py │ └── utils.py └── setup.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/README.md -------------------------------------------------------------------------------- /assets/sacre_coeur_A.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/assets/sacre_coeur_A.jpg -------------------------------------------------------------------------------- /assets/sacre_coeur_B.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/assets/sacre_coeur_B.jpg -------------------------------------------------------------------------------- /assets/toronto_A.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/assets/toronto_A.jpg -------------------------------------------------------------------------------- /assets/toronto_B.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/assets/toronto_B.jpg -------------------------------------------------------------------------------- /data/.gitignore: -------------------------------------------------------------------------------- 1 | * 2 | !.gitignore -------------------------------------------------------------------------------- /demo/demo_3D_effect.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/demo/demo_3D_effect.py -------------------------------------------------------------------------------- /demo/demo_fundamental.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/demo/demo_fundamental.py -------------------------------------------------------------------------------- /demo/demo_match.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/demo/demo_match.py -------------------------------------------------------------------------------- /demo/demo_match_opencv_sift.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/demo/demo_match_opencv_sift.py -------------------------------------------------------------------------------- /demo/demo_match_tiny.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/demo/demo_match_tiny.py -------------------------------------------------------------------------------- /demo/gif/.gitignore: -------------------------------------------------------------------------------- 1 | * 2 | !.gitignore -------------------------------------------------------------------------------- /experiments/eval_roma_outdoor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/experiments/eval_roma_outdoor.py -------------------------------------------------------------------------------- /experiments/eval_tiny_roma_v1_outdoor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/experiments/eval_tiny_roma_v1_outdoor.py -------------------------------------------------------------------------------- /experiments/roma_indoor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/experiments/roma_indoor.py -------------------------------------------------------------------------------- /experiments/train_roma_outdoor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/experiments/train_roma_outdoor.py -------------------------------------------------------------------------------- /experiments/train_tiny_roma_v1_outdoor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/experiments/train_tiny_roma_v1_outdoor.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/requirements.txt -------------------------------------------------------------------------------- /romatch/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/__init__.py -------------------------------------------------------------------------------- /romatch/benchmarks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/benchmarks/__init__.py -------------------------------------------------------------------------------- /romatch/benchmarks/hpatches_sequences_homog_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/benchmarks/hpatches_sequences_homog_benchmark.py -------------------------------------------------------------------------------- /romatch/benchmarks/megadepth_dense_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/benchmarks/megadepth_dense_benchmark.py -------------------------------------------------------------------------------- /romatch/benchmarks/megadepth_pose_estimation_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/benchmarks/megadepth_pose_estimation_benchmark.py -------------------------------------------------------------------------------- /romatch/benchmarks/megadepth_pose_estimation_benchmark_poselib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/benchmarks/megadepth_pose_estimation_benchmark_poselib.py -------------------------------------------------------------------------------- /romatch/benchmarks/scannet_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/benchmarks/scannet_benchmark.py -------------------------------------------------------------------------------- /romatch/checkpointing/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/checkpointing/__init__.py -------------------------------------------------------------------------------- /romatch/checkpointing/checkpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/checkpointing/checkpoint.py -------------------------------------------------------------------------------- /romatch/datasets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/datasets/__init__.py -------------------------------------------------------------------------------- /romatch/datasets/megadepth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/datasets/megadepth.py -------------------------------------------------------------------------------- /romatch/datasets/scannet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/datasets/scannet.py -------------------------------------------------------------------------------- /romatch/losses/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/losses/__init__.py -------------------------------------------------------------------------------- /romatch/losses/robust_loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/losses/robust_loss.py -------------------------------------------------------------------------------- /romatch/losses/robust_loss_tiny_roma.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/losses/robust_loss_tiny_roma.py -------------------------------------------------------------------------------- /romatch/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/models/__init__.py -------------------------------------------------------------------------------- /romatch/models/encoders.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/models/encoders.py -------------------------------------------------------------------------------- /romatch/models/matcher.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/models/matcher.py -------------------------------------------------------------------------------- /romatch/models/model_zoo/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/models/model_zoo/__init__.py -------------------------------------------------------------------------------- /romatch/models/model_zoo/roma_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/models/model_zoo/roma_models.py -------------------------------------------------------------------------------- /romatch/models/tiny.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/models/tiny.py -------------------------------------------------------------------------------- /romatch/models/transformer/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/models/transformer/__init__.py -------------------------------------------------------------------------------- /romatch/models/transformer/dinov2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/models/transformer/dinov2.py -------------------------------------------------------------------------------- /romatch/models/transformer/layers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/models/transformer/layers/__init__.py -------------------------------------------------------------------------------- /romatch/models/transformer/layers/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/models/transformer/layers/attention.py -------------------------------------------------------------------------------- /romatch/models/transformer/layers/block.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/models/transformer/layers/block.py -------------------------------------------------------------------------------- /romatch/models/transformer/layers/dino_head.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/models/transformer/layers/dino_head.py -------------------------------------------------------------------------------- /romatch/models/transformer/layers/drop_path.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/models/transformer/layers/drop_path.py -------------------------------------------------------------------------------- /romatch/models/transformer/layers/layer_scale.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/models/transformer/layers/layer_scale.py -------------------------------------------------------------------------------- /romatch/models/transformer/layers/mlp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/models/transformer/layers/mlp.py -------------------------------------------------------------------------------- /romatch/models/transformer/layers/patch_embed.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/models/transformer/layers/patch_embed.py -------------------------------------------------------------------------------- /romatch/models/transformer/layers/swiglu_ffn.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/models/transformer/layers/swiglu_ffn.py -------------------------------------------------------------------------------- /romatch/train/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/train/__init__.py -------------------------------------------------------------------------------- /romatch/train/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/train/train.py -------------------------------------------------------------------------------- /romatch/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/utils/__init__.py -------------------------------------------------------------------------------- /romatch/utils/kde.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/utils/kde.py -------------------------------------------------------------------------------- /romatch/utils/local_correlation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/utils/local_correlation.py -------------------------------------------------------------------------------- /romatch/utils/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/utils/transforms.py -------------------------------------------------------------------------------- /romatch/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/romatch/utils/utils.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Vincentqyw/RoMa/HEAD/setup.py --------------------------------------------------------------------------------