jasmine314342/picknplace-bimanual-360

picknplace36020260201

LeRobot dataset for robot manipulation.

Dataset Info

Features

• —action: float32 [12]
• —observation.state: float32 [12]
• —observation.images.head: video (480x640, av1)
• —observation.images.left_wrist: video (480x640, av1)
• —observation.images.right_wrist: video (480x640, av1)
• —timestamp: float32 [1]
• —frame_index: int64 [1]
• —episode_index: int64 [1]
• —index: int64 [1]
• —task_index: int64 [1]

Tasks

• —Use left arm to pick up the banana and place it on the plate
• —Use left arm to pick up the corn and place it on the plate
• —Use right arm to pick up the banana and place it on the plate
• —Use right arm to pick up the corn and place it on the plate
• —Use left arm to pick up the bread and place it on the plate
• —Use right arm to pick up the bread and place it on the plate
• —Use left arm to pick up the yogurt bottle and place it in the bin
• —Use left arm to pick up the used tissue and place it in the bin
• —Use right arm to pick up the yogurt bottle and place it in the bin
• —Use right arm to pick up the used tissue and place it in the bin
• —Use left arm to pick up the ketchup bottle and place it in the bin
• —Use right arm to pick up the ketchup bottle and place it in the bin

Usage

from lerobot.common.datasets.lerobot_dataset import LeRobotDataset

dataset = LeRobotDataset("REPO_ID")