NONHUMAN-RESEARCH/pick-and-place-one-fruit

---
license: apache-2.0
task_categories:
- robotics
tags:
- LeRobot
configs:
- config_name: default
data_files: data/*/*.parquet
---

This dataset was created using [LeRobot](https://github.com/huggingface/lerobot).

## Dataset tasks prompts

### Experiment 1

- **Banana:** "Identify the long yellow banana lying flat on the table. Approach and grasp the curved yellow fruit near its middle section. Lift the banana and deposit it carefully into the yellow basket."
- **Strawberry:** "Locate the small red strawberry positioned on the table surface. Securely grasp the round red fruit with the robotic gripper. Transport the strawberry to the yellow basket and release it inside."

### Experiment 2

- **Banana:** "Locate the long curved banana on the table, grasp it by the middle, and place it into the yellow basket."
- **Strawberry:** "Locate the small red strawberry on the table, grasp it securely, and place it into the yellow basket."

### Experiment 3

- **Banana:** "Pick the banana from the table and place it in the basket."
- **Strawberry:** "Pick the strawberry from the table and place it in the basket."

## Dataset Description

- **Homepage:** [More Information Needed]
- **Paper:** [More Information Needed]
- **License:** apache-2.0

## Dataset Structure

[meta/info.json](meta/info.json):
```json
{
"codebase_version": "v3.0",
"robot_type": "bi_piper_follower",
"total_episodes": 522,
"total_frames": 194659,
"total_tasks": 2,
"chunks_size": 1000,
"data_files_size_in_mb": 100,
"video_files_size_in_mb": 500,
"fps": 50,
"splits": {
"train": "0:522"
},
"data_path": "data/chunk-{chunk_index:03d}/file-{file_index:03d}.parquet",
"video_path": "videos/{video_key}/chunk-{chunk_index:03d}/file-{file_index:03d}.mp4",
"features": {
"action": {
"dtype": "float32",
"names": [
"left_shoulder_pan.pos",
"left_shoulder_lift.pos",
"left_elbow_flex.pos",
"left_forearm_roll.pos",
"left_wrist_flex.pos",
"left_wrist_roll.pos",
"left_gripper.pos",
"right_shoulder_pan.pos",
"right_shoulder_lift.pos",
"right_elbow_flex.pos",
"right_forearm_roll.pos",
"right_wrist_flex.pos",
"right_wrist_roll.pos",
"right_gripper.pos"
],
"shape": [
14
]
},
"observation.state": {
"dtype": "float32",
"names": [
"left_shoulder_pan.pos",
"left_shoulder_lift.pos",
"left_elbow_flex.pos",
"left_forearm_roll.pos",
"left_wrist_flex.pos",
"left_wrist_roll.pos",
"left_gripper.pos",
"right_shoulder_pan.pos",
"right_shoulder_lift.pos",
"right_elbow_flex.pos",
"right_forearm_roll.pos",
"right_wrist_flex.pos",
"right_wrist_roll.pos",
"right_gripper.pos"
],
"shape": [
14
]
},
"observation.images.left": {
"dtype": "video",
"shape": [
376,
672,
3
],
"names": [
"height",
"width",
"channels"
],
"info": {
"video.height": 376,
"video.width": 672,
"video.codec": "av1",
"video.pix_fmt": "yuv420p",
"video.is_depth_map": false,
"video.fps": 50,
"video.channels": 3,
"has_audio": false
}
},
"observation.images.top": {
"dtype": "video",
"shape": [
376,
672,
3
],
"names": [
"height",
"width",
"channels"
],
"info": {
"video.height": 376,
"video.width": 672,
"video.codec": "av1",
"video.pix_fmt": "yuv420p",
"video.is_depth_map": false,
"video.fps": 50,
"video.channels": 3,
"has_audio": false
}
},
"observation.images.right": {
"dtype": "video",
"shape": [
376,
672,
3
],
"names": [
"height",
"width",
"channels"
],
"info": {
"video.height": 376,
"video.width": 672,
"video.codec": "av1",
"video.pix_fmt": "yuv420p",
"video.is_depth_map": false,
"video.fps": 50,
"video.channels": 3,
"has_audio": false
}
},
"timestamp": {
"dtype": "float32",
"shape": [
1
],
"names": null
},
"frame_index": {
"dtype": "int64",
"shape": [
1
],
"names": null
},
"episode_index": {
"dtype": "int64",
"shape": [
1
],
"names": null
},
"index": {
"dtype": "int64",
"shape": [
1
],
"names": null
},
"task_index": {
"dtype": "int64",
"shape": [
1
],
"names": null
}
}
}
```

## Citation

**BibTeX:**

```bibtex
[More Information Needed]
```