Robotics
PyTorch
tactile-sensing
multimodal

Add baseline checkpoints

#1
by StanFu - opened
README.md ADDED
@@ -0,0 +1,29 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ library_name: pytorch
4
+ tags:
5
+ - robotics
6
+ - tactile-sensing
7
+ - multimodal
8
+ datasets:
9
+ - TUM-ICS/Hide-and-Seek
10
+ ---
11
+
12
+ # HAS-Bench Baselines
13
+
14
+ Pretrained baseline checkpoints for the [Hide-and-Seek tactile dataset](https://huggingface.co/datasets/TUM-ICS/Hide-and-Seek): a dual-head multimodal classifier with an object head and a weight head, trained with the public release pipeline (seed 42).
15
+
16
+ | File | Modalities |
17
+ |---|---|
18
+ | `m123/best_model.pt` | tactile point cloud + raw skin signals + virtual skin wrenches |
19
+ | `m12345/best_model.pt` | + end-effector pose + wrist F/T |
20
+
21
+ Each directory also contains the `last_metrics.json` written at the end of training.
22
+
23
+ ## Usage
24
+
25
+ ```bash
26
+ hf download TUM-ICS/HAS-Bench-baselines --local-dir checkpoints
27
+ ```
28
+
29
+ Dataloaders and evaluation scripts: https://github.com/TUM-ICS/Tactile-Hide-and-Seek
m123/best_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:0281881557edca98d31f836bd0f8764573c97de49cc6c6fb5112ca86455c5a55
3
+ size 6968475
m123/last_metrics.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "validation": {
3
+ "loss": 0.41280303381904654,
4
+ "object_acc": 0.9346707818930041,
5
+ "weight_acc": 0.875,
6
+ "joint_acc": 0.8461934156378601
7
+ }
8
+ }
m12345/best_model.pt ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:a89dd9182fc8b522ff6b7b466d220dea819c8dda7ae4d813137db17262c695f5
3
+ size 10306331
m12345/last_metrics.json ADDED
@@ -0,0 +1,8 @@
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "validation": {
3
+ "loss": 0.4156359742345756,
4
+ "object_acc": 0.9382716049382716,
5
+ "weight_acc": 0.885545267489712,
6
+ "joint_acc": 0.856224279835391
7
+ }
8
+ }