├── .gitattributes ├── .gitignore ├── LICENSE ├── README.md ├── ckpt ├── awdesc-ca │ └── readme.md ├── awdesc-t16 │ └── readme.md └── awdesc-t32 │ └── readme.md ├── configs ├── AWDesc_eva.yaml ├── AWDesc_train_CA.yaml ├── AWDesc_train_Tiny.yaml ├── SuperPoint_train.yaml └── readme.md ├── data_utils ├── __init__.py ├── dataset_tools.py ├── megadepth_train_dataset.py └── megadepth_train_dataset_dl.py ├── evaluation_hpatch ├── .ipynb_checkpoints │ └── HPatches-Sequences-Matching-Benchmark-checkpoint.ipynb ├── __init__.py ├── export.py ├── hpatch_related │ ├── hpatch_dataset.py │ └── hpatch_evaluation.py ├── hpatches_sequences │ ├── HPatches-Sequences-Matching-Benchmark.ipynb │ ├── README.md │ ├── cache │ │ └── mtldesc.npy │ ├── convert_to_png.sh │ ├── download.sh │ ├── download_cache.sh │ └── readme.md ├── models │ ├── MTLDesc.py │ └── __init__.py └── utils │ ├── common.py │ ├── d2net_pyramid.py │ ├── d2net_utils.py │ ├── evaluation_tools.py │ ├── evaluator.py │ ├── logger.py │ └── utils.py ├── nets ├── __init__.py ├── network.py ├── utils.py └── vit │ ├── __init__.py │ ├── vit_seg_configs.py │ ├── vit_seg_modeling.py │ └── vit_seg_modeling_resnet_skip.py ├── readme.md ├── requirements.txt ├── train.py ├── trainers ├── __init__.py ├── awdesc_trainer.py ├── base_trainer.py ├── mtldesc_trainer.py ├── superpoint_trainer.py └── utils.py └── utils ├── evaluation_tools.py ├── logger.py └── utils.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/README.md -------------------------------------------------------------------------------- /ckpt/awdesc-ca/readme.md: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /ckpt/awdesc-t16/readme.md: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /ckpt/awdesc-t32/readme.md: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /configs/AWDesc_eva.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/configs/AWDesc_eva.yaml -------------------------------------------------------------------------------- /configs/AWDesc_train_CA.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/configs/AWDesc_train_CA.yaml -------------------------------------------------------------------------------- /configs/AWDesc_train_Tiny.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/configs/AWDesc_train_Tiny.yaml -------------------------------------------------------------------------------- /configs/SuperPoint_train.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/configs/SuperPoint_train.yaml -------------------------------------------------------------------------------- /configs/readme.md: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /data_utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/data_utils/__init__.py -------------------------------------------------------------------------------- /data_utils/dataset_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/data_utils/dataset_tools.py -------------------------------------------------------------------------------- /data_utils/megadepth_train_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/data_utils/megadepth_train_dataset.py -------------------------------------------------------------------------------- /data_utils/megadepth_train_dataset_dl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/data_utils/megadepth_train_dataset_dl.py -------------------------------------------------------------------------------- /evaluation_hpatch/.ipynb_checkpoints/HPatches-Sequences-Matching-Benchmark-checkpoint.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/evaluation_hpatch/.ipynb_checkpoints/HPatches-Sequences-Matching-Benchmark-checkpoint.ipynb -------------------------------------------------------------------------------- /evaluation_hpatch/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /evaluation_hpatch/export.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/evaluation_hpatch/export.py -------------------------------------------------------------------------------- /evaluation_hpatch/hpatch_related/hpatch_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/evaluation_hpatch/hpatch_related/hpatch_dataset.py -------------------------------------------------------------------------------- /evaluation_hpatch/hpatch_related/hpatch_evaluation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/evaluation_hpatch/hpatch_related/hpatch_evaluation.py -------------------------------------------------------------------------------- /evaluation_hpatch/hpatches_sequences/HPatches-Sequences-Matching-Benchmark.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/evaluation_hpatch/hpatches_sequences/HPatches-Sequences-Matching-Benchmark.ipynb -------------------------------------------------------------------------------- /evaluation_hpatch/hpatches_sequences/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/evaluation_hpatch/hpatches_sequences/README.md -------------------------------------------------------------------------------- /evaluation_hpatch/hpatches_sequences/cache/mtldesc.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/evaluation_hpatch/hpatches_sequences/cache/mtldesc.npy -------------------------------------------------------------------------------- /evaluation_hpatch/hpatches_sequences/convert_to_png.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/evaluation_hpatch/hpatches_sequences/convert_to_png.sh -------------------------------------------------------------------------------- /evaluation_hpatch/hpatches_sequences/download.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/evaluation_hpatch/hpatches_sequences/download.sh -------------------------------------------------------------------------------- /evaluation_hpatch/hpatches_sequences/download_cache.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/evaluation_hpatch/hpatches_sequences/download_cache.sh -------------------------------------------------------------------------------- /evaluation_hpatch/hpatches_sequences/readme.md: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /evaluation_hpatch/models/MTLDesc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/evaluation_hpatch/models/MTLDesc.py -------------------------------------------------------------------------------- /evaluation_hpatch/models/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/evaluation_hpatch/models/__init__.py -------------------------------------------------------------------------------- /evaluation_hpatch/utils/common.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/evaluation_hpatch/utils/common.py -------------------------------------------------------------------------------- /evaluation_hpatch/utils/d2net_pyramid.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/evaluation_hpatch/utils/d2net_pyramid.py -------------------------------------------------------------------------------- /evaluation_hpatch/utils/d2net_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/evaluation_hpatch/utils/d2net_utils.py -------------------------------------------------------------------------------- /evaluation_hpatch/utils/evaluation_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/evaluation_hpatch/utils/evaluation_tools.py -------------------------------------------------------------------------------- /evaluation_hpatch/utils/evaluator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/evaluation_hpatch/utils/evaluator.py -------------------------------------------------------------------------------- /evaluation_hpatch/utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/evaluation_hpatch/utils/logger.py -------------------------------------------------------------------------------- /evaluation_hpatch/utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/evaluation_hpatch/utils/utils.py -------------------------------------------------------------------------------- /nets/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/nets/__init__.py -------------------------------------------------------------------------------- /nets/network.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/nets/network.py -------------------------------------------------------------------------------- /nets/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/nets/utils.py -------------------------------------------------------------------------------- /nets/vit/__init__.py: -------------------------------------------------------------------------------- 1 | 2 | -------------------------------------------------------------------------------- /nets/vit/vit_seg_configs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/nets/vit/vit_seg_configs.py -------------------------------------------------------------------------------- /nets/vit/vit_seg_modeling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/nets/vit/vit_seg_modeling.py -------------------------------------------------------------------------------- /nets/vit/vit_seg_modeling_resnet_skip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/nets/vit/vit_seg_modeling_resnet_skip.py -------------------------------------------------------------------------------- /readme.md: -------------------------------------------------------------------------------- 1 | Attention Weighted Local Descriptors 2 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/requirements.txt -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/train.py -------------------------------------------------------------------------------- /trainers/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/trainers/__init__.py -------------------------------------------------------------------------------- /trainers/awdesc_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/trainers/awdesc_trainer.py -------------------------------------------------------------------------------- /trainers/base_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/trainers/base_trainer.py -------------------------------------------------------------------------------- /trainers/mtldesc_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/trainers/mtldesc_trainer.py -------------------------------------------------------------------------------- /trainers/superpoint_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/trainers/superpoint_trainer.py -------------------------------------------------------------------------------- /trainers/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/trainers/utils.py -------------------------------------------------------------------------------- /utils/evaluation_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/utils/evaluation_tools.py -------------------------------------------------------------------------------- /utils/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/utils/logger.py -------------------------------------------------------------------------------- /utils/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/vignywang/AWDesc/HEAD/utils/utils.py --------------------------------------------------------------------------------