- คำอธิบาย :
การจัดการบนโต๊ะด้วยวัตถุ 17 ชิ้น
หน้าแรก : https://ai.googleblog.com/2022/12/rt-1-robotics-transformer-for-real.html
ซอร์สโค้ด :
tfds.robotics.rtx.Fractal20220817Data
รุ่น :
-
0.1.0
(ค่าเริ่มต้น): การเปิดตัวครั้งแรก
-
ขนาดการดาวน์โหลด :
Unknown size
ขนาดชุดข้อมูล :
111.38 GiB
แคชอัตโนมัติ ( เอกสาร ): No
แยก :
แยก | ตัวอย่าง |
---|---|
'train' | 87,212 |
- โครงสร้างคุณสมบัติ :
FeaturesDict({
'aspects': FeaturesDict({
'already_success': bool,
'feasible': bool,
'has_aspects': bool,
'success': bool,
'undesirable': bool,
}),
'attributes': FeaturesDict({
'collection_mode': int64,
'collection_mode_name': string,
'data_type': int64,
'data_type_name': string,
'env': int64,
'env_name': string,
'location': int64,
'location_name': string,
'objects_family': int64,
'objects_family_name': string,
'task_family': int64,
'task_family_name': string,
}),
'steps': Dataset({
'action': FeaturesDict({
'base_displacement_vector': Tensor(shape=(2,), dtype=float32),
'base_displacement_vertical_rotation': Tensor(shape=(1,), dtype=float32),
'gripper_closedness_action': Tensor(shape=(1,), dtype=float32, description=continuous gripper position),
'rotation_delta': Tensor(shape=(3,), dtype=float32, description=rpy commanded orientation displacement, in base-relative frame),
'terminate_episode': Tensor(shape=(3,), dtype=int32),
'world_vector': Tensor(shape=(3,), dtype=float32, description=commanded end-effector displacement, in base-relative frame),
}),
'is_first': bool,
'is_last': bool,
'is_terminal': bool,
'observation': FeaturesDict({
'base_pose_tool_reached': Tensor(shape=(7,), dtype=float32, description=end-effector base-relative position+quaternion pose),
'gripper_closed': Tensor(shape=(1,), dtype=float32),
'gripper_closedness_commanded': Tensor(shape=(1,), dtype=float32, description=continuous gripper position),
'height_to_bottom': Tensor(shape=(1,), dtype=float32, description=height of end-effector from ground),
'image': Image(shape=(256, 320, 3), dtype=uint8),
'natural_language_embedding': Tensor(shape=(512,), dtype=float32),
'natural_language_instruction': string,
'orientation_box': Tensor(shape=(2, 3), dtype=float32),
'orientation_start': Tensor(shape=(4,), dtype=float32),
'robot_orientation_positions_box': Tensor(shape=(3, 3), dtype=float32),
'rotation_delta_to_go': Tensor(shape=(3,), dtype=float32, description=rotational displacement from current orientation to target),
'src_rotation': Tensor(shape=(4,), dtype=float32),
'vector_to_go': Tensor(shape=(3,), dtype=float32, description=displacement from current end-effector position to target),
'workspace_bounds': Tensor(shape=(3, 3), dtype=float32),
}),
'reward': Scalar(shape=(), dtype=float32),
}),
})
- เอกสารคุณสมบัติ :
คุณสมบัติ | ระดับ | รูปร่าง | ประเภทD | คำอธิบาย |
---|---|---|---|---|
คุณสมบัติDict | ||||
ด้าน | คุณสมบัติDict | ลักษณะเซสชันสำหรับการให้คะแนนของคราวด์คอมพิวท์ | ||
ด้าน/แล้ว_ความสำเร็จ | เทนเซอร์ | บูล | ||
ด้าน/เป็นไปได้ | เทนเซอร์ | บูล | ||
ด้าน / has_aspects | เทนเซอร์ | บูล | ||
ด้าน/ความสำเร็จ | เทนเซอร์ | บูล | ||
ด้าน/ที่ไม่พึงประสงค์ | เทนเซอร์ | บูล | ||
คุณลักษณะ | คุณสมบัติDict | |||
คุณลักษณะ/คอลเลกชัน_โหมด | เทนเซอร์ | int64 | ||
คุณลักษณะ/collection_mode_name | เทนเซอร์ | เชือก | ||
คุณลักษณะ/data_type | เทนเซอร์ | int64 | ||
คุณลักษณะ/data_type_name | เทนเซอร์ | เชือก | ||
คุณสมบัติ/สภาพแวดล้อม | เทนเซอร์ | int64 | ||
คุณลักษณะ/env_name | เทนเซอร์ | เชือก | ||
คุณสมบัติ/สถานที่ | เทนเซอร์ | int64 | ||
คุณลักษณะ/location_name | เทนเซอร์ | เชือก | ||
คุณลักษณะ/objects_family | เทนเซอร์ | int64 | ||
คุณลักษณะ/objects_family_name | เทนเซอร์ | เชือก | ||
คุณลักษณะ/task_family | เทนเซอร์ | int64 | ||
คุณลักษณะ/task_family_name | เทนเซอร์ | เชือก | ||
ขั้นตอน | ชุดข้อมูล | |||
ขั้นตอน/การดำเนินการ | คุณสมบัติDict | |||
ขั้นตอน/การกระทำ/base_displacement_vector | เทนเซอร์ | (2,) | ลอย32 | |
ขั้นตอน/การกระทำ/base_displacement_vertical_rotation | เทนเซอร์ | (1,) | ลอย32 | |
ขั้นตอน/การกระทำ/อุปกรณ์จับยึด_ปิดแน่น_การกระทำ | เทนเซอร์ | (1,) | ลอย32 | ตำแหน่งกริปเปอร์ต่อเนื่อง |
ขั้นตอน/การกระทำ/rotation_delta | เทนเซอร์ | (3,) | ลอย32 | rpy กำหนดทิศทางการกระจัดในกรอบฐานสัมพัทธ์ |
ขั้นตอน/การกระทำ/terminate_episode | เทนเซอร์ | (3,) | int32 | |
ขั้นตอน/การกระทำ/world_vector | เทนเซอร์ | (3,) | ลอย32 | ได้รับคำสั่งให้เปลี่ยนตำแหน่งเอ็ฟเฟ็กเตอร์ในกรอบฐานสัมพัทธ์ |
ขั้นตอน/is_first | เทนเซอร์ | บูล | ||
ขั้นตอน/is_last | เทนเซอร์ | บูล | ||
ขั้นตอน/is_terminal | เทนเซอร์ | บูล | ||
ขั้นตอน/การสังเกต | คุณสมบัติDict | |||
ขั้นตอน/การสังเกต/base_pose_tool_reached | เทนเซอร์ | (7,) | ลอย32 | ตำแหน่งสัมพันธ์ฐานเอฟเฟกต์ปลาย + ท่าควอเทอร์เนียน |
ขั้นตอน/การสังเกต/gripper_closed | เทนเซอร์ | (1,) | ลอย32 | |
ขั้นตอน/การสังเกต/gripper_closedness_commanded | เทนเซอร์ | (1,) | ลอย32 | ตำแหน่งกริปเปอร์ต่อเนื่อง |
ขั้นตอน/การสังเกต/height_to_bottom | เทนเซอร์ | (1,) | ลอย32 | ความสูงของเอนด์เอฟเฟกต์จากพื้นดิน |
ขั้นตอน/การสังเกต/ภาพ | ภาพ | (256, 320, 3) | uint8 | |
ขั้นตอน/การสังเกต/natural_Language_embedding | เทนเซอร์ | (512,) | ลอย32 | |
ขั้นตอน/การสังเกต/natural_Language_instruction | เทนเซอร์ | เชือก | ||
ขั้นตอน/การสังเกต/orientation_box | เทนเซอร์ | (2, 3) | ลอย32 | |
ขั้นตอน/การสังเกต/orientation_start | เทนเซอร์ | (4,) | ลอย32 | |
ขั้นตอน/การสังเกต/robot_orientation_positions_box | เทนเซอร์ | (3, 3) | ลอย32 | |
ขั้นตอน/การสังเกต/rotation_delta_to_go | เทนเซอร์ | (3,) | ลอย32 | การกระจัดแบบหมุนจากการวางแนวปัจจุบันไปยังเป้าหมาย |
ขั้นตอน/การสังเกต/src_rotation | เทนเซอร์ | (4,) | ลอย32 | |
ขั้นตอน/การสังเกต/vector_to_go | เทนเซอร์ | (3,) | ลอย32 | การกระจัดจากตำแหน่งเอฟเฟกต์ปลายปัจจุบันไปยังเป้าหมาย |
ขั้นตอน/การสังเกต/พื้นที่ทำงาน_bounds | เทนเซอร์ | (3, 3) | ลอย32 | |
ขั้นตอน/รางวัล | สเกลาร์ | ลอย32 |
คีย์ภายใต้การดูแล (ดู
as_supervised
doc ):None
รูปภาพ ( tfds.show_examples ): ไม่รองรับ
ตัวอย่าง ( tfds.as_dataframe ):
- การอ้างอิง :
@article{brohan2022rt,
title={Rt-1: Robotics transformer for real-world control at scale},
author={Brohan, Anthony and Brown, Noah and Carbajal, Justice and Chebotar, Yevgen and Dabis, Joseph and Finn, Chelsea and Gopalakrishnan, Keerthana and Hausman, Karol and Herzog, Alex and Hsu, Jasmine and others},
journal={arXiv preprint arXiv:2212.06817},
year={2022}
}