text_audio_pretraining_for_GAN
收藏资源简介:
bw_jh_dataset 是一个多源机器人操作数据集,提供机器人夹爪和机械臂的3D点跟踪数据,以及校准的多视角RGB视频、语言指令和成功标签。数据集包含三个子集:droid(27,615个episodes,约280 GB,2个外部ZED视角,来自真实Franka机器人,仅包含成功且有语言指令的episodes)、hrdexdb(1,601个episodes,约998 GB,22个校准相机,来自真实xArm6+灵巧手)、robocasa(40,026个episodes,约273 GB,2个agent视角加腕部ego相机,来自仿真Panda机器人)。每个episode存储为一个目录,包含多视角MP4视频、gripper.npz(256个3D点,T×256×3)、robot.npz(512个3D点,T×512×3)、object.npz(256个3D点,T×256×3,robocasa中每个操作对象单独一个文件)、meta.json(包含相机内外参、语言指令、成功标签、帧率、帧数、任务类别等)。所有3D点位于view1相机坐标系(OpenCV约定,米制)。对于droid和hrdexdb,相机为静态支架;对于robocasa,view1/view2相机安装在移动基座上,当基座移动时,meta.json中提供每帧的相机外参(T_cam_world_perframe)。数据集适用于机器人操作任务,如3D点跟踪、多视角重建、语言引导的操纵等。许可证为CC-BY-4.0。
bw_jh_dataset is a multi-source robot manipulation dataset providing 3D point tracking data for robot grippers and arms, along with calibrated multi-view RGB videos, language instructions, and success labels. The dataset consists of three subsets: droid (27,615 episodes, ~280 GB, 2 external ZED views, from real Franka robot, only includes successful episodes with language instructions), hrdexdb (1,601 episodes, ~998 GB, 22 calibrated cameras, from real xArm6+dexterous hand), and robocasa (40,026 episodes, ~273 GB, 2 agent views plus wrist ego camera, from simulated Panda robot). Each episode is stored as a directory containing multi-view MP4 videos, gripper.npz (256 3D points, T×256×3), robot.npz (512 3D points, T×512×3), object.npz (256 3D points, T×256×3, with separate files for each manipulated object in robocasa), and meta.json (including camera intrinsics/extrinsics, language instructions, success labels, frame rate, frame count, task category, etc.). All 3D points are in the view1 camera coordinate system (OpenCV convention, meters). For droid and hrdexdb, cameras are static; for robocasa, view1/view2 cameras are mounted on a mobile base, and meta.json provides per-frame camera extrinsics (T_cam_world_perframe) when the base moves. The dataset is suitable for robot manipulation tasks such as 3D point tracking, multi-view reconstruction, language-guided manipulation, etc. License: CC-BY-4.0.
bw_jh_dataset — 机器人3D点轨迹数据集
数据集概述
一个统一的多源机器人操作数据集,包含机器人夹爪和机械臂的3D点轨迹、校准后的多视角RGB视频、语言指令和成功标签。数据集分为三个子集,总规模约1,551 GB。
数据子集
| 子集 | 轨迹数 | 大小 | 视角 | 来源 |
|---|---|---|---|---|
droid |
27,615(另有~1,050条在droid-11中) | ~280 GB | 2个外部ZED视角 | DROID(真实,Franka) |
hrdexdb |
1,601 | ~998 GB | 22个标定相机 | HRDexDB(真实,xArm6+灵巧手) |
robocasa |
40,026 | ~273 GB | 2个agent视角+腕部相机 | RoboCasa(仿真,Panda) |
数据获取
- 每个子集以tar分片形式提供,大于50 GB的文件被拆分为
*.tar.partNN片段,需先拼接再解压。 - 每个分片包含对应的
*.txt/*.manifest.txt文件,列出其包含的轨迹ID。
样本格式(各子集一致)
每个轨迹为一个目录,内容如下:
| 文件 | 说明 |
|---|---|
view*.mp4 |
RGB视频(hrdexdb最多view23;robocasa含ego.mp4) |
gripper.npz |
3D夹爪点轨迹,形状(T, 256, 3),float32 |
robot.npz |
3D机械臂点轨迹,形状(T, 512, 3),float32 |
object*.npz |
被操作物体的3D轨迹,形状(T, 256, 3),float32(robocasa多物体任务有多个文件) |
meta.json |
标定参数+标签信息 |
关键细节:
- 所有3D点均位于view1相机坐标系(OpenCV约定,单位:米)
meta.json包含视角标定(T_cam_world4×4矩阵、内参)、语言指令、成功标签、fps、帧数、任务类别、uuid等字段- droid/hrdexdb相机为静态固定,单次标定即可;robocasa中相机随移动底座运动,移动轨迹使用
T_cam_world_perframe逐帧标定
RoboCasa被操作物体轨迹
- 每个robocasa轨迹包含每个被操作物体的
object*.npz文件 - 物体选择采用确定性方法:仅当物体在轨迹中移动超过5 cm时才被包含
- 放置目标(如锅、餐盒)和干扰物保持静止,不包含在内
- 多物体任务生成
object.npz、object2.npz等,meta.json → object列出所有被操作物体
DROID子集说明
- 仅包含带语言指令的成功轨迹(27,615条 = 成功 ∩ 带语言注释)
meta.json已修补,包含验证过的指令,无需后处理droid/droid_language_recovered.json记录了语言恢复的来源追踪信息
RoboCasa移动底座过滤
robocasa/base_motion_filter/按底座移动情况分类每个轨迹:
| 类别 | 轨迹数(占比) | 含义 |
|---|---|---|
static |
31,193(54.7%) | 底座从不移动,可直接使用 |
settle |
24,167(42.4%) | 底座移动后静止≥30帧,裁剪至静止尾部即可 |
moving |
1,679(2.9%) | 结束时底座仍在移动,丢弃 |
97.1%的轨迹保持可用。tools/robocasa_base_filter.py可自动应用此过滤。
可视化工具
tools/visualize_episode.py可将3D点轨迹重新投影到所有视角,生成拼接多视角视频,用于验证标定和npz一致性。示例用法:
bash
python tools/visualize_episode.py --split droid --episode 0 --root <extracted_root>
python tools/visualize_episode.py --split hrdexdb --episode allegro_v5__apple__3 --root <extracted_root>
来源与许可
droid:源自DROID数据集(CC-BY-4.0),语言注释合并自官方聚合注释和KarlP/droid 75k发布robocasa:源自RoboCasa数据集hrdexdb:HRDexDB多相机遥操作录制- 成功标签来自源数据集自身标签;8条标签矛盾的droid轨迹已被排除




