T-Rex Dataset
收藏资源简介:
T-Rex数据集是一个大规模、100小时的触觉反应数据集,通过一种数据高效的配方收集,优先考虑基本运动原语(22个原语、200多个对象、5400多个轨迹);其中约50小时以LeRobot v3.0格式开源。数据集用于训练触觉反应策略,支持12个需要精细力控制和可变形物体操作的任务。
The T-Rex Dataset is a large-scale, 100-hour tactile response dataset collected via a data-efficient recipe, prioritizing basic movement primitives (22 primitives, over 200 objects, and more than 5400 trajectories). Approximately 50 hours of this dataset are open-sourced in the LeRobot v3.0 format. This dataset is utilized for training tactile response policies and supports 12 tasks that require fine force control and deformable object manipulation.
数据集概述
T-Rex(Tactile-Reactive Dexterous Manipulation) 是一个面向触觉反应型灵巧操作的大规模数据集。
- 数据集规模:总时长约 100 小时,其中开源约 50 小时。
- 数据构成:包含 5,400 多条轨迹,覆盖 22 种基本运动基元,涉及 200 多个物体。
- 采集平台:使用双臂 Dexmate Vega-1 机器人,配备两个 Sharpa Wave 灵巧手。
- 传感器模态:
- 三路 RGB 视频:头部、左手腕、右手腕。
- 状态与动作:当前与目标关节位置。
- 触觉传感器:每指尖 10 张 原始灰度图像、估计变形图、估计的 6 维力/力矩。
- 数据格式:以 LeRobot v3.0 格式发布,托管于 Hugging Face Hub: https://huggingface.co/datasets/zekaiwang/trex_dataset
- 数据使用:提供独立的 快速启动套件,包含 Colab 笔记本,支持无需完整下载即可浏览、检查与重放数据。快速启动地址:https://github.com/ZhuoyangLiu2005/T-Rex/blob/main/dataset_quickstart/README.md
方法亮点
- 异步混合专家变换器(Asynchronous Mixture-of-Transformers, MoT):基于 Qwen3-VL-2B 骨干网络,包含潜在(推理)、动作、触觉三个专家,以不同速率运行(动作去噪约 5 Hz,触觉精炼约 20 Hz),通过级联流匹配耦合,使策略能在动作块内对接触做出响应。
- 时序触觉 VQ-VAE:对高频力/变形信号进行时序量化编码,嵌入模型中,实时编码。
- 三阶段训练:大规模无触觉预训练 → 触觉反应中段训练 → 任务特定后训练。
- 性能提升:在 12 项 精细力控与可变形物体操作任务上,平均成功率比最强基线 高 30% 以上。
模型仓库
提供在 Hugging Face Hub 上的检查点,可直接用于任务微调:
| 检查点 | 阶段 | 说明 |
|---|---|---|
| miniFranka/T-Rex_pretrain_mecka22k_epoch1 | 预训练 | 约 22k 无触觉 episode 上的 VLM-动作对齐(1 epoch)。 |
| miniFranka/T-Rex_midtrain_mecka23k_ucb100_vqvae_epoch6 | 中段训练 | 触觉反应型(级联流 + 嵌入 VQ-VAE),6 epochs。推荐从此开始微调。 |
硬件与遥操作栈
完整的遥操作与数据采集硬件代码位于 hardware_code/ 目录,包括:
- Manus 数据手套 + VIVE 追踪器 遥操作方案。
- 全臂逆运动学与碰撞避免 控制。
- 相机/触觉流式传输与同步录制(HDF5 + MP4 + 无损压缩触觉视频)。
- 机器人端推理客户端(支持慢/快协议服务器)。
数据准备(自定义任务)
用户需将自己的任务 episode 按指定目录结构组织,并转换为以下两种格式之一进行后训练:
- JSON 格式(默认):通过
utils/gen_json_tac_deltabase_eef_bimanual_parallel.py生成。 - LeRobot v3.0 格式(可选):通过
utils/convert_inlab_to_lerobot.py转换。
触觉码默认由嵌入的 VQ-VAE 实时编码,无需预烘焙。
知识引用
- 项目主页:https://tactile-rex.github.io/
- 论文(arXiv):https://arxiv.org/abs/2606.17055
- 数据集(Hugging Face):https://huggingface.co/datasets/zekaiwang/trex_dataset




