You can deploy this model and start earning money today!

xjh19972/
STHN-two-stages

Name: STHN-two-stages
Availability: InStock

safetensors

model_hub_mixin

pytorch_model_hub_mixin

xjh19972/boson-nighttime

2405.20470

Views

No views yet

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

json

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

xjh19972/STHN-two-stages

STHN — Two-Stages Model

Config

xjh19972/
STHN-two-stages