search

arplaboratory / satellite-thermal-geo-localization

[IROS 2023] Official repository for "Long-range UAV Thermal Geo-localization with Satellite Imagery"
https://xjh19971.github.io/STGL/
MIT License
55 stars 8 forks source link

Ask for trained models #21

Closed mrwangyou closed 5 months ago

mrwangyou commented 6 months ago

Hi Jiuhong,

I've been working on UAV geo-localization and find the approach fascinating and potentially very useful. Thank you for your great effort in thermal geo-localization. Since the training step costs a lot in time and resources, I wonder if would it be possible for you to share the pretrained model weights?

Thank you.

xjh19971 commented 5 months ago

Thank you for your interest, and apologies for the delay. We have provided pretrained weights for ResNet50+GeM in our latest repository (https://github.com/arplaboratory/STHN/tree/main) as a baseline.

The download link is: https://drive.google.com/drive/folders/19Z0wqeDlJyzTZB1qc355G2WEww7I9rTB

However, due to updates in the code, these pretrained weights may not be compatible with this repository. We recommend using them with https://github.com/arplaboratory/STHN/tree/main/global_pipeline, which offers the same functionality as this repository.

mrwangyou commented 5 months ago

Thank you for providing the pretrain weights. They will help a lot with my work.