CurriculumLoc is a PyTorch implementation for our paper "CurriculumLoc: Enhancing Cross-Domain Geolocalization through Multi-Stage Refinement". If you use this code for your research, please cite our paper. For additional questions contact us via huboni@mail.nwpu.edu.cn or huboni7@gmail.com.
We test this repo with Python 3.10, PyTorch 1.12.1, and CUDA 11.3. However, it should be runnable with recent PyTorch versions. You can install by conda with our prove environment.taml.
conda env create -f environment.yaml
We test our models on two datasets. One dataset is ALTO, it can be download at here. Another is our TerraTrack, TerraTrack is being prepared.... All of these datasets contain some challenging environmental variations, as shown in below table.
python train_terra.py
python match_localization.py
If you're using CurculumLoc in your research or applications, please cite using this BibTeX:
@misc{hu2023curriculumloc,
title={CurriculumLoc: Enhancing Cross-Domain Geolocalization through Multi-Stage Refinement},
author={Boni Hu and Lin Chen and Runjian Chen and Shuhui Bu and Pengcheng Han and Haowei Li},
year={2023},
eprint={2311.11604},
archivePrefix={arXiv},
primaryClass={cs.CV}
}