├── .gitignore ├── LICENSE ├── README.md ├── config ├── __init__.py └── defaults.py ├── data ├── __init__.py ├── pointclouds.py └── preprocess │ ├── __init__.py │ ├── compute_kpt_pairs.py │ ├── compute_overlap.py │ ├── compute_radius.py │ └── fuse_fragments_3DMatch.py ├── docker ├── Dockerfile └── build.sh ├── evaluation ├── __init__.py ├── eval_geomreg_3dmatch.py ├── eval_geomreg_3dmatch.sh ├── eval_geomreg_eth.py └── eval_geomreg_eth.sh ├── figures └── pipeline.png ├── models ├── __init__.py ├── base_model.py ├── modules.py └── mvdesc.py ├── requirements.txt ├── scripts ├── __init__.py ├── configs │ ├── ours_3dmatch.yaml │ └── ours_eth.yaml ├── engine_utils.py ├── main_mvdesc.py └── ours_3dmatch │ ├── net_cnn_16.pth │ ├── net_embed_16.pth │ ├── net_pool_16.pth │ └── net_renderer_16.pth ├── soft_renderer ├── __init__.py ├── cuda │ ├── __init__.py │ ├── jit.py │ ├── soft_rasterize_cuda.cpp │ ├── soft_rasterize_cuda_kernel.cu │ └── utils.cuh └── transform.py └── utils ├── __init__.py ├── io.py ├── log.py ├── loss.py └── meters.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/README.md -------------------------------------------------------------------------------- /config/__init__.py: -------------------------------------------------------------------------------- 1 | from .defaults import M as mvdesc_cfg 2 | -------------------------------------------------------------------------------- /config/defaults.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/config/defaults.py -------------------------------------------------------------------------------- /data/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/pointclouds.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/data/pointclouds.py -------------------------------------------------------------------------------- /data/preprocess/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /data/preprocess/compute_kpt_pairs.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/data/preprocess/compute_kpt_pairs.py -------------------------------------------------------------------------------- /data/preprocess/compute_overlap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/data/preprocess/compute_overlap.py -------------------------------------------------------------------------------- /data/preprocess/compute_radius.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/data/preprocess/compute_radius.py -------------------------------------------------------------------------------- /data/preprocess/fuse_fragments_3DMatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/data/preprocess/fuse_fragments_3DMatch.py -------------------------------------------------------------------------------- /docker/Dockerfile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/docker/Dockerfile -------------------------------------------------------------------------------- /docker/build.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/docker/build.sh -------------------------------------------------------------------------------- /evaluation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /evaluation/eval_geomreg_3dmatch.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/evaluation/eval_geomreg_3dmatch.py -------------------------------------------------------------------------------- /evaluation/eval_geomreg_3dmatch.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/evaluation/eval_geomreg_3dmatch.sh -------------------------------------------------------------------------------- /evaluation/eval_geomreg_eth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/evaluation/eval_geomreg_eth.py -------------------------------------------------------------------------------- /evaluation/eval_geomreg_eth.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/evaluation/eval_geomreg_eth.sh -------------------------------------------------------------------------------- /figures/pipeline.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/figures/pipeline.png -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/models/base_model.py -------------------------------------------------------------------------------- /models/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/models/modules.py -------------------------------------------------------------------------------- /models/mvdesc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/models/mvdesc.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /scripts/configs/ours_3dmatch.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/scripts/configs/ours_3dmatch.yaml -------------------------------------------------------------------------------- /scripts/configs/ours_eth.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/scripts/configs/ours_eth.yaml -------------------------------------------------------------------------------- /scripts/engine_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/scripts/engine_utils.py -------------------------------------------------------------------------------- /scripts/main_mvdesc.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/scripts/main_mvdesc.py -------------------------------------------------------------------------------- /scripts/ours_3dmatch/net_cnn_16.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/scripts/ours_3dmatch/net_cnn_16.pth -------------------------------------------------------------------------------- /scripts/ours_3dmatch/net_embed_16.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/scripts/ours_3dmatch/net_embed_16.pth -------------------------------------------------------------------------------- /scripts/ours_3dmatch/net_pool_16.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/scripts/ours_3dmatch/net_pool_16.pth -------------------------------------------------------------------------------- /scripts/ours_3dmatch/net_renderer_16.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/scripts/ours_3dmatch/net_renderer_16.pth -------------------------------------------------------------------------------- /soft_renderer/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /soft_renderer/cuda/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /soft_renderer/cuda/jit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/soft_renderer/cuda/jit.py -------------------------------------------------------------------------------- /soft_renderer/cuda/soft_rasterize_cuda.cpp: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/soft_renderer/cuda/soft_rasterize_cuda.cpp -------------------------------------------------------------------------------- /soft_renderer/cuda/soft_rasterize_cuda_kernel.cu: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/soft_renderer/cuda/soft_rasterize_cuda_kernel.cu -------------------------------------------------------------------------------- /soft_renderer/cuda/utils.cuh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/soft_renderer/cuda/utils.cuh -------------------------------------------------------------------------------- /soft_renderer/transform.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/soft_renderer/transform.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /utils/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/utils/io.py -------------------------------------------------------------------------------- /utils/log.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/utils/log.py -------------------------------------------------------------------------------- /utils/loss.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/utils/loss.py -------------------------------------------------------------------------------- /utils/meters.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/craigleili/3DLocalMultiViewDesc/HEAD/utils/meters.py --------------------------------------------------------------------------------