├── .gitignore ├── LICENSE ├── README.md ├── assets ├── scannet_eval_list.txt └── teaser.png ├── components ├── __init__.py ├── evaluators.py ├── extractors.py ├── load_component.py ├── matchers.py └── readers.py ├── datadump ├── check_training_data.py ├── configs │ ├── fmbench_root.yaml │ ├── fmbench_sp.yaml │ ├── gl3d.yaml │ ├── scannet_root.yaml │ ├── scannet_sp.yaml │ ├── yfcc_root.yaml │ └── yfcc_sp.yaml ├── dump.py └── dumper │ ├── base_dumper.py │ ├── fmbench.py │ ├── gl3d_train.py │ ├── scannet.py │ └── yfcc.py ├── demo ├── configs │ ├── nn_config.yaml │ └── sgm_config.yaml ├── demo.py ├── demo_1.jpg └── demo_2.jpg ├── evaluation ├── configs │ ├── cost │ │ ├── sg_cost.yaml │ │ └── sgm_cost.yaml │ └── eval │ │ ├── fm_eval_nn.yaml │ │ ├── fm_eval_sg.yaml │ │ ├── fm_eval_sgm.yaml │ │ ├── scannet_eval_nn.yaml │ │ ├── scannet_eval_sg.yaml │ │ ├── scannet_eval_sgm.yaml │ │ ├── yfcc_eval_nn.yaml │ │ ├── yfcc_eval_sg.yaml │ │ └── yfcc_eval_sgm.yaml ├── eval_cost.py └── evaluate.py ├── requirements.txt ├── sgmnet ├── __init__.py └── match_model.py ├── superglue ├── __init__.py └── match_model.py ├── superpoint ├── __init__.py └── superpoint.py ├── train ├── config.py ├── configs │ ├── sg.yaml │ └── sgm.yaml ├── dataset.py ├── loss.py ├── main.py ├── train.py ├── train_sg.sh ├── train_sgm.sh └── valid.py └── utils ├── __init__.py ├── data_utils.py ├── evaluation_utils.py ├── fm_utils.py ├── metrics.py ├── train_utils.py └── transformations.py /.gitignore: -------------------------------------------------------------------------------- 1 | *.pyc -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/README.md -------------------------------------------------------------------------------- /assets/scannet_eval_list.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/assets/scannet_eval_list.txt -------------------------------------------------------------------------------- /assets/teaser.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/assets/teaser.png -------------------------------------------------------------------------------- /components/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/components/__init__.py -------------------------------------------------------------------------------- /components/evaluators.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/components/evaluators.py -------------------------------------------------------------------------------- /components/extractors.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/components/extractors.py -------------------------------------------------------------------------------- /components/load_component.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/components/load_component.py -------------------------------------------------------------------------------- /components/matchers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/components/matchers.py -------------------------------------------------------------------------------- /components/readers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/components/readers.py -------------------------------------------------------------------------------- /datadump/check_training_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/datadump/check_training_data.py -------------------------------------------------------------------------------- /datadump/configs/fmbench_root.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/datadump/configs/fmbench_root.yaml -------------------------------------------------------------------------------- /datadump/configs/fmbench_sp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/datadump/configs/fmbench_sp.yaml -------------------------------------------------------------------------------- /datadump/configs/gl3d.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/datadump/configs/gl3d.yaml -------------------------------------------------------------------------------- /datadump/configs/scannet_root.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/datadump/configs/scannet_root.yaml -------------------------------------------------------------------------------- /datadump/configs/scannet_sp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/datadump/configs/scannet_sp.yaml -------------------------------------------------------------------------------- /datadump/configs/yfcc_root.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/datadump/configs/yfcc_root.yaml -------------------------------------------------------------------------------- /datadump/configs/yfcc_sp.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/datadump/configs/yfcc_sp.yaml -------------------------------------------------------------------------------- /datadump/dump.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/datadump/dump.py -------------------------------------------------------------------------------- /datadump/dumper/base_dumper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/datadump/dumper/base_dumper.py -------------------------------------------------------------------------------- /datadump/dumper/fmbench.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/datadump/dumper/fmbench.py -------------------------------------------------------------------------------- /datadump/dumper/gl3d_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/datadump/dumper/gl3d_train.py -------------------------------------------------------------------------------- /datadump/dumper/scannet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/datadump/dumper/scannet.py -------------------------------------------------------------------------------- /datadump/dumper/yfcc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/datadump/dumper/yfcc.py -------------------------------------------------------------------------------- /demo/configs/nn_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/demo/configs/nn_config.yaml -------------------------------------------------------------------------------- /demo/configs/sgm_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/demo/configs/sgm_config.yaml -------------------------------------------------------------------------------- /demo/demo.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/demo/demo.py -------------------------------------------------------------------------------- /demo/demo_1.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/demo/demo_1.jpg -------------------------------------------------------------------------------- /demo/demo_2.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/demo/demo_2.jpg -------------------------------------------------------------------------------- /evaluation/configs/cost/sg_cost.yaml: -------------------------------------------------------------------------------- 1 | net_channels: 128 2 | layer_num: 9 3 | head: 4 4 | use_score_encoding: True -------------------------------------------------------------------------------- /evaluation/configs/cost/sgm_cost.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/evaluation/configs/cost/sgm_cost.yaml -------------------------------------------------------------------------------- /evaluation/configs/eval/fm_eval_nn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/evaluation/configs/eval/fm_eval_nn.yaml -------------------------------------------------------------------------------- /evaluation/configs/eval/fm_eval_sg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/evaluation/configs/eval/fm_eval_sg.yaml -------------------------------------------------------------------------------- /evaluation/configs/eval/fm_eval_sgm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/evaluation/configs/eval/fm_eval_sgm.yaml -------------------------------------------------------------------------------- /evaluation/configs/eval/scannet_eval_nn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/evaluation/configs/eval/scannet_eval_nn.yaml -------------------------------------------------------------------------------- /evaluation/configs/eval/scannet_eval_sg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/evaluation/configs/eval/scannet_eval_sg.yaml -------------------------------------------------------------------------------- /evaluation/configs/eval/scannet_eval_sgm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/evaluation/configs/eval/scannet_eval_sgm.yaml -------------------------------------------------------------------------------- /evaluation/configs/eval/yfcc_eval_nn.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/evaluation/configs/eval/yfcc_eval_nn.yaml -------------------------------------------------------------------------------- /evaluation/configs/eval/yfcc_eval_sg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/evaluation/configs/eval/yfcc_eval_sg.yaml -------------------------------------------------------------------------------- /evaluation/configs/eval/yfcc_eval_sgm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/evaluation/configs/eval/yfcc_eval_sgm.yaml -------------------------------------------------------------------------------- /evaluation/eval_cost.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/evaluation/eval_cost.py -------------------------------------------------------------------------------- /evaluation/evaluate.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/evaluation/evaluate.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/requirements.txt -------------------------------------------------------------------------------- /sgmnet/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/sgmnet/__init__.py -------------------------------------------------------------------------------- /sgmnet/match_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/sgmnet/match_model.py -------------------------------------------------------------------------------- /superglue/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/superglue/__init__.py -------------------------------------------------------------------------------- /superglue/match_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/superglue/match_model.py -------------------------------------------------------------------------------- /superpoint/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/superpoint/__init__.py -------------------------------------------------------------------------------- /superpoint/superpoint.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/superpoint/superpoint.py -------------------------------------------------------------------------------- /train/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/train/config.py -------------------------------------------------------------------------------- /train/configs/sg.yaml: -------------------------------------------------------------------------------- 1 | net_channels: 128 2 | layer_num: 9 3 | head: 4 4 | use_score_encoding: True 5 | p_th: 0.2 -------------------------------------------------------------------------------- /train/configs/sgm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/train/configs/sgm.yaml -------------------------------------------------------------------------------- /train/dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/train/dataset.py -------------------------------------------------------------------------------- /train/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/train/loss.py -------------------------------------------------------------------------------- /train/main.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/train/main.py -------------------------------------------------------------------------------- /train/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/train/train.py -------------------------------------------------------------------------------- /train/train_sg.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/train/train_sg.sh -------------------------------------------------------------------------------- /train/train_sgm.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/train/train_sgm.sh -------------------------------------------------------------------------------- /train/valid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/train/valid.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/data_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/utils/data_utils.py -------------------------------------------------------------------------------- /utils/evaluation_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/utils/evaluation_utils.py -------------------------------------------------------------------------------- /utils/fm_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/utils/fm_utils.py -------------------------------------------------------------------------------- /utils/metrics.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/utils/metrics.py -------------------------------------------------------------------------------- /utils/train_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/utils/train_utils.py -------------------------------------------------------------------------------- /utils/transformations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vdvchen/SGMNet/HEAD/utils/transformations.py --------------------------------------------------------------------------------