xjh19972/STHN-one-stage

STHN — One-Stage Model

This is the one-stage variant of STHN (Satellite-Thermal Homography Network) for UAV satellite-thermal geo-localization. It performs iterative homography estimation in a single refinement stage (no fine-stage crop refinement).

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
• Two-stage variant: https://huggingface.co/xjh19972/STHN-two-stages

Config

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