├── .gitignore ├── LICENSE ├── README.md ├── add_pca.py ├── add_pca.sh ├── assets └── vlad_buff_pipeline.jpg ├── cluster_analysis.py ├── dataloaders ├── GSVCitiesDataloader.py ├── GSVCitiesDataset.py ├── MapillaryDataset.py ├── PittsburgDataset.py └── val │ ├── AmstertimeDataset.py │ ├── BaiduDataset.py │ ├── MapillaryDataset.py │ ├── MapillaryTestDataset.py │ ├── NordlandDataset.py │ ├── PittsburghDataset.py │ ├── SPEDDataset.py │ ├── SfsmDataset.py │ ├── StluciaDataset.py │ └── Tokyo247Dataset.py ├── datasets ├── Nordland │ ├── Nordland_dbImages.npy │ ├── Nordland_gt.npy │ └── Nordland_qImages.npy ├── Pittsburgh │ ├── pitts250k_test_dbImages.npy │ ├── pitts250k_test_gt.npy │ ├── pitts250k_test_qImages.npy │ ├── pitts30k_test_dbImages.npy │ ├── pitts30k_test_gt.npy │ ├── pitts30k_test_qImages.npy │ ├── pitts30k_val_dbImages.npy │ ├── pitts30k_val_gt.npy │ └── pitts30k_val_qImages.npy ├── SPED │ ├── SPED_dbImages.npy │ ├── SPED_gt.npy │ └── SPED_qImages.npy ├── amstertime │ ├── amstertime_dbImages.npy │ └── amstertime_qImages.npy ├── baidu │ ├── baidu_dbImages.npy │ ├── baidu_dbgImages_gt.npy │ ├── baidu_qImages.npy │ └── baidu_qImages_gt.npy ├── msls_test │ ├── msls_test_dbImages.npy │ └── msls_test_qImages.npy ├── msls_val │ ├── msls_val_dbImages.npy │ ├── msls_val_pIdx.npy │ ├── msls_val_qIdx.npy │ └── msls_val_qImages.npy ├── sfsm │ ├── sfsm_dbImages.npy │ └── sfsm_qImages.npy ├── st_lucia │ ├── st_lucia_dbImages.npy │ └── st_lucia_qImages.npy └── tokyo247 │ ├── tokyo247_dbImages.npy │ └── tokyo247_qImages.npy ├── environment.yml ├── eval.py ├── eval.sh ├── hubconf.py ├── models ├── __init__.py ├── aggregators │ ├── __init__.py │ ├── aggregation.py │ ├── convap.py │ ├── cosplace.py │ ├── gem.py │ ├── mixvpr.py │ └── salad.py ├── backbones │ ├── __init__.py │ ├── dinov2.py │ └── resnet.py └── helper.py ├── predict_analysis.sh ├── predictions.py ├── sweepParams.yml ├── train.py ├── utils ├── __init__.py ├── losses.py └── validation.py └── vpr_model.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/README.md -------------------------------------------------------------------------------- /add_pca.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/add_pca.py -------------------------------------------------------------------------------- /add_pca.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/add_pca.sh -------------------------------------------------------------------------------- /assets/vlad_buff_pipeline.jpg: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/assets/vlad_buff_pipeline.jpg -------------------------------------------------------------------------------- /cluster_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/cluster_analysis.py -------------------------------------------------------------------------------- /dataloaders/GSVCitiesDataloader.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/dataloaders/GSVCitiesDataloader.py -------------------------------------------------------------------------------- /dataloaders/GSVCitiesDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/dataloaders/GSVCitiesDataset.py -------------------------------------------------------------------------------- /dataloaders/MapillaryDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/dataloaders/MapillaryDataset.py -------------------------------------------------------------------------------- /dataloaders/PittsburgDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/dataloaders/PittsburgDataset.py -------------------------------------------------------------------------------- /dataloaders/val/AmstertimeDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/dataloaders/val/AmstertimeDataset.py -------------------------------------------------------------------------------- /dataloaders/val/BaiduDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/dataloaders/val/BaiduDataset.py -------------------------------------------------------------------------------- /dataloaders/val/MapillaryDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/dataloaders/val/MapillaryDataset.py -------------------------------------------------------------------------------- /dataloaders/val/MapillaryTestDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/dataloaders/val/MapillaryTestDataset.py -------------------------------------------------------------------------------- /dataloaders/val/NordlandDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/dataloaders/val/NordlandDataset.py -------------------------------------------------------------------------------- /dataloaders/val/PittsburghDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/dataloaders/val/PittsburghDataset.py -------------------------------------------------------------------------------- /dataloaders/val/SPEDDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/dataloaders/val/SPEDDataset.py -------------------------------------------------------------------------------- /dataloaders/val/SfsmDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/dataloaders/val/SfsmDataset.py -------------------------------------------------------------------------------- /dataloaders/val/StluciaDataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/dataloaders/val/StluciaDataset.py -------------------------------------------------------------------------------- /dataloaders/val/Tokyo247Dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/dataloaders/val/Tokyo247Dataset.py -------------------------------------------------------------------------------- /datasets/Nordland/Nordland_dbImages.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/Nordland/Nordland_dbImages.npy -------------------------------------------------------------------------------- /datasets/Nordland/Nordland_gt.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/Nordland/Nordland_gt.npy -------------------------------------------------------------------------------- /datasets/Nordland/Nordland_qImages.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/Nordland/Nordland_qImages.npy -------------------------------------------------------------------------------- /datasets/Pittsburgh/pitts250k_test_dbImages.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/Pittsburgh/pitts250k_test_dbImages.npy -------------------------------------------------------------------------------- /datasets/Pittsburgh/pitts250k_test_gt.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/Pittsburgh/pitts250k_test_gt.npy -------------------------------------------------------------------------------- /datasets/Pittsburgh/pitts250k_test_qImages.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/Pittsburgh/pitts250k_test_qImages.npy -------------------------------------------------------------------------------- /datasets/Pittsburgh/pitts30k_test_dbImages.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/Pittsburgh/pitts30k_test_dbImages.npy -------------------------------------------------------------------------------- /datasets/Pittsburgh/pitts30k_test_gt.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/Pittsburgh/pitts30k_test_gt.npy -------------------------------------------------------------------------------- /datasets/Pittsburgh/pitts30k_test_qImages.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/Pittsburgh/pitts30k_test_qImages.npy -------------------------------------------------------------------------------- /datasets/Pittsburgh/pitts30k_val_dbImages.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/Pittsburgh/pitts30k_val_dbImages.npy -------------------------------------------------------------------------------- /datasets/Pittsburgh/pitts30k_val_gt.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/Pittsburgh/pitts30k_val_gt.npy -------------------------------------------------------------------------------- /datasets/Pittsburgh/pitts30k_val_qImages.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/Pittsburgh/pitts30k_val_qImages.npy -------------------------------------------------------------------------------- /datasets/SPED/SPED_dbImages.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/SPED/SPED_dbImages.npy -------------------------------------------------------------------------------- /datasets/SPED/SPED_gt.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/SPED/SPED_gt.npy -------------------------------------------------------------------------------- /datasets/SPED/SPED_qImages.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/SPED/SPED_qImages.npy -------------------------------------------------------------------------------- /datasets/amstertime/amstertime_dbImages.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/amstertime/amstertime_dbImages.npy -------------------------------------------------------------------------------- /datasets/amstertime/amstertime_qImages.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/amstertime/amstertime_qImages.npy -------------------------------------------------------------------------------- /datasets/baidu/baidu_dbImages.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/baidu/baidu_dbImages.npy -------------------------------------------------------------------------------- /datasets/baidu/baidu_dbgImages_gt.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/baidu/baidu_dbgImages_gt.npy -------------------------------------------------------------------------------- /datasets/baidu/baidu_qImages.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/baidu/baidu_qImages.npy -------------------------------------------------------------------------------- /datasets/baidu/baidu_qImages_gt.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/baidu/baidu_qImages_gt.npy -------------------------------------------------------------------------------- /datasets/msls_test/msls_test_dbImages.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/msls_test/msls_test_dbImages.npy -------------------------------------------------------------------------------- /datasets/msls_test/msls_test_qImages.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/msls_test/msls_test_qImages.npy -------------------------------------------------------------------------------- /datasets/msls_val/msls_val_dbImages.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/msls_val/msls_val_dbImages.npy -------------------------------------------------------------------------------- /datasets/msls_val/msls_val_pIdx.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/msls_val/msls_val_pIdx.npy -------------------------------------------------------------------------------- /datasets/msls_val/msls_val_qIdx.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/msls_val/msls_val_qIdx.npy -------------------------------------------------------------------------------- /datasets/msls_val/msls_val_qImages.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/msls_val/msls_val_qImages.npy -------------------------------------------------------------------------------- /datasets/sfsm/sfsm_dbImages.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/sfsm/sfsm_dbImages.npy -------------------------------------------------------------------------------- /datasets/sfsm/sfsm_qImages.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/sfsm/sfsm_qImages.npy -------------------------------------------------------------------------------- /datasets/st_lucia/st_lucia_dbImages.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/st_lucia/st_lucia_dbImages.npy -------------------------------------------------------------------------------- /datasets/st_lucia/st_lucia_qImages.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/st_lucia/st_lucia_qImages.npy -------------------------------------------------------------------------------- /datasets/tokyo247/tokyo247_dbImages.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/tokyo247/tokyo247_dbImages.npy -------------------------------------------------------------------------------- /datasets/tokyo247/tokyo247_qImages.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/datasets/tokyo247/tokyo247_qImages.npy -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/environment.yml -------------------------------------------------------------------------------- /eval.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/eval.py -------------------------------------------------------------------------------- /eval.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/eval.sh -------------------------------------------------------------------------------- /hubconf.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/hubconf.py -------------------------------------------------------------------------------- /models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /models/aggregators/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/models/aggregators/__init__.py -------------------------------------------------------------------------------- /models/aggregators/aggregation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/models/aggregators/aggregation.py -------------------------------------------------------------------------------- /models/aggregators/convap.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/models/aggregators/convap.py -------------------------------------------------------------------------------- /models/aggregators/cosplace.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/models/aggregators/cosplace.py -------------------------------------------------------------------------------- /models/aggregators/gem.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/models/aggregators/gem.py -------------------------------------------------------------------------------- /models/aggregators/mixvpr.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/models/aggregators/mixvpr.py -------------------------------------------------------------------------------- /models/aggregators/salad.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/models/aggregators/salad.py -------------------------------------------------------------------------------- /models/backbones/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/models/backbones/__init__.py -------------------------------------------------------------------------------- /models/backbones/dinov2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/models/backbones/dinov2.py -------------------------------------------------------------------------------- /models/backbones/resnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/models/backbones/resnet.py -------------------------------------------------------------------------------- /models/helper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/models/helper.py -------------------------------------------------------------------------------- /predict_analysis.sh: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/predict_analysis.sh -------------------------------------------------------------------------------- /predictions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/predictions.py -------------------------------------------------------------------------------- /sweepParams.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/sweepParams.yml -------------------------------------------------------------------------------- /train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/train.py -------------------------------------------------------------------------------- /utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/utils/__init__.py -------------------------------------------------------------------------------- /utils/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/utils/losses.py -------------------------------------------------------------------------------- /utils/validation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/utils/validation.py -------------------------------------------------------------------------------- /vpr_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Ahmedest61/VLAD-BuFF/HEAD/vpr_model.py --------------------------------------------------------------------------------