├── .gitattributes ├── .gitignore ├── Pretrain_gis.py ├── Pretrain_mm.py ├── README.md ├── Rerank.py ├── Retrieval.py ├── configs ├── Pretrain_gis.yaml ├── Pretrain_mm.yaml ├── Rerank.yaml └── Retrieval.yaml ├── data └── datasets.zip ├── dataset ├── __init__.py ├── mgeo_dataset.py └── utils.py ├── models ├── model_pretrain_gis.py ├── model_pretrain_mm.py ├── model_rerank.py ├── model_retrieval.py ├── tokenization_bert.py └── xbert.py ├── optim ├── __init__.py ├── adafactor.py ├── adahessian.py ├── adamp.py ├── adamw.py ├── lookahead.py ├── nadam.py ├── novograd.py ├── nvnovograd.py ├── optim_factory.py ├── radam.py ├── rmsprop_tf.py └── sgdp.py ├── output └── __init__.py ├── prepare_data ├── calculate_geographic_context.py └── download_pretrain_models.sh ├── requirements.txt ├── resources ├── gis_config.json ├── hz_aoi.txt ├── hz_roads.txt ├── location_in_aoi.demo ├── location_near_aoi.demo ├── location_near_road.demo └── text_location_pair.demo ├── run_gis_encoder_pretrain.sh ├── run_mm_pretrain.sh ├── run_rerank.sh ├── run_retrieval.sh ├── scheduler ├── __init__.py ├── cosine_lr.py ├── plateau_lr.py ├── scheduler.py ├── scheduler_factory.py ├── step_lr.py └── tanh_lr.py └── utils.py /.gitattributes: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/.gitattributes -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/.gitignore -------------------------------------------------------------------------------- /Pretrain_gis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/Pretrain_gis.py -------------------------------------------------------------------------------- /Pretrain_mm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/Pretrain_mm.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/README.md -------------------------------------------------------------------------------- /Rerank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/Rerank.py -------------------------------------------------------------------------------- /Retrieval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/Retrieval.py -------------------------------------------------------------------------------- /configs/Pretrain_gis.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/configs/Pretrain_gis.yaml -------------------------------------------------------------------------------- /configs/Pretrain_mm.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/configs/Pretrain_mm.yaml -------------------------------------------------------------------------------- /configs/Rerank.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/configs/Rerank.yaml -------------------------------------------------------------------------------- /configs/Retrieval.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/configs/Retrieval.yaml -------------------------------------------------------------------------------- /data/datasets.zip: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/data/datasets.zip -------------------------------------------------------------------------------- /dataset/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/dataset/__init__.py -------------------------------------------------------------------------------- /dataset/mgeo_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/dataset/mgeo_dataset.py -------------------------------------------------------------------------------- /dataset/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/dataset/utils.py -------------------------------------------------------------------------------- /models/model_pretrain_gis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/models/model_pretrain_gis.py -------------------------------------------------------------------------------- /models/model_pretrain_mm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/models/model_pretrain_mm.py -------------------------------------------------------------------------------- /models/model_rerank.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/models/model_rerank.py -------------------------------------------------------------------------------- /models/model_retrieval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/models/model_retrieval.py -------------------------------------------------------------------------------- /models/tokenization_bert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/models/tokenization_bert.py -------------------------------------------------------------------------------- /models/xbert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/models/xbert.py -------------------------------------------------------------------------------- /optim/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/optim/__init__.py -------------------------------------------------------------------------------- /optim/adafactor.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/optim/adafactor.py -------------------------------------------------------------------------------- /optim/adahessian.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/optim/adahessian.py -------------------------------------------------------------------------------- /optim/adamp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/optim/adamp.py -------------------------------------------------------------------------------- /optim/adamw.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/optim/adamw.py -------------------------------------------------------------------------------- /optim/lookahead.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/optim/lookahead.py -------------------------------------------------------------------------------- /optim/nadam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/optim/nadam.py -------------------------------------------------------------------------------- /optim/novograd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/optim/novograd.py -------------------------------------------------------------------------------- /optim/nvnovograd.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/optim/nvnovograd.py -------------------------------------------------------------------------------- /optim/optim_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/optim/optim_factory.py -------------------------------------------------------------------------------- /optim/radam.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/optim/radam.py -------------------------------------------------------------------------------- /optim/rmsprop_tf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/optim/rmsprop_tf.py -------------------------------------------------------------------------------- /optim/sgdp.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/optim/sgdp.py -------------------------------------------------------------------------------- /output/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /prepare_data/calculate_geographic_context.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/prepare_data/calculate_geographic_context.py -------------------------------------------------------------------------------- /prepare_data/download_pretrain_models.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/prepare_data/download_pretrain_models.sh -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/requirements.txt -------------------------------------------------------------------------------- /resources/gis_config.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/resources/gis_config.json -------------------------------------------------------------------------------- /resources/hz_aoi.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/resources/hz_aoi.txt -------------------------------------------------------------------------------- /resources/hz_roads.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/resources/hz_roads.txt -------------------------------------------------------------------------------- /resources/location_in_aoi.demo: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/resources/location_in_aoi.demo -------------------------------------------------------------------------------- /resources/location_near_aoi.demo: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/resources/location_near_aoi.demo -------------------------------------------------------------------------------- /resources/location_near_road.demo: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/resources/location_near_road.demo -------------------------------------------------------------------------------- /resources/text_location_pair.demo: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/resources/text_location_pair.demo -------------------------------------------------------------------------------- /run_gis_encoder_pretrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/run_gis_encoder_pretrain.sh -------------------------------------------------------------------------------- /run_mm_pretrain.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/run_mm_pretrain.sh -------------------------------------------------------------------------------- /run_rerank.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/run_rerank.sh -------------------------------------------------------------------------------- /run_retrieval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/run_retrieval.sh -------------------------------------------------------------------------------- /scheduler/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/scheduler/__init__.py -------------------------------------------------------------------------------- /scheduler/cosine_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/scheduler/cosine_lr.py -------------------------------------------------------------------------------- /scheduler/plateau_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/scheduler/plateau_lr.py -------------------------------------------------------------------------------- /scheduler/scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/scheduler/scheduler.py -------------------------------------------------------------------------------- /scheduler/scheduler_factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/scheduler/scheduler_factory.py -------------------------------------------------------------------------------- /scheduler/step_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/scheduler/step_lr.py -------------------------------------------------------------------------------- /scheduler/tanh_lr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/scheduler/tanh_lr.py -------------------------------------------------------------------------------- /utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/PhantomGrapes/MGeo/HEAD/utils.py --------------------------------------------------------------------------------