banana_red
收藏资源简介:
该数据集是一个机器人控制数据集,专为so_follower机器人设计。它包含1个任务、1个片段和580帧数据,帧率为30 fps。数据集存储为Parquet文件,总数据大小为100 MB,视频文件大小为200 MB。数据特征包括动作(action)和观察状态(observation.state),均为6维浮点数组,对应机器人的关节位置(如肩部平移、肩部升降、肘部弯曲、腕部弯曲、腕部旋转和夹爪位置)。观察图像(observation.images.front)为视频格式,分辨率为480x640,3通道,使用AV1编解码器。此外,还包含时间戳、帧索引、片段索引、索引和任务索引等元数据。数据集适用于机器人学习任务,如模仿学习或强化学习。
This dataset is a robotic control dataset specifically designed for the so_follower robot. It contains 1 task, 1 segment, and 580 frames of data with a frame rate of 30 fps. The dataset is stored as Parquet files, with a total size of 100 MB, while the accompanying video files have a size of 200 MB. The data features include action and observation.state, both of which are 6-dimensional floating-point arrays corresponding to the robot's joint positions (e.g., shoulder translation, shoulder elevation, elbow flexion, wrist flexion, wrist rotation, and gripper position). The observation image (observation.images.front) is in video format, with a resolution of 480x640, 3 channels, and uses the AV1 codec. Additionally, it includes metadata such as timestamps, frame indices, segment indices, indexes, and task indices. This dataset is suitable for robotic learning tasks such as imitation learning or reinforcement learning.
数据集概述
数据集名称:banana_red
主页地址:https://huggingface.co/datasets/rslxcvg/banana_red
许可证:Apache-2.0
任务类别:机器人学(Robotics)
数据集规模与构成
- 总片段数(Episodes):1
- 总帧数(Frames):580
- 总任务数(Tasks):1
- 帧率(FPS):30
- 数据文件大小:约100 MB
- 视频文件大小:约200 MB
- 数据集划分:仅包含训练集(train: "0:1")
数据格式
- 数据文件格式:Parquet(路径:
data/*/*.parquet) - 视频文件格式:MP4(视频编码为AV1,分辨率480×640,帧率30 FPS)
- 数据分块:chunks_size = 1000
特征结构
| 特征名 | 数据类型 | 形状 | 说明 |
|---|---|---|---|
| action | float32 | (6,) | 机器人6自由度动作(肩部、肘部、腕部、夹爪位置) |
| observation.state | float32 | (6,) | 机器人6自由度观测状态(同动作维度) |
| observation.images.front | video | (480, 640, 3) | 前置摄像头视频帧(RGB,高度480,宽度640) |
| timestamp | float32 | (1,) | 时间戳 |
| frame_index | int64 | (1,) | 帧索引 |
| episode_index | int64 | (1,) | 片段索引 |
| index | int64 | (1,) | 全局索引 |
| task_index | int64 | (1,) | 任务索引 |
机器人信息
- 机器人类型:so_follower



