AllenBi21 commited on
Commit
5314f01
·
verified ·
1 Parent(s): 589519a

Add model card

Browse files
Files changed (1) hide show
  1. README.md +64 -0
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).