eval_xvla-so101-4tasks-aug_sort_25
收藏资源简介:
该数据集是一个机器人控制数据集,使用LeRobot工具创建。它包含来自so_follower类型机器人的演示数据,适用于机器人模仿学习或强化学习任务。数据集规模为19个完整episodes,共计57,809帧数据,对应1个任务。数据以分块Parquet文件存储,并包含关联的MP4格式视频文件。数据特征包括:机器人的6维动作指令(对应肩部平移、肩部升降、肘部弯曲、腕部弯曲、腕部旋转和夹爪位置);观测状态(与动作相同的6维关节位置);两个图像观测(image和image2),均为RGB视频,分辨率480x640,帧率30fps;以及时间戳、帧索引、episode索引、全局索引和任务索引等元数据。视频采用AV1编码。数据集遵循Apache-2.0许可证。
This dataset is a robot control dataset created using the LeRobot tool. It contains demonstration data from so_follower type robots, suitable for robot imitation learning or reinforcement learning tasks. The dataset consists of 19 complete episodes, totaling 57,809 frames of data, corresponding to 1 task. Data is stored in chunked Parquet files and includes associated video files in MP4 format. Features include: 6-dimensional action commands for the robot (corresponding to shoulder translation, shoulder elevation, elbow flexion, wrist flexion, wrist rotation, and gripper position); observation states (6-dimensional joint positions identical to the actions); two image observations (image and image2), both RGB videos with a resolution of 480x640 and a frame rate of 30 fps; and metadata such as timestamps, frame index, episode index, global index, and task index. Videos are encoded with AV1. The dataset is licensed under Apache-2.0.
数据集概览
- 数据集名称: eval_xvla-so101-4tasks-aug_sort_25
- 许可证: Apache-2.0
- 任务类别: 机器人技术 (Robotics)
- 标签: LeRobot
数据集结构
- 机器人类型: so_follower
- 总片段数: 20
- 总帧数: 61173
- 总任务数: 1
- 块大小: 1000
- 数据文件大小: 100 MB
- 视频文件大小: 200 MB
- 帧率: 30 FPS
- 数据划分:
- 训练集: 片段 0 至 19 (共20个片段)
数据特征
| 特征 | 数据类型 | 形状 | 说明 |
|---|---|---|---|
| action | float32 | [6] | 包含6个关节动作:shoulder_pan.pos, shoulder_lift.pos, elbow_flex.pos, wrist_flex.pos, wrist_roll.pos, gripper.pos |
| observation.state | float32 | [6] | 包含6个关节状态,字段同 action |
| observation.images.image | video | [480, 640, 3] | 主摄像头视频,AV1编码,30 FPS,高度480像素,宽度640像素,3通道 |
| observation.images.image2 | video | [480, 640, 3] | 另一视角摄像头视频,参数与 image 一致 |
| timestamp | float32 | [1] | 时间戳 |
| frame_index | int64 | [1] | 帧索引 |
| episode_index | int64 | [1] | 片段索引 |
| index | int64 | [1] | 全局索引 |
| task_index | int64 | [1] | 任务索引 |
数据文件路径
- 数据文件:
data/chunk-{chunk_index:03d}/file-{file_index:03d}.parquet - 视频文件:
videos/{video_key}/chunk-{chunk_index:03d}/file-{file_index:03d}.mp4
在线可视化
- 可使用 LeRobot 可视化工具:https://huggingface.co/spaces/lerobot/visualize_dataset?path=hjkso1406/eval_xvla-so101-4tasks-aug_sort_25




