Skip to content

MFRGN: Multi-scale Feature Representation Generalization Network for Ground-to-Aerial Geo-localization

Notifications You must be signed in to change notification settings

ytao-wang/MFRGN

Repository files navigation

This repo contains the official implementation of the ACM MM 2024 paper

MFRGN: Multi-scale Feature Representation Generalization Network for Ground-to-Aerial Geo-localization

Yuntao Wang, Jinpu Zhang, Ruonan Wei, Wenbo Gao, Yuehuan Wang*

paper(MM'24)

This code is based on the Sample4Geo framework.

Details of the datasets, training and inference can be found in Sample4Geo.

News

  • 17 Sep 2025 We provide our all model checkpoints on google drive: CVUSA,CVACT,VIGOR (3.06G, including distance_dict, convnext_base, the model with crop, sample, and crop&sample), University-1652 (0.6G, including D2S and S2D).

  • 12 Mar 2025 We now support training and testing University-1652 by adding the Class TimmModel_u to mfrgn.py. It is worth noting that the full shared network is used here, which is different from the TimmModel (which only shares the backbone for ground and aerial view). And we provide our pretrained results: checkpoint_u1652 [BaiduYun, Password: 1234].

  • 12 Dec 2024 We now provide supplementary results on University-1652.

Methods Drone2Sat
R@1 / AP
Sat2Drone
R@1 / AP
Sample4Geo 92.65 / 93.81 95.65 / 91.39
Ours 94.33 / 95.24 96.15 / 93.94

Dataset Preparation

To accelerate training/test time, you can run data_preparation.py, which implements image transformation (from '.jpg'/'.png' to '.pt') and cropping (similar to SAFA).

When you process images from '.jpg'/'.png' to '.pt', you should set ext='pt' in sample4geo/dataset/*.py

Also, if you are experiencing network errors about the backbone, you may need to download the backbone weights offline and put them into the pretrained folder.

Results

We provide our pretrained results: MFRGN-pretained.zip [BaiduYun, Password: 1234], which contains pretrained weight files or files necessary to train certain network configurations (e.g. distance_dict, convnext backbone weights).

About

MFRGN: Multi-scale Feature Representation Generalization Network for Ground-to-Aerial Geo-localization

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages