Add model card
Browse files
README.md
ADDED
|
@@ -0,0 +1,64 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
---
|
| 2 |
+
license: cc-by-nc-4.0
|
| 3 |
+
library_name: pytorch
|
| 4 |
+
pipeline_tag: feature-extraction
|
| 5 |
+
tags:
|
| 6 |
+
- tactile
|
| 7 |
+
- robotics
|
| 8 |
+
- gelsight
|
| 9 |
+
- masked-autoencoder
|
| 10 |
+
- multimodal
|
| 11 |
+
- representation-learning
|
| 12 |
+
---
|
| 13 |
+
|
| 14 |
+
# HTT — Heterogeneous Tactile Transformer
|
| 15 |
+
|
| 16 |
+
A multimodal tactile representation model. One shared transformer backbone
|
| 17 |
+
encodes four different tactile sensors into a common **192-dimensional**
|
| 18 |
+
embedding space, pretrained with masked-autoencoder reconstruction and
|
| 19 |
+
cross-modal alignment. Feed a raw sensor reading, get a feature vector for any
|
| 20 |
+
downstream head (classification, force / slip estimation, policy learning).
|
| 21 |
+
|
| 22 |
+
| Modality | Type | Raw input |
|
| 23 |
+
|---|---|---|
|
| 24 |
+
| `gsmini` | vision (GelSight Mini) | uint8 image `[224, 224, 3]` |
|
| 25 |
+
| `9dtact` | vision (9DTact) | uint8 image `[224, 224, 3]` |
|
| 26 |
+
| `xela` | taxel array | float `[T, 72]` |
|
| 27 |
+
| `tac02` | taxel array | float `[T, 66]` |
|
| 28 |
+
|
| 29 |
+
## Checkpoint
|
| 30 |
+
|
| 31 |
+
| | |
|
| 32 |
+
|---|---|
|
| 33 |
+
| File | `htt_4sensors_best.pth` (~69 MB) |
|
| 34 |
+
| Contents | `model_state_dict` = 4 encoders + shared 9-layer trunk + 4 decoders (17.1 M params) |
|
| 35 |
+
| Embedding dim | 192 |
|
| 36 |
+
| SHA-256 | `024f4c3a067168197d0a6996bbca5c03e744ed5abd1d35a666dbf78e7ac673f0` |
|
| 37 |
+
|
| 38 |
+
Slim inference/finetune checkpoint (optimizer / predictor states dropped).
|
| 39 |
+
|
| 40 |
+
## Usage
|
| 41 |
+
|
| 42 |
+
Use it with the **HTT** package (contains the architecture, preprocessing, and
|
| 43 |
+
examples). Download the weights into `checkpoints/`:
|
| 44 |
+
|
| 45 |
+
```bash
|
| 46 |
+
hf download AllenBi21/HTT htt_4sensors_best.pth --local-dir checkpoints
|
| 47 |
+
```
|
| 48 |
+
|
| 49 |
+
```python
|
| 50 |
+
import numpy as np
|
| 51 |
+
from htt import HTT
|
| 52 |
+
|
| 53 |
+
model = HTT(modality="gsmini") # loads checkpoints/htt_4sensors_best.pth
|
| 54 |
+
frame = np.random.randint(0, 256, (224, 224, 3), dtype=np.uint8) # your sensor frame
|
| 55 |
+
emb = model(frame) # -> torch.Tensor [1, 192]
|
| 56 |
+
```
|
| 57 |
+
|
| 58 |
+
Read the raw-input contract before feeding your own data — the model returns
|
| 59 |
+
garbage on out-of-distribution inputs.
|
| 60 |
+
|
| 61 |
+
## License
|
| 62 |
+
|
| 63 |
+
Released under **CC BY-NC 4.0** (non-commercial). Portions are derived from
|
| 64 |
+
Meta's V-JEPA / DINOv2 (Apache-2.0 and CC-BY-NC-4.0).
|