Mirror weights from dcher95/TTE; update model card
Browse files- README.md +39 -0
- config.json +11 -0
- model.safetensors +3 -0
README.md
ADDED
|
@@ -0,0 +1,39 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
library_name: tte
|
| 3 |
+
pipeline_tag: feature-extraction
|
| 4 |
+
tags: [location-encoder, geospatial, remote-sensing, sentinel-2, voronoi]
|
| 5 |
+
license: mit
|
| 6 |
+
---
|
| 7 |
+
|
| 8 |
+
# Tessellating the Earth (TTE) — location encoder
|
| 9 |
+
|
| 10 |
+
Maps a geographic coordinate `(lat, lon)` to a learned embedding via a learnable
|
| 11 |
+
Spherical Voronoi partition of S² with global semantic tokens. ECCV 2026.
|
| 12 |
+
|
| 13 |
+
Daniel Cher, Hamza Iqbal, Eric Xing, Brian Wei, Nathan Jacobs — Washington University in St. Louis ([MVRL](https://mvrl.cse.wustl.edu/)).
|
| 14 |
+
|
| 15 |
+
- Code: https://github.com/mvrl/TTE
|
| 16 |
+
- Project page: https://dcher95.github.io/TTE/
|
| 17 |
+
|
| 18 |
+
```python
|
| 19 |
+
import torch
|
| 20 |
+
from tte import TTE
|
| 21 |
+
|
| 22 |
+
model = TTE.from_pretrained("MVRL/TTE").eval()
|
| 23 |
+
coords = torch.tensor([[37.77, -122.42],
|
| 24 |
+
[-3.12, 60.02]])
|
| 25 |
+
emb = model.encode(coords)
|
| 26 |
+
```
|
| 27 |
+
|
| 28 |
+
Image backbone used during training (not needed here): frozen SSL4EO-S12 MAE ViT-L/16.
|
| 29 |
+
|
| 30 |
+
## Citation
|
| 31 |
+
|
| 32 |
+
```bibtex
|
| 33 |
+
@inproceedings{cher2026tte,
|
| 34 |
+
title = {Tessellating the Earth: Learnable Spherical Voronoi Partitions for Location Encoding},
|
| 35 |
+
author = {Cher, Daniel and Iqbal, Hamza and Xing, Eric and Wei, Brian and Jacobs, Nathan},
|
| 36 |
+
booktitle = {European Conference on Computer Vision (ECCV)},
|
| 37 |
+
year = {2026}
|
| 38 |
+
}
|
| 39 |
+
```
|
config.json
ADDED
|
@@ -0,0 +1,11 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
{
|
| 2 |
+
"hidden_dim": 512,
|
| 3 |
+
"n_registers": 64,
|
| 4 |
+
"n_reslayers": 2,
|
| 5 |
+
"n_sites": 4096,
|
| 6 |
+
"normalize": true,
|
| 7 |
+
"output_dim": 512,
|
| 8 |
+
"register_fixed_gate": 0.5,
|
| 9 |
+
"site_embed_dim": 384,
|
| 10 |
+
"use_cosine_attention": false
|
| 11 |
+
}
|
model.safetensors
ADDED
|
@@ -0,0 +1,3 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
version https://git-lfs.github.com/spec/v1
|
| 2 |
+
oid sha256:a3e6fabf7f72d6943887d9cb5fd58ea45138267af31716489616f75e2fd47aae
|
| 3 |
+
size 13814728
|