five

avyuktsachdeva/pick_place_side_wrist_no_crop

收藏
Hugging Face2026-05-28 更新2026-05-31 收录
下载链接:
https://hf-mirror.com/datasets/avyuktsachdeva/pick_place_side_wrist_no_crop
下载链接
链接失效反馈
官方服务:
资源简介:
该数据集由LeRobot创建,是一个机器人学数据集,专门用于机器人控制和视觉任务。数据集包含100个episodes,总计46782个frames,数据以.parquet文件格式存储,视频以.mp4格式存储,帧率为15fps。特征包括两个摄像头图像(cam0和cam1,均为3通道、240x320分辨率)、状态观测(3维浮点数)、动作(4维浮点数),以及时间戳、帧索引、episode索引等元数据。机器人类型为Unity,数据分割为训练集(索引0到100)。数据集大小为100MB(数据文件)和200MB(视频文件),适用于机器人学习算法的训练和评估。

This dataset was created using LeRobot and is a robotics dataset designed for robot control and vision tasks. It contains 100 episodes with a total of 46,782 frames, stored in .parquet files for data and .mp4 files for videos at 15 fps. Features include two camera images (cam0 and cam1, both with 3 channels and 240x320 resolution), state observations (3-dimensional float32), actions (4-dimensional float32), and metadata such as timestamp, frame index, and episode index. The robot type is Unity, and the data is split into a training set (indices 0 to 100). The dataset size is 100 MB for data files and 200 MB for video files, suitable for training and evaluating robot learning algorithms.
提供机构:
avyuktsachdeva
搜集汇总
数据集介绍
main_image_url
构建方式
pick_place_side_wrist_no_crop数据集依托LeRobot框架构建,旨在为机器人操作任务提供高质量的仿真数据。该数据集包含100个完整episode,共计46782帧时序数据,所有数据均源自Unity仿真环境。数据以分块形式存储于parquet文件及对应的mp4视频文件中,其中parquet文件承载结构化特征数据,视频文件则记录高帧率图像序列。特征空间涵盖两个视角的视觉观测图像(cam0与cam1,分辨率240×320)、3维机器人状态向量以及4维动作指令,并辅以时间戳、帧索引等元信息,形成一套结构严谨的机器人操作行为数据集。
特点
该数据集的核心特点在于其针对机器人抓取与放置任务(pick-place)的精细设计,专注于侧面腕部视角且未进行图像裁剪,保留了原始视野的完整性。数据采集频率为15 FPS,确保动作序列的时序分辨率。特征维度中,状态空间为3维,动作空间为4维,反映了任务在位置与姿态控制上的复杂度。所有数据已统一划分为训练集(100个episode),无需额外拆分,便于直接用于模仿学习或强化学习算法的训练与评估。此外,数据集采用Apache-2.0许可协议,具有开放的学术和商业使用权限。
使用方法
使用本数据集时,可基于LeRobot库加载parquet文件与视频数据,通过定义的数据路径(data_path与video_path)索引对应的chunk与file。用户需按照features字典中定义的键名(如observation.image.cam0)访问多模态输入,并利用state与action字段构建策略模型的输入-输出对。数据以episode为单位组织,可结合episode_index与frame_index进行时序对齐,适用于离线模仿学习(如行为克隆)或在线交互式训练场景。建议配合Unity仿真环境回放验证,以提升模型在真实或虚拟环境中的迁移性能。
背景与挑战
背景概述
在机器人学习领域,模仿学习与强化学习依赖于高质量、多样化的演示数据集来训练机器人完成精细操作任务。pick_place_side_wrist_no_crop数据集由Hugging Face社区基于LeRobot框架创建,专注于机器人抓取与放置操作,采用Unity仿真环境生成,包含100个演示片段、共计46782帧图像和对应动作状态数据,旨在为机器人操作策略学习提供标准化训练资源。该数据集通过双视角摄像头(cam0与cam1)记录240×320分辨率的视觉信息,并结合三维机器人状态与四维动作空间,形成了包含视觉-状态-动作多模态对齐的完整样本。作为LeRobot生态中标准化数据格式的代表,该数据集为机器人操作策略的迁移学习、多任务泛化及仿真到现实的算法研究提供了可复现的基准,推动了机器人数据集在格式统一性、模态完整性方面的发展。
当前挑战
该数据集所应对的核心领域挑战在于:机器人抓取与放置操作中,如何从有限演示中高效学习具备鲁棒性的视觉-运动策略。具体而言,数据集构建面临以下挑战:其一,仿真环境与真实物理世界之间存在显著域差异,仿真中采集的数据可能因光照、摩擦、抓取接触等参数简化而难以直接迁移至实际机器人平台;其二,数据集仅包含单一任务(pick_place_side_wrist_no_crop),动作空间为4维(可能包含位置与夹爪控制),状态空间为3维,这种低维表示虽简化了学习复杂度,却限制了策略对复杂环境变形的适应能力;其三,从构建过程看,数据以15帧/秒的低频率采样,且训练集仅含100个片段,较少的样本量使得数据多样性不足,难以覆盖抓取角度、物体形状及放置位置的广泛变化,易导致过拟合与泛化性能下降。
常用场景
经典使用场景
在机器人学习领域,pick_place_side_wrist_no_crop数据集为模仿学习与行为克隆提供了高质量的示范轨迹。该数据集源自Unity仿真环境,包含100个演示片段,总计超过四万六千帧,记录了机械臂从侧面视角结合腕部摄像头完成抓取与放置任务的完整过程。每个时间步均提供双摄像头图像、三自由度状态向量及四维动作指令,形成典型的视觉-状态-动作三元组结构,尤其适合用于训练端到端的机器人操作策略,是研究基于视觉的灵巧操作行为的经典基准资源。
衍生相关工作
围绕该数据集衍生出一系列具有影响力的研究工作,典型代表包括基于行为克隆的端到端操作策略优化、隐式策略与扩散策略在机器人任务中的首次适配,以及视觉-语言模型与动作生成的交叉探索。部分工作将其作为仿真基准,验证了基于Transformer的序列建模框架在长期依赖动作预测中的有效性。此外,该数据集还催生了针对多视角融合与数据增强技术的专项研究,推动了LeRobot开源生态中仿真数据标准化流程的建立,成为机器人社区比较不同模仿学习架构性能的重要参照。
数据集最近研究
最新研究方向
该数据集聚焦于机器人操作领域的精细抓取与放置任务,特别强调腕部侧向运动与无裁剪视觉输入的耦合。当前前沿研究方向集中于此数据集在预训练视觉-运动策略中的应用,例如结合扩散策略或基于Transformer的模仿学习模型,通过其提供的同步双视角图像与低维状态-动作流,推动具身智能体在非结构化环境中实现鲁棒性操作。热点事件上,该数据集的发布恰逢LeRobot生态在开源机器人数据标准化上的努力,其100个高质量轨迹、统一的Parquet数据格式与Apache-2.0许可,为跨平台基准测试和策略泛化研究提供了关键资源。这一设计降低了机器人数据采集与复用的门槛,催化了从单一场景模拟到零样本迁移的范式转换,对简化机器人技能学习的研究链路具有里程碑意义。
以上内容由遇见数据集搜集并总结生成
二维码
社区交流群
二维码
科研交流群
商业服务