xjh19972/STHN-two-stages

STHN — Two-Stages Model

This is the two-stages variant of STHN (Satellite-Thermal Homography Network) for UAV satellite-thermal geo-localization. It performs coarse homography estimation followed by a fine-stage refinement on a cropped region around the coarse prediction.

Pushed to the Hub using the PytorchModelHubMixin integration.

• Code: https://github.com/arplaboratory/STHN
• Paper: https://arxiv.org/abs/2405.20470
• Demo code: https://github.com/arplaboratory/STHN/blob/main/STHN_demo.py
• One-stage variant: https://huggingface.co/xjh19972/STHN-one-stage

Config

{
  "model_config": {
    "corr_level": 4,
    "database_size": 1536,
    "fine_padding": 0,
    "iters_lev0": 6,
    "iters_lev1": 6,
    "resize_width": 256,
    "two_stages": true
  }
}