रोबोमिमिक_ph

  • विवरण :

Robomimic कुशल मानव डेटासेट को 1 कुशल ऑपरेटर द्वारा RoboTurk प्लेटफॉर्म का उपयोग करके एकत्र किया गया था (परिवहन के अपवाद के साथ, जिसमें 2 कुशल ऑपरेटर एक साथ काम कर रहे थे)। प्रत्येक डेटासेट में 200 सफल प्रक्षेपवक्र होते हैं।

प्रत्येक कार्य के दो संस्करण हैं: एक कम आयामी अवलोकन ( low_dim ) के साथ, और एक चित्र ( image ) के साथ।

डेटासेट चरण और एपिसोड का प्रतिनिधित्व करने के लिए आरएलडीएस प्रारूप का पालन करते हैं।

विभाजित करना उदाहरण
'train' 200
  • पर्यवेक्षित कुंजियाँ ( as_supervised doc देखें): None

  • चित्र ( tfds.show_examples ): समर्थित नहीं है।

  • उदाहरण ( tfds.as_dataframe ): गुम।

  • उद्धरण :

@inproceedings{robomimic2021,
  title={What Matters in Learning from Offline Human Demonstrations for Robot Manipulation},
  author={Ajay Mandlekar and Danfei Xu and Josiah Wong and Soroush Nasiriany
          and Chen Wang and Rohun Kulkarni and Li Fei-Fei and Silvio Savarese
          and Yuke Zhu and Roberto Mart\'{i}n-Mart\'{i}n},
  booktitle={Conference on Robot Learning},
  year={2021}
}

Robomimic_ph/lift_ph_image (डिफ़ॉल्ट कॉन्फ़िगरेशन)

  • डाउनलोड आकार : 798.43 MiB

  • डेटासेट का आकार : 114.47 MiB

  • ऑटो-कैश्ड ( दस्तावेज़ीकरण ): हाँ

  • फ़ीचर संरचना :

FeaturesDict({
    '20_percent': bool,
    '20_percent_train': bool,
    '20_percent_valid': bool,
    '50_percent': bool,
    '50_percent_train': bool,
    '50_percent_valid': bool,
    'episode_id': string,
    'horizon': int32,
    'steps': Dataset({
        'action': Tensor(shape=(7,), dtype=float64),
        'discount': int32,
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': FeaturesDict({
            'agentview_image': Image(shape=(84, 84, 3), dtype=uint8),
            'object': Tensor(shape=(10,), dtype=float64),
            'robot0_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot0_eef_vel_ang': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_vel_lin': Tensor(shape=(3,), dtype=float64),
            'robot0_eye_in_hand_image': Image(shape=(84, 84, 3), dtype=uint8),
            'robot0_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot0_gripper_qvel': Tensor(shape=(2,), dtype=float64),
            'robot0_joint_pos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_cos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_sin': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_vel': Tensor(shape=(7,), dtype=float64),
        }),
        'reward': float64,
        'states': Tensor(shape=(32,), dtype=float64),
    }),
    'train': bool,
    'valid': bool,
})
  • फ़ीचर दस्तावेज़ीकरण :
विशेषता कक्षा आकार डीटाइप विवरण
विशेषताएं डिक्ट
इसे स्वीकार करो टेन्सर बूल
20_प्रतिशत_ट्रेन टेन्सर बूल
20_प्रतिशत_वैध टेन्सर बूल
50 प्रतिशत टेन्सर बूल
50_प्रतिशत_ट्रेन टेन्सर बूल
50_प्रतिशत_वैध टेन्सर बूल
प्रकरण_आईडी टेन्सर डोरी
क्षितिज टेन्सर int32
कदम डेटासेट
कदम / कार्रवाई टेन्सर (7,) फ्लोट64
चरण/छूट टेन्सर int32
चरण/is_first टेन्सर बूल
चरण/is_last टेन्सर बूल
कदम/is_terminal टेन्सर बूल
चरण/अवलोकन विशेषताएं डिक्ट
कदम/अवलोकन/agentview_image छवि (84, 84, 3) uint8
चरण/अवलोकन/ऑब्जेक्ट टेन्सर (10,) फ्लोट64
चरण/अवलोकन/robot0_eef_pos टेन्सर (3,) फ्लोट64 अंत-प्रभावक स्थिति
चरण/अवलोकन/robot0_eef_quat टेन्सर (4,) फ्लोट64 अंत-प्रभावक अभिविन्यास
चरण/अवलोकन/robot0_eef_vel_ang टेन्सर (3,) फ्लोट64 अंत-प्रभावक कोणीय वेग
चरण/अवलोकन/robot0_eef_vel_lin टेन्सर (3,) फ्लोट64 अंत-प्रभावकार कार्टेशियन वेग
चरण/अवलोकन/robot0_eye_in_hand_image छवि (84, 84, 3) uint8
चरण/अवलोकन/robot0_gripper_qpos टेन्सर (2,) फ्लोट64 ग्रिपर की स्थिति
चरण/अवलोकन/robot0_gripper_qvel टेन्सर (2,) फ्लोट64 ग्रिपर वेग
चरण/अवलोकन/robot0_joint_pos टेन्सर (7,) फ्लोट64 7DOF संयुक्त स्थिति
चरण/अवलोकन/robot0_joint_pos_cos टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_pos_sin टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_vel टेन्सर (7,) फ्लोट64 7DOF संयुक्त वेग
कदम / इनाम टेन्सर फ्लोट64
चरण/राज्य टेन्सर (32,) फ्लोट64
रेलगाड़ी टेन्सर बूल
वैध टेन्सर बूल

रोबोमिक_फ/लिफ्ट_फ_लो_डिम

  • डाउनलोड आकार : 17.69 MiB

  • डेटासेट का आकार : 8.50 MiB

  • ऑटो-कैश्ड ( दस्तावेज़ीकरण ): हाँ

  • फ़ीचर संरचना :

FeaturesDict({
    '20_percent': bool,
    '20_percent_train': bool,
    '20_percent_valid': bool,
    '50_percent': bool,
    '50_percent_train': bool,
    '50_percent_valid': bool,
    'episode_id': string,
    'horizon': int32,
    'steps': Dataset({
        'action': Tensor(shape=(7,), dtype=float64),
        'discount': int32,
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': FeaturesDict({
            'object': Tensor(shape=(10,), dtype=float64),
            'robot0_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot0_eef_vel_ang': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_vel_lin': Tensor(shape=(3,), dtype=float64),
            'robot0_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot0_gripper_qvel': Tensor(shape=(2,), dtype=float64),
            'robot0_joint_pos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_cos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_sin': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_vel': Tensor(shape=(7,), dtype=float64),
        }),
        'reward': float64,
        'states': Tensor(shape=(32,), dtype=float64),
    }),
    'train': bool,
    'valid': bool,
})
  • फ़ीचर दस्तावेज़ीकरण :
विशेषता कक्षा आकार डीटाइप विवरण
विशेषताएं डिक्ट
इसे स्वीकार करो टेन्सर बूल
20_प्रतिशत_ट्रेन टेन्सर बूल
20_प्रतिशत_वैध टेन्सर बूल
50 प्रतिशत टेन्सर बूल
50_प्रतिशत_ट्रेन टेन्सर बूल
50_प्रतिशत_वैध टेन्सर बूल
प्रकरण_आईडी टेन्सर डोरी
क्षितिज टेन्सर int32
कदम डेटासेट
कदम / कार्रवाई टेन्सर (7,) फ्लोट64
चरण/छूट टेन्सर int32
चरण/is_first टेन्सर बूल
चरण/is_last टेन्सर बूल
कदम/is_terminal टेन्सर बूल
चरण/अवलोकन विशेषताएं डिक्ट
चरण/अवलोकन/ऑब्जेक्ट टेन्सर (10,) फ्लोट64
चरण/अवलोकन/robot0_eef_pos टेन्सर (3,) फ्लोट64 अंत-प्रभावक स्थिति
चरण/अवलोकन/robot0_eef_quat टेन्सर (4,) फ्लोट64 अंत-प्रभावक अभिविन्यास
चरण/अवलोकन/robot0_eef_vel_ang टेन्सर (3,) फ्लोट64 अंत-प्रभावक कोणीय वेग
चरण/अवलोकन/robot0_eef_vel_lin टेन्सर (3,) फ्लोट64 अंत-प्रभावकार कार्टेशियन वेग
चरण/अवलोकन/robot0_gripper_qpos टेन्सर (2,) फ्लोट64 ग्रिपर की स्थिति
चरण/अवलोकन/robot0_gripper_qvel टेन्सर (2,) फ्लोट64 ग्रिपर वेग
चरण/अवलोकन/robot0_joint_pos टेन्सर (7,) फ्लोट64 7DOF संयुक्त स्थिति
चरण/अवलोकन/robot0_joint_pos_cos टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_pos_sin टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_vel टेन्सर (7,) फ्लोट64 7DOF संयुक्त वेग
कदम / इनाम टेन्सर फ्लोट64
चरण/राज्य टेन्सर (32,) फ्लोट64
रेलगाड़ी टेन्सर बूल
वैध टेन्सर बूल

Robomimic_ph/can_ph_image

  • डाउनलोड आकार : 1.87 GiB

  • डेटासेट का आकार : 474.55 MiB

  • ऑटो-कैश्ड ( दस्तावेज़ीकरण ): नहीं

  • फ़ीचर संरचना :

FeaturesDict({
    '20_percent': bool,
    '20_percent_train': bool,
    '20_percent_valid': bool,
    '50_percent': bool,
    '50_percent_train': bool,
    '50_percent_valid': bool,
    'episode_id': string,
    'horizon': int32,
    'steps': Dataset({
        'action': Tensor(shape=(7,), dtype=float64),
        'discount': int32,
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': FeaturesDict({
            'agentview_image': Image(shape=(84, 84, 3), dtype=uint8),
            'object': Tensor(shape=(14,), dtype=float64),
            'robot0_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot0_eef_vel_ang': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_vel_lin': Tensor(shape=(3,), dtype=float64),
            'robot0_eye_in_hand_image': Image(shape=(84, 84, 3), dtype=uint8),
            'robot0_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot0_gripper_qvel': Tensor(shape=(2,), dtype=float64),
            'robot0_joint_pos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_cos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_sin': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_vel': Tensor(shape=(7,), dtype=float64),
        }),
        'reward': float64,
        'states': Tensor(shape=(71,), dtype=float64),
    }),
    'train': bool,
    'valid': bool,
})
  • फ़ीचर दस्तावेज़ीकरण :
विशेषता कक्षा आकार डीटाइप विवरण
विशेषताएं डिक्ट
इसे स्वीकार करो टेन्सर बूल
20_प्रतिशत_ट्रेन टेन्सर बूल
20_प्रतिशत_वैध टेन्सर बूल
50 प्रतिशत टेन्सर बूल
50_प्रतिशत_ट्रेन टेन्सर बूल
50_प्रतिशत_वैध टेन्सर बूल
प्रकरण_आईडी टेन्सर डोरी
क्षितिज टेन्सर int32
कदम डेटासेट
कदम / कार्रवाई टेन्सर (7,) फ्लोट64
चरण/छूट टेन्सर int32
चरण/is_first टेन्सर बूल
चरण/is_last टेन्सर बूल
कदम/is_terminal टेन्सर बूल
चरण/अवलोकन विशेषताएं डिक्ट
कदम/अवलोकन/agentview_image छवि (84, 84, 3) uint8
चरण/अवलोकन/ऑब्जेक्ट टेन्सर (14,) फ्लोट64
चरण/अवलोकन/robot0_eef_pos टेन्सर (3,) फ्लोट64 अंत-प्रभावक स्थिति
चरण/अवलोकन/robot0_eef_quat टेन्सर (4,) फ्लोट64 अंत-प्रभावक अभिविन्यास
चरण/अवलोकन/robot0_eef_vel_ang टेन्सर (3,) फ्लोट64 अंत-प्रभावक कोणीय वेग
चरण/अवलोकन/robot0_eef_vel_lin टेन्सर (3,) फ्लोट64 अंत-प्रभावकार कार्टेशियन वेग
चरण/अवलोकन/robot0_eye_in_hand_image छवि (84, 84, 3) uint8
चरण/अवलोकन/robot0_gripper_qpos टेन्सर (2,) फ्लोट64 ग्रिपर की स्थिति
चरण/अवलोकन/robot0_gripper_qvel टेन्सर (2,) फ्लोट64 ग्रिपर वेग
चरण/अवलोकन/robot0_joint_pos टेन्सर (7,) फ्लोट64 7DOF संयुक्त स्थिति
चरण/अवलोकन/robot0_joint_pos_cos टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_pos_sin टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_vel टेन्सर (7,) फ्लोट64 7DOF संयुक्त वेग
कदम / इनाम टेन्सर फ्लोट64
चरण/राज्य टेन्सर (71,) फ्लोट64
रेलगाड़ी टेन्सर बूल
वैध टेन्सर बूल

Robomimic_ph/can_ph_low_dim

  • डाउनलोड आकार : 43.38 MiB

  • डेटासेट का आकार : 27.73 MiB

  • ऑटो-कैश्ड ( दस्तावेज़ीकरण ): हाँ

  • फ़ीचर संरचना :

FeaturesDict({
    '20_percent': bool,
    '20_percent_train': bool,
    '20_percent_valid': bool,
    '50_percent': bool,
    '50_percent_train': bool,
    '50_percent_valid': bool,
    'episode_id': string,
    'horizon': int32,
    'steps': Dataset({
        'action': Tensor(shape=(7,), dtype=float64),
        'discount': int32,
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': FeaturesDict({
            'object': Tensor(shape=(14,), dtype=float64),
            'robot0_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot0_eef_vel_ang': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_vel_lin': Tensor(shape=(3,), dtype=float64),
            'robot0_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot0_gripper_qvel': Tensor(shape=(2,), dtype=float64),
            'robot0_joint_pos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_cos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_sin': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_vel': Tensor(shape=(7,), dtype=float64),
        }),
        'reward': float64,
        'states': Tensor(shape=(71,), dtype=float64),
    }),
    'train': bool,
    'valid': bool,
})
  • फ़ीचर दस्तावेज़ीकरण :
विशेषता कक्षा आकार डीटाइप विवरण
विशेषताएं डिक्ट
इसे स्वीकार करो टेन्सर बूल
20_प्रतिशत_ट्रेन टेन्सर बूल
20_प्रतिशत_वैध टेन्सर बूल
50 प्रतिशत टेन्सर बूल
50_प्रतिशत_ट्रेन टेन्सर बूल
50_प्रतिशत_वैध टेन्सर बूल
प्रकरण_आईडी टेन्सर डोरी
क्षितिज टेन्सर int32
कदम डेटासेट
कदम / कार्रवाई टेन्सर (7,) फ्लोट64
चरण/छूट टेन्सर int32
चरण/is_first टेन्सर बूल
चरण/is_last टेन्सर बूल
कदम/is_terminal टेन्सर बूल
चरण/अवलोकन विशेषताएं डिक्ट
चरण/अवलोकन/ऑब्जेक्ट टेन्सर (14,) फ्लोट64
चरण/अवलोकन/robot0_eef_pos टेन्सर (3,) फ्लोट64 अंत-प्रभावक स्थिति
चरण/अवलोकन/robot0_eef_quat टेन्सर (4,) फ्लोट64 अंत-प्रभावक अभिविन्यास
चरण/अवलोकन/robot0_eef_vel_ang टेन्सर (3,) फ्लोट64 अंत-प्रभावक कोणीय वेग
चरण/अवलोकन/robot0_eef_vel_lin टेन्सर (3,) फ्लोट64 अंत-प्रभावकार कार्टेशियन वेग
चरण/अवलोकन/robot0_gripper_qpos टेन्सर (2,) फ्लोट64 ग्रिपर की स्थिति
चरण/अवलोकन/robot0_gripper_qvel टेन्सर (2,) फ्लोट64 ग्रिपर वेग
चरण/अवलोकन/robot0_joint_pos टेन्सर (7,) फ्लोट64 7DOF संयुक्त स्थिति
चरण/अवलोकन/robot0_joint_pos_cos टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_pos_sin टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_vel टेन्सर (7,) फ्लोट64 7DOF संयुक्त वेग
कदम / इनाम टेन्सर फ्लोट64
चरण/राज्य टेन्सर (71,) फ्लोट64
रेलगाड़ी टेन्सर बूल
वैध टेन्सर बूल

Robomimic_ph/square_ph_image

  • डाउनलोड आकार : 2.42 GiB

  • डेटासेट का आकार : 401.28 MiB

  • ऑटो-कैश्ड ( दस्तावेज़ीकरण ): नहीं

  • फ़ीचर संरचना :

FeaturesDict({
    '20_percent': bool,
    '20_percent_train': bool,
    '20_percent_valid': bool,
    '50_percent': bool,
    '50_percent_train': bool,
    '50_percent_valid': bool,
    'episode_id': string,
    'horizon': int32,
    'steps': Dataset({
        'action': Tensor(shape=(7,), dtype=float64),
        'discount': int32,
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': FeaturesDict({
            'agentview_image': Image(shape=(84, 84, 3), dtype=uint8),
            'object': Tensor(shape=(14,), dtype=float64),
            'robot0_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot0_eef_vel_ang': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_vel_lin': Tensor(shape=(3,), dtype=float64),
            'robot0_eye_in_hand_image': Image(shape=(84, 84, 3), dtype=uint8),
            'robot0_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot0_gripper_qvel': Tensor(shape=(2,), dtype=float64),
            'robot0_joint_pos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_cos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_sin': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_vel': Tensor(shape=(7,), dtype=float64),
        }),
        'reward': float64,
        'states': Tensor(shape=(45,), dtype=float64),
    }),
    'train': bool,
    'valid': bool,
})
  • फ़ीचर दस्तावेज़ीकरण :
विशेषता कक्षा आकार डीटाइप विवरण
विशेषताएं डिक्ट
इसे स्वीकार करो टेन्सर बूल
20_प्रतिशत_ट्रेन टेन्सर बूल
20_प्रतिशत_वैध टेन्सर बूल
50 प्रतिशत टेन्सर बूल
50_प्रतिशत_ट्रेन टेन्सर बूल
50_प्रतिशत_वैध टेन्सर बूल
प्रकरण_आईडी टेन्सर डोरी
क्षितिज टेन्सर int32
कदम डेटासेट
कदम / कार्रवाई टेन्सर (7,) फ्लोट64
चरण/छूट टेन्सर int32
चरण/is_first टेन्सर बूल
चरण/is_last टेन्सर बूल
कदम/is_terminal टेन्सर बूल
चरण/अवलोकन विशेषताएं डिक्ट
कदम/अवलोकन/agentview_image छवि (84, 84, 3) uint8
चरण/अवलोकन/ऑब्जेक्ट टेन्सर (14,) फ्लोट64
चरण/अवलोकन/robot0_eef_pos टेन्सर (3,) फ्लोट64 अंत-प्रभावक स्थिति
चरण/अवलोकन/robot0_eef_quat टेन्सर (4,) फ्लोट64 अंत-प्रभावक अभिविन्यास
चरण/अवलोकन/robot0_eef_vel_ang टेन्सर (3,) फ्लोट64 अंत-प्रभावक कोणीय वेग
चरण/अवलोकन/robot0_eef_vel_lin टेन्सर (3,) फ्लोट64 अंत-प्रभावकार कार्टेशियन वेग
चरण/अवलोकन/robot0_eye_in_hand_image छवि (84, 84, 3) uint8
चरण/अवलोकन/robot0_gripper_qpos टेन्सर (2,) फ्लोट64 ग्रिपर की स्थिति
चरण/अवलोकन/robot0_gripper_qvel टेन्सर (2,) फ्लोट64 ग्रिपर वेग
चरण/अवलोकन/robot0_joint_pos टेन्सर (7,) फ्लोट64 7DOF संयुक्त स्थिति
चरण/अवलोकन/robot0_joint_pos_cos टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_pos_sin टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_vel टेन्सर (7,) फ्लोट64 7DOF संयुक्त वेग
कदम / इनाम टेन्सर फ्लोट64
चरण/राज्य टेन्सर (45,) फ्लोट64
रेलगाड़ी टेन्सर बूल
वैध टेन्सर बूल

Robomimic_ph/square_ph_low_dim

  • डाउनलोड आकार : 47.69 MiB

  • डेटासेट का आकार : 29.91 MiB

  • ऑटो-कैश्ड ( दस्तावेज़ीकरण ): हाँ

  • फ़ीचर संरचना :

FeaturesDict({
    '20_percent': bool,
    '20_percent_train': bool,
    '20_percent_valid': bool,
    '50_percent': bool,
    '50_percent_train': bool,
    '50_percent_valid': bool,
    'episode_id': string,
    'horizon': int32,
    'steps': Dataset({
        'action': Tensor(shape=(7,), dtype=float64),
        'discount': int32,
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': FeaturesDict({
            'object': Tensor(shape=(14,), dtype=float64),
            'robot0_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot0_eef_vel_ang': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_vel_lin': Tensor(shape=(3,), dtype=float64),
            'robot0_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot0_gripper_qvel': Tensor(shape=(2,), dtype=float64),
            'robot0_joint_pos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_cos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_sin': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_vel': Tensor(shape=(7,), dtype=float64),
        }),
        'reward': float64,
        'states': Tensor(shape=(45,), dtype=float64),
    }),
    'train': bool,
    'valid': bool,
})
  • फ़ीचर दस्तावेज़ीकरण :
विशेषता कक्षा आकार डीटाइप विवरण
विशेषताएं डिक्ट
इसे स्वीकार करो टेन्सर बूल
20_प्रतिशत_ट्रेन टेन्सर बूल
20_प्रतिशत_वैध टेन्सर बूल
50 प्रतिशत टेन्सर बूल
50_प्रतिशत_ट्रेन टेन्सर बूल
50_प्रतिशत_वैध टेन्सर बूल
प्रकरण_आईडी टेन्सर डोरी
क्षितिज टेन्सर int32
कदम डेटासेट
कदम / कार्रवाई टेन्सर (7,) फ्लोट64
चरण/छूट टेन्सर int32
चरण/is_first टेन्सर बूल
चरण/is_last टेन्सर बूल
कदम/is_terminal टेन्सर बूल
चरण/अवलोकन विशेषताएं डिक्ट
चरण/अवलोकन/ऑब्जेक्ट टेन्सर (14,) फ्लोट64
चरण/अवलोकन/robot0_eef_pos टेन्सर (3,) फ्लोट64 अंत-प्रभावक स्थिति
चरण/अवलोकन/robot0_eef_quat टेन्सर (4,) फ्लोट64 अंत-प्रभावक अभिविन्यास
चरण/अवलोकन/robot0_eef_vel_ang टेन्सर (3,) फ्लोट64 अंत-प्रभावक कोणीय वेग
चरण/अवलोकन/robot0_eef_vel_lin टेन्सर (3,) फ्लोट64 अंत-प्रभावकार कार्टेशियन वेग
चरण/अवलोकन/robot0_gripper_qpos टेन्सर (2,) फ्लोट64 ग्रिपर की स्थिति
चरण/अवलोकन/robot0_gripper_qvel टेन्सर (2,) फ्लोट64 ग्रिपर वेग
चरण/अवलोकन/robot0_joint_pos टेन्सर (7,) फ्लोट64 7DOF संयुक्त स्थिति
चरण/अवलोकन/robot0_joint_pos_cos टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_pos_sin टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_vel टेन्सर (7,) फ्लोट64 7DOF संयुक्त वेग
कदम / इनाम टेन्सर फ्लोट64
चरण/राज्य टेन्सर (45,) फ्लोट64
रेलगाड़ी टेन्सर बूल
वैध टेन्सर बूल

Robomimic_ph/transport_ph_image

  • डाउनलोड का आकार : 15.07 GiB

  • डेटासेट का आकार : 3.64 GiB

  • ऑटो-कैश्ड ( दस्तावेज़ीकरण ): नहीं

  • फ़ीचर संरचना :

FeaturesDict({
    '20_percent': bool,
    '20_percent_train': bool,
    '20_percent_valid': bool,
    '50_percent': bool,
    '50_percent_train': bool,
    '50_percent_valid': bool,
    'episode_id': string,
    'horizon': int32,
    'steps': Dataset({
        'action': Tensor(shape=(14,), dtype=float64),
        'discount': int32,
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': FeaturesDict({
            'object': Tensor(shape=(41,), dtype=float64),
            'robot0_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot0_eef_vel_ang': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_vel_lin': Tensor(shape=(3,), dtype=float64),
            'robot0_eye_in_hand_image': Image(shape=(84, 84, 3), dtype=uint8),
            'robot0_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot0_gripper_qvel': Tensor(shape=(2,), dtype=float64),
            'robot0_joint_pos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_cos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_sin': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_vel': Tensor(shape=(7,), dtype=float64),
            'robot1_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot1_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot1_eef_vel_ang': Tensor(shape=(3,), dtype=float64),
            'robot1_eef_vel_lin': Tensor(shape=(3,), dtype=float64),
            'robot1_eye_in_hand_image': Image(shape=(84, 84, 3), dtype=uint8),
            'robot1_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot1_gripper_qvel': Tensor(shape=(2,), dtype=float64),
            'robot1_joint_pos': Tensor(shape=(7,), dtype=float64),
            'robot1_joint_pos_cos': Tensor(shape=(7,), dtype=float64),
            'robot1_joint_pos_sin': Tensor(shape=(7,), dtype=float64),
            'robot1_joint_vel': Tensor(shape=(7,), dtype=float64),
            'shouldercamera0_image': Image(shape=(84, 84, 3), dtype=uint8),
            'shouldercamera1_image': Image(shape=(84, 84, 3), dtype=uint8),
        }),
        'reward': float64,
        'states': Tensor(shape=(115,), dtype=float64),
    }),
    'train': bool,
    'valid': bool,
})
  • फ़ीचर दस्तावेज़ीकरण :
विशेषता कक्षा आकार डीटाइप विवरण
विशेषताएं डिक्ट
इसे स्वीकार करो टेन्सर बूल
20_प्रतिशत_ट्रेन टेन्सर बूल
20_प्रतिशत_वैध टेन्सर बूल
50 प्रतिशत टेन्सर बूल
50_प्रतिशत_ट्रेन टेन्सर बूल
50_प्रतिशत_वैध टेन्सर बूल
प्रकरण_आईडी टेन्सर डोरी
क्षितिज टेन्सर int32
कदम डेटासेट
कदम / कार्रवाई टेन्सर (14,) फ्लोट64
चरण/छूट टेन्सर int32
चरण/is_first टेन्सर बूल
चरण/is_last टेन्सर बूल
कदम/is_terminal टेन्सर बूल
चरण/अवलोकन विशेषताएं डिक्ट
चरण/अवलोकन/ऑब्जेक्ट टेन्सर (41,) फ्लोट64
चरण/अवलोकन/robot0_eef_pos टेन्सर (3,) फ्लोट64 अंत-प्रभावक स्थिति
चरण/अवलोकन/robot0_eef_quat टेन्सर (4,) फ्लोट64 अंत-प्रभावक अभिविन्यास
चरण/अवलोकन/robot0_eef_vel_ang टेन्सर (3,) फ्लोट64 अंत-प्रभावक कोणीय वेग
चरण/अवलोकन/robot0_eef_vel_lin टेन्सर (3,) फ्लोट64 अंत-प्रभावकार कार्टेशियन वेग
चरण/अवलोकन/robot0_eye_in_hand_image छवि (84, 84, 3) uint8
चरण/अवलोकन/robot0_gripper_qpos टेन्सर (2,) फ्लोट64 ग्रिपर की स्थिति
चरण/अवलोकन/robot0_gripper_qvel टेन्सर (2,) फ्लोट64 ग्रिपर वेग
चरण/अवलोकन/robot0_joint_pos टेन्सर (7,) फ्लोट64 7DOF संयुक्त स्थिति
चरण/अवलोकन/robot0_joint_pos_cos टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_pos_sin टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_vel टेन्सर (7,) फ्लोट64 7DOF संयुक्त वेग
चरण/अवलोकन/robot1_eef_pos टेन्सर (3,) फ्लोट64 अंत-प्रभावक स्थिति
चरण/अवलोकन/robot1_eef_quat टेन्सर (4,) फ्लोट64 अंत-प्रभावक अभिविन्यास
चरण/अवलोकन/robot1_eef_vel_ang टेन्सर (3,) फ्लोट64 अंत-प्रभावक कोणीय वेग
चरण/अवलोकन/robot1_eef_vel_lin टेन्सर (3,) फ्लोट64 अंत-प्रभावकार कार्टेशियन वेग
चरण/अवलोकन/robot1_eye_in_hand_image छवि (84, 84, 3) uint8
चरण/अवलोकन/robot1_gripper_qpos टेन्सर (2,) फ्लोट64 ग्रिपर की स्थिति
चरण/अवलोकन/robot1_gripper_qvel टेन्सर (2,) फ्लोट64 ग्रिपर वेग
चरण/अवलोकन/robot1_joint_pos टेन्सर (7,) फ्लोट64 7DOF संयुक्त स्थिति
चरण/अवलोकन/robot1_joint_pos_cos टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot1_joint_pos_sin टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot1_joint_vel टेन्सर (7,) फ्लोट64 7DOF संयुक्त वेग
स्टेप्स/ऑब्जर्वेशन/शोल्डरकैमरा0_इमेज छवि (84, 84, 3) uint8
स्टेप्स/ऑब्जर्वेशन/शोल्डरकैमरा1_इमेज छवि (84, 84, 3) uint8
कदम / इनाम टेन्सर फ्लोट64
चरण/राज्य टेन्सर (115,) फ्लोट64
रेलगाड़ी टेन्सर बूल
वैध टेन्सर बूल

Robomimic_ph/transport_ph_low_dim

  • डाउनलोड आकार : 294.70 MiB

  • डेटासेट का आकार : 208.05 MiB

  • ऑटो-कैश्ड ( दस्तावेज़ीकरण ): केवल जब shuffle_files=False (ट्रेन)

  • फ़ीचर संरचना :

FeaturesDict({
    '20_percent': bool,
    '20_percent_train': bool,
    '20_percent_valid': bool,
    '50_percent': bool,
    '50_percent_train': bool,
    '50_percent_valid': bool,
    'episode_id': string,
    'horizon': int32,
    'steps': Dataset({
        'action': Tensor(shape=(14,), dtype=float64),
        'discount': int32,
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': FeaturesDict({
            'object': Tensor(shape=(41,), dtype=float64),
            'robot0_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot0_eef_vel_ang': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_vel_lin': Tensor(shape=(3,), dtype=float64),
            'robot0_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot0_gripper_qvel': Tensor(shape=(2,), dtype=float64),
            'robot0_joint_pos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_cos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_sin': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_vel': Tensor(shape=(7,), dtype=float64),
            'robot1_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot1_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot1_eef_vel_ang': Tensor(shape=(3,), dtype=float64),
            'robot1_eef_vel_lin': Tensor(shape=(3,), dtype=float64),
            'robot1_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot1_gripper_qvel': Tensor(shape=(2,), dtype=float64),
            'robot1_joint_pos': Tensor(shape=(7,), dtype=float64),
            'robot1_joint_pos_cos': Tensor(shape=(7,), dtype=float64),
            'robot1_joint_pos_sin': Tensor(shape=(7,), dtype=float64),
            'robot1_joint_vel': Tensor(shape=(7,), dtype=float64),
        }),
        'reward': float64,
        'states': Tensor(shape=(115,), dtype=float64),
    }),
    'train': bool,
    'valid': bool,
})
  • फ़ीचर दस्तावेज़ीकरण :
विशेषता कक्षा आकार डीटाइप विवरण
विशेषताएं डिक्ट
इसे स्वीकार करो टेन्सर बूल
20_प्रतिशत_ट्रेन टेन्सर बूल
20_प्रतिशत_वैध टेन्सर बूल
50 प्रतिशत टेन्सर बूल
50_प्रतिशत_ट्रेन टेन्सर बूल
50_प्रतिशत_वैध टेन्सर बूल
प्रकरण_आईडी टेन्सर डोरी
क्षितिज टेन्सर int32
कदम डेटासेट
कदम / कार्रवाई टेन्सर (14,) फ्लोट64
चरण/छूट टेन्सर int32
चरण/is_first टेन्सर बूल
चरण/is_last टेन्सर बूल
कदम/is_terminal टेन्सर बूल
चरण/अवलोकन विशेषताएं डिक्ट
चरण/अवलोकन/ऑब्जेक्ट टेन्सर (41,) फ्लोट64
चरण/अवलोकन/robot0_eef_pos टेन्सर (3,) फ्लोट64 अंत-प्रभावक स्थिति
चरण/अवलोकन/robot0_eef_quat टेन्सर (4,) फ्लोट64 अंत-प्रभावक अभिविन्यास
चरण/अवलोकन/robot0_eef_vel_ang टेन्सर (3,) फ्लोट64 अंत-प्रभावक कोणीय वेग
चरण/अवलोकन/robot0_eef_vel_lin टेन्सर (3,) फ्लोट64 अंत-प्रभावकार कार्टेशियन वेग
चरण/अवलोकन/robot0_gripper_qpos टेन्सर (2,) फ्लोट64 ग्रिपर की स्थिति
चरण/अवलोकन/robot0_gripper_qvel टेन्सर (2,) फ्लोट64 ग्रिपर वेग
चरण/अवलोकन/robot0_joint_pos टेन्सर (7,) फ्लोट64 7DOF संयुक्त स्थिति
चरण/अवलोकन/robot0_joint_pos_cos टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_pos_sin टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_vel टेन्सर (7,) फ्लोट64 7DOF संयुक्त वेग
चरण/अवलोकन/robot1_eef_pos टेन्सर (3,) फ्लोट64 अंत-प्रभावक स्थिति
चरण/अवलोकन/robot1_eef_quat टेन्सर (4,) फ्लोट64 अंत-प्रभावक अभिविन्यास
चरण/अवलोकन/robot1_eef_vel_ang टेन्सर (3,) फ्लोट64 अंत-प्रभावक कोणीय वेग
चरण/अवलोकन/robot1_eef_vel_lin टेन्सर (3,) फ्लोट64 अंत-प्रभावकार कार्टेशियन वेग
चरण/अवलोकन/robot1_gripper_qpos टेन्सर (2,) फ्लोट64 ग्रिपर की स्थिति
चरण/अवलोकन/robot1_gripper_qvel टेन्सर (2,) फ्लोट64 ग्रिपर वेग
चरण/अवलोकन/robot1_joint_pos टेन्सर (7,) फ्लोट64 7DOF संयुक्त स्थिति
चरण/अवलोकन/robot1_joint_pos_cos टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot1_joint_pos_sin टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot1_joint_vel टेन्सर (7,) फ्लोट64 7DOF संयुक्त वेग
कदम / इनाम टेन्सर फ्लोट64
चरण/राज्य टेन्सर (115,) फ्लोट64
रेलगाड़ी टेन्सर बूल
वैध टेन्सर बूल

Robomimic_ph/tool_hang_ph_image

  • डाउनलोड का आकार : 61.96 GiB

  • डेटासेट का आकार : 9.10 GiB

  • ऑटो-कैश्ड ( दस्तावेज़ीकरण ): नहीं

  • फ़ीचर संरचना :

FeaturesDict({
    'episode_id': string,
    'horizon': int32,
    'steps': Dataset({
        'action': Tensor(shape=(7,), dtype=float64),
        'discount': int32,
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': FeaturesDict({
            'object': Tensor(shape=(44,), dtype=float64),
            'robot0_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot0_eef_vel_ang': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_vel_lin': Tensor(shape=(3,), dtype=float64),
            'robot0_eye_in_hand_image': Image(shape=(240, 240, 3), dtype=uint8),
            'robot0_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot0_gripper_qvel': Tensor(shape=(2,), dtype=float64),
            'robot0_joint_pos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_cos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_sin': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_vel': Tensor(shape=(7,), dtype=float64),
            'sideview_image': Image(shape=(240, 240, 3), dtype=uint8),
        }),
        'reward': float64,
        'states': Tensor(shape=(58,), dtype=float64),
    }),
    'train': bool,
    'valid': bool,
})
  • फ़ीचर दस्तावेज़ीकरण :
विशेषता कक्षा आकार डीटाइप विवरण
विशेषताएं डिक्ट
प्रकरण_आईडी टेन्सर डोरी
क्षितिज टेन्सर int32
कदम डेटासेट
कदम / कार्रवाई टेन्सर (7,) फ्लोट64
चरण/छूट टेन्सर int32
चरण/is_first टेन्सर बूल
चरण/is_last टेन्सर बूल
कदम/is_terminal टेन्सर बूल
चरण/अवलोकन विशेषताएं डिक्ट
चरण/अवलोकन/ऑब्जेक्ट टेन्सर (44,) फ्लोट64
चरण/अवलोकन/robot0_eef_pos टेन्सर (3,) फ्लोट64 अंत-प्रभावक स्थिति
चरण/अवलोकन/robot0_eef_quat टेन्सर (4,) फ्लोट64 अंत-प्रभावक अभिविन्यास
चरण/अवलोकन/robot0_eef_vel_ang टेन्सर (3,) फ्लोट64 अंत-प्रभावक कोणीय वेग
चरण/अवलोकन/robot0_eef_vel_lin टेन्सर (3,) फ्लोट64 अंत-प्रभावकार कार्टेशियन वेग
चरण/अवलोकन/robot0_eye_in_hand_image छवि (240, 240, 3) uint8
चरण/अवलोकन/robot0_gripper_qpos टेन्सर (2,) फ्लोट64 ग्रिपर की स्थिति
चरण/अवलोकन/robot0_gripper_qvel टेन्सर (2,) फ्लोट64 ग्रिपर वेग
चरण/अवलोकन/robot0_joint_pos टेन्सर (7,) फ्लोट64 7DOF संयुक्त स्थिति
चरण/अवलोकन/robot0_joint_pos_cos टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_pos_sin टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_vel टेन्सर (7,) फ्लोट64 7DOF संयुक्त वेग
चरण/अवलोकन/sideview_image छवि (240, 240, 3) uint8
कदम / इनाम टेन्सर फ्लोट64
चरण/राज्य टेन्सर (58,) फ्लोट64
रेलगाड़ी टेन्सर बूल
वैध टेन्सर बूल

robomimic_ph/tool_hang_ph_low_dim

  • डाउनलोड आकार : 192.29 MiB

  • डेटासेट का आकार : 121.77 MiB

  • ऑटो-कैश्ड ( दस्तावेज़ीकरण ): हाँ

  • फ़ीचर संरचना :

FeaturesDict({
    'episode_id': string,
    'horizon': int32,
    'steps': Dataset({
        'action': Tensor(shape=(7,), dtype=float64),
        'discount': int32,
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': FeaturesDict({
            'object': Tensor(shape=(44,), dtype=float64),
            'robot0_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot0_eef_vel_ang': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_vel_lin': Tensor(shape=(3,), dtype=float64),
            'robot0_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot0_gripper_qvel': Tensor(shape=(2,), dtype=float64),
            'robot0_joint_pos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_cos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_sin': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_vel': Tensor(shape=(7,), dtype=float64),
        }),
        'reward': float64,
        'states': Tensor(shape=(58,), dtype=float64),
    }),
    'train': bool,
    'valid': bool,
})
  • फ़ीचर दस्तावेज़ीकरण :
विशेषता कक्षा आकार डीटाइप विवरण
विशेषताएं डिक्ट
प्रकरण_आईडी टेन्सर डोरी
क्षितिज टेन्सर int32
कदम डेटासेट
कदम / कार्रवाई टेन्सर (7,) फ्लोट64
चरण/छूट टेन्सर int32
चरण/is_first टेन्सर बूल
चरण/is_last टेन्सर बूल
कदम/is_terminal टेन्सर बूल
चरण/अवलोकन विशेषताएं डिक्ट
चरण/अवलोकन/ऑब्जेक्ट टेन्सर (44,) फ्लोट64
चरण/अवलोकन/robot0_eef_pos टेन्सर (3,) फ्लोट64 अंत-प्रभावक स्थिति
चरण/अवलोकन/robot0_eef_quat टेन्सर (4,) फ्लोट64 अंत-प्रभावक अभिविन्यास
चरण/अवलोकन/robot0_eef_vel_ang टेन्सर (3,) फ्लोट64 अंत-प्रभावक कोणीय वेग
चरण/अवलोकन/robot0_eef_vel_lin टेन्सर (3,) फ्लोट64 अंत-प्रभावकार कार्टेशियन वेग
चरण/अवलोकन/robot0_gripper_qpos टेन्सर (2,) फ्लोट64 ग्रिपर की स्थिति
चरण/अवलोकन/robot0_gripper_qvel टेन्सर (2,) फ्लोट64 ग्रिपर वेग
चरण/अवलोकन/robot0_joint_pos टेन्सर (7,) फ्लोट64 7DOF संयुक्त स्थिति
चरण/अवलोकन/robot0_joint_pos_cos टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_pos_sin टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_vel टेन्सर (7,) फ्लोट64 7DOF संयुक्त वेग
कदम / इनाम टेन्सर फ्लोट64
चरण/राज्य टेन्सर (58,) फ्लोट64
रेलगाड़ी टेन्सर बूल
वैध टेन्सर बूल
,

  • विवरण :

Robomimic कुशल मानव डेटासेट को 1 कुशल ऑपरेटर द्वारा RoboTurk प्लेटफॉर्म का उपयोग करके एकत्र किया गया था (परिवहन के अपवाद के साथ, जिसमें 2 कुशल ऑपरेटर एक साथ काम कर रहे थे)। प्रत्येक डेटासेट में 200 सफल प्रक्षेपवक्र होते हैं।

प्रत्येक कार्य के दो संस्करण हैं: एक कम आयामी अवलोकन ( low_dim ) के साथ, और एक चित्र ( image ) के साथ।

डेटासेट चरण और एपिसोड का प्रतिनिधित्व करने के लिए आरएलडीएस प्रारूप का पालन करते हैं।

विभाजित करना उदाहरण
'train' 200
  • पर्यवेक्षित कुंजियाँ ( as_supervised doc देखें): None

  • चित्र ( tfds.show_examples ): समर्थित नहीं है।

  • उदाहरण ( tfds.as_dataframe ): गुम।

  • उद्धरण :

@inproceedings{robomimic2021,
  title={What Matters in Learning from Offline Human Demonstrations for Robot Manipulation},
  author={Ajay Mandlekar and Danfei Xu and Josiah Wong and Soroush Nasiriany
          and Chen Wang and Rohun Kulkarni and Li Fei-Fei and Silvio Savarese
          and Yuke Zhu and Roberto Mart\'{i}n-Mart\'{i}n},
  booktitle={Conference on Robot Learning},
  year={2021}
}

Robomimic_ph/lift_ph_image (डिफ़ॉल्ट कॉन्फ़िगरेशन)

  • डाउनलोड आकार : 798.43 MiB

  • डेटासेट का आकार : 114.47 MiB

  • ऑटो-कैश्ड ( दस्तावेज़ीकरण ): हाँ

  • फ़ीचर संरचना :

FeaturesDict({
    '20_percent': bool,
    '20_percent_train': bool,
    '20_percent_valid': bool,
    '50_percent': bool,
    '50_percent_train': bool,
    '50_percent_valid': bool,
    'episode_id': string,
    'horizon': int32,
    'steps': Dataset({
        'action': Tensor(shape=(7,), dtype=float64),
        'discount': int32,
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': FeaturesDict({
            'agentview_image': Image(shape=(84, 84, 3), dtype=uint8),
            'object': Tensor(shape=(10,), dtype=float64),
            'robot0_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot0_eef_vel_ang': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_vel_lin': Tensor(shape=(3,), dtype=float64),
            'robot0_eye_in_hand_image': Image(shape=(84, 84, 3), dtype=uint8),
            'robot0_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot0_gripper_qvel': Tensor(shape=(2,), dtype=float64),
            'robot0_joint_pos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_cos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_sin': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_vel': Tensor(shape=(7,), dtype=float64),
        }),
        'reward': float64,
        'states': Tensor(shape=(32,), dtype=float64),
    }),
    'train': bool,
    'valid': bool,
})
  • फ़ीचर दस्तावेज़ीकरण :
विशेषता कक्षा आकार डीटाइप विवरण
विशेषताएं डिक्ट
इसे स्वीकार करो टेन्सर बूल
20_प्रतिशत_ट्रेन टेन्सर बूल
20_प्रतिशत_वैध टेन्सर बूल
50 प्रतिशत टेन्सर बूल
50_प्रतिशत_ट्रेन टेन्सर बूल
50_प्रतिशत_वैध टेन्सर बूल
प्रकरण_आईडी टेन्सर डोरी
क्षितिज टेन्सर int32
कदम डेटासेट
कदम / कार्रवाई टेन्सर (7,) फ्लोट64
चरण/छूट टेन्सर int32
चरण/is_first टेन्सर बूल
चरण/is_last टेन्सर बूल
कदम/is_terminal टेन्सर बूल
चरण/अवलोकन विशेषताएं डिक्ट
कदम/अवलोकन/agentview_image छवि (84, 84, 3) uint8
चरण/अवलोकन/ऑब्जेक्ट टेन्सर (10,) फ्लोट64
चरण/अवलोकन/robot0_eef_pos टेन्सर (3,) फ्लोट64 अंत-प्रभावक स्थिति
चरण/अवलोकन/robot0_eef_quat टेन्सर (4,) फ्लोट64 अंत-प्रभावक अभिविन्यास
चरण/अवलोकन/robot0_eef_vel_ang टेन्सर (3,) फ्लोट64 अंत-प्रभावक कोणीय वेग
चरण/अवलोकन/robot0_eef_vel_lin टेन्सर (3,) फ्लोट64 अंत-प्रभावकार कार्टेशियन वेग
चरण/अवलोकन/robot0_eye_in_hand_image छवि (84, 84, 3) uint8
चरण/अवलोकन/robot0_gripper_qpos टेन्सर (2,) फ्लोट64 ग्रिपर की स्थिति
चरण/अवलोकन/robot0_gripper_qvel टेन्सर (2,) फ्लोट64 ग्रिपर वेग
चरण/अवलोकन/robot0_joint_pos टेन्सर (7,) फ्लोट64 7DOF संयुक्त स्थिति
चरण/अवलोकन/robot0_joint_pos_cos टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_pos_sin टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_vel टेन्सर (7,) फ्लोट64 7DOF संयुक्त वेग
कदम / इनाम टेन्सर फ्लोट64
चरण/राज्य टेन्सर (32,) फ्लोट64
रेलगाड़ी टेन्सर बूल
वैध टेन्सर बूल

रोबोमिक_फ/लिफ्ट_फ_लो_डिम

  • डाउनलोड आकार : 17.69 MiB

  • डेटासेट का आकार : 8.50 MiB

  • ऑटो-कैश्ड ( दस्तावेज़ीकरण ): हाँ

  • फ़ीचर संरचना :

FeaturesDict({
    '20_percent': bool,
    '20_percent_train': bool,
    '20_percent_valid': bool,
    '50_percent': bool,
    '50_percent_train': bool,
    '50_percent_valid': bool,
    'episode_id': string,
    'horizon': int32,
    'steps': Dataset({
        'action': Tensor(shape=(7,), dtype=float64),
        'discount': int32,
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': FeaturesDict({
            'object': Tensor(shape=(10,), dtype=float64),
            'robot0_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot0_eef_vel_ang': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_vel_lin': Tensor(shape=(3,), dtype=float64),
            'robot0_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot0_gripper_qvel': Tensor(shape=(2,), dtype=float64),
            'robot0_joint_pos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_cos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_sin': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_vel': Tensor(shape=(7,), dtype=float64),
        }),
        'reward': float64,
        'states': Tensor(shape=(32,), dtype=float64),
    }),
    'train': bool,
    'valid': bool,
})
  • फ़ीचर दस्तावेज़ीकरण :
विशेषता कक्षा आकार डीटाइप विवरण
विशेषताएं डिक्ट
इसे स्वीकार करो टेन्सर बूल
20_प्रतिशत_ट्रेन टेन्सर बूल
20_प्रतिशत_वैध टेन्सर बूल
50 प्रतिशत टेन्सर बूल
50_प्रतिशत_ट्रेन टेन्सर बूल
50_प्रतिशत_वैध टेन्सर बूल
प्रकरण_आईडी टेन्सर डोरी
क्षितिज टेन्सर int32
कदम डेटासेट
कदम / कार्रवाई टेन्सर (7,) फ्लोट64
चरण/छूट टेन्सर int32
चरण/is_first टेन्सर बूल
चरण/is_last टेन्सर बूल
कदम/is_terminal टेन्सर बूल
चरण/अवलोकन विशेषताएं डिक्ट
चरण/अवलोकन/ऑब्जेक्ट टेन्सर (10,) फ्लोट64
चरण/अवलोकन/robot0_eef_pos टेन्सर (3,) फ्लोट64 अंत-प्रभावक स्थिति
चरण/अवलोकन/robot0_eef_quat टेन्सर (4,) फ्लोट64 अंत-प्रभावक अभिविन्यास
चरण/अवलोकन/robot0_eef_vel_ang टेन्सर (3,) फ्लोट64 अंत-प्रभावक कोणीय वेग
चरण/अवलोकन/robot0_eef_vel_lin टेन्सर (3,) फ्लोट64 अंत-प्रभावकार कार्टेशियन वेग
चरण/अवलोकन/robot0_gripper_qpos टेन्सर (2,) फ्लोट64 ग्रिपर की स्थिति
चरण/अवलोकन/robot0_gripper_qvel टेन्सर (2,) फ्लोट64 ग्रिपर वेग
चरण/अवलोकन/robot0_joint_pos टेन्सर (7,) फ्लोट64 7DOF संयुक्त स्थिति
चरण/अवलोकन/robot0_joint_pos_cos टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_pos_sin टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_vel टेन्सर (7,) फ्लोट64 7DOF संयुक्त वेग
कदम / इनाम टेन्सर फ्लोट64
चरण/राज्य टेन्सर (32,) फ्लोट64
रेलगाड़ी टेन्सर बूल
वैध टेन्सर बूल

Robomimic_ph/can_ph_image

  • डाउनलोड आकार : 1.87 GiB

  • डेटासेट का आकार : 474.55 MiB

  • ऑटो-कैश्ड ( दस्तावेज़ीकरण ): नहीं

  • फ़ीचर संरचना :

FeaturesDict({
    '20_percent': bool,
    '20_percent_train': bool,
    '20_percent_valid': bool,
    '50_percent': bool,
    '50_percent_train': bool,
    '50_percent_valid': bool,
    'episode_id': string,
    'horizon': int32,
    'steps': Dataset({
        'action': Tensor(shape=(7,), dtype=float64),
        'discount': int32,
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': FeaturesDict({
            'agentview_image': Image(shape=(84, 84, 3), dtype=uint8),
            'object': Tensor(shape=(14,), dtype=float64),
            'robot0_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot0_eef_vel_ang': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_vel_lin': Tensor(shape=(3,), dtype=float64),
            'robot0_eye_in_hand_image': Image(shape=(84, 84, 3), dtype=uint8),
            'robot0_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot0_gripper_qvel': Tensor(shape=(2,), dtype=float64),
            'robot0_joint_pos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_cos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_sin': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_vel': Tensor(shape=(7,), dtype=float64),
        }),
        'reward': float64,
        'states': Tensor(shape=(71,), dtype=float64),
    }),
    'train': bool,
    'valid': bool,
})
  • फ़ीचर दस्तावेज़ीकरण :
विशेषता कक्षा आकार डीटाइप विवरण
विशेषताएं डिक्ट
इसे स्वीकार करो टेन्सर बूल
20_प्रतिशत_ट्रेन टेन्सर बूल
20_प्रतिशत_वैध टेन्सर बूल
50 प्रतिशत टेन्सर बूल
50_प्रतिशत_ट्रेन टेन्सर बूल
50_प्रतिशत_वैध टेन्सर बूल
प्रकरण_आईडी टेन्सर डोरी
क्षितिज टेन्सर int32
कदम डेटासेट
कदम / कार्रवाई टेन्सर (7,) फ्लोट64
चरण/छूट टेन्सर int32
चरण/is_first टेन्सर बूल
चरण/is_last टेन्सर बूल
कदम/is_terminal टेन्सर बूल
चरण/अवलोकन विशेषताएं डिक्ट
कदम/अवलोकन/agentview_image छवि (84, 84, 3) uint8
चरण/अवलोकन/ऑब्जेक्ट टेन्सर (14,) फ्लोट64
चरण/अवलोकन/robot0_eef_pos टेन्सर (3,) फ्लोट64 अंत-प्रभावक स्थिति
चरण/अवलोकन/robot0_eef_quat टेन्सर (4,) फ्लोट64 अंत-प्रभावक अभिविन्यास
चरण/अवलोकन/robot0_eef_vel_ang टेन्सर (3,) फ्लोट64 अंत-प्रभावक कोणीय वेग
चरण/अवलोकन/robot0_eef_vel_lin टेन्सर (3,) फ्लोट64 अंत-प्रभावकार कार्टेशियन वेग
चरण/अवलोकन/robot0_eye_in_hand_image छवि (84, 84, 3) uint8
चरण/अवलोकन/robot0_gripper_qpos टेन्सर (2,) फ्लोट64 ग्रिपर की स्थिति
चरण/अवलोकन/robot0_gripper_qvel टेन्सर (2,) फ्लोट64 ग्रिपर वेग
चरण/अवलोकन/robot0_joint_pos टेन्सर (7,) फ्लोट64 7DOF संयुक्त स्थिति
चरण/अवलोकन/robot0_joint_pos_cos टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_pos_sin टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_vel टेन्सर (7,) फ्लोट64 7DOF संयुक्त वेग
कदम / इनाम टेन्सर फ्लोट64
चरण/राज्य टेन्सर (71,) फ्लोट64
रेलगाड़ी टेन्सर बूल
वैध टेन्सर बूल

Robomimic_ph/can_ph_low_dim

  • डाउनलोड आकार : 43.38 MiB

  • डेटासेट का आकार : 27.73 MiB

  • ऑटो-कैश्ड ( दस्तावेज़ीकरण ): हाँ

  • फ़ीचर संरचना :

FeaturesDict({
    '20_percent': bool,
    '20_percent_train': bool,
    '20_percent_valid': bool,
    '50_percent': bool,
    '50_percent_train': bool,
    '50_percent_valid': bool,
    'episode_id': string,
    'horizon': int32,
    'steps': Dataset({
        'action': Tensor(shape=(7,), dtype=float64),
        'discount': int32,
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': FeaturesDict({
            'object': Tensor(shape=(14,), dtype=float64),
            'robot0_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot0_eef_vel_ang': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_vel_lin': Tensor(shape=(3,), dtype=float64),
            'robot0_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot0_gripper_qvel': Tensor(shape=(2,), dtype=float64),
            'robot0_joint_pos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_cos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_sin': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_vel': Tensor(shape=(7,), dtype=float64),
        }),
        'reward': float64,
        'states': Tensor(shape=(71,), dtype=float64),
    }),
    'train': bool,
    'valid': bool,
})
  • फ़ीचर दस्तावेज़ीकरण :
विशेषता कक्षा आकार डीटाइप विवरण
विशेषताएं डिक्ट
इसे स्वीकार करो टेन्सर बूल
20_प्रतिशत_ट्रेन टेन्सर बूल
20_प्रतिशत_वैध टेन्सर बूल
50 प्रतिशत टेन्सर बूल
50_प्रतिशत_ट्रेन टेन्सर बूल
50_प्रतिशत_वैध टेन्सर बूल
प्रकरण_आईडी टेन्सर डोरी
क्षितिज टेन्सर int32
कदम डेटासेट
कदम / कार्रवाई टेन्सर (7,) फ्लोट64
चरण/छूट टेन्सर int32
चरण/is_first टेन्सर बूल
चरण/is_last टेन्सर बूल
कदम/is_terminal टेन्सर बूल
चरण/अवलोकन विशेषताएं डिक्ट
चरण/अवलोकन/ऑब्जेक्ट टेन्सर (14,) फ्लोट64
चरण/अवलोकन/robot0_eef_pos टेन्सर (3,) फ्लोट64 अंत-प्रभावक स्थिति
चरण/अवलोकन/robot0_eef_quat टेन्सर (4,) फ्लोट64 अंत-प्रभावक अभिविन्यास
चरण/अवलोकन/robot0_eef_vel_ang टेन्सर (3,) फ्लोट64 अंत-प्रभावक कोणीय वेग
चरण/अवलोकन/robot0_eef_vel_lin टेन्सर (3,) फ्लोट64 अंत-प्रभावकार कार्टेशियन वेग
चरण/अवलोकन/robot0_gripper_qpos टेन्सर (2,) फ्लोट64 ग्रिपर की स्थिति
चरण/अवलोकन/robot0_gripper_qvel टेन्सर (2,) फ्लोट64 ग्रिपर वेग
चरण/अवलोकन/robot0_joint_pos टेन्सर (7,) फ्लोट64 7DOF संयुक्त स्थिति
चरण/अवलोकन/robot0_joint_pos_cos टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_pos_sin टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_vel टेन्सर (7,) फ्लोट64 7DOF संयुक्त वेग
कदम / इनाम टेन्सर फ्लोट64
चरण/राज्य टेन्सर (71,) फ्लोट64
रेलगाड़ी टेन्सर बूल
वैध टेन्सर बूल

Robomimic_ph/square_ph_image

  • डाउनलोड आकार : 2.42 GiB

  • डेटासेट का आकार : 401.28 MiB

  • ऑटो-कैश्ड ( दस्तावेज़ीकरण ): नहीं

  • फ़ीचर संरचना :

FeaturesDict({
    '20_percent': bool,
    '20_percent_train': bool,
    '20_percent_valid': bool,
    '50_percent': bool,
    '50_percent_train': bool,
    '50_percent_valid': bool,
    'episode_id': string,
    'horizon': int32,
    'steps': Dataset({
        'action': Tensor(shape=(7,), dtype=float64),
        'discount': int32,
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': FeaturesDict({
            'agentview_image': Image(shape=(84, 84, 3), dtype=uint8),
            'object': Tensor(shape=(14,), dtype=float64),
            'robot0_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot0_eef_vel_ang': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_vel_lin': Tensor(shape=(3,), dtype=float64),
            'robot0_eye_in_hand_image': Image(shape=(84, 84, 3), dtype=uint8),
            'robot0_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot0_gripper_qvel': Tensor(shape=(2,), dtype=float64),
            'robot0_joint_pos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_cos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_sin': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_vel': Tensor(shape=(7,), dtype=float64),
        }),
        'reward': float64,
        'states': Tensor(shape=(45,), dtype=float64),
    }),
    'train': bool,
    'valid': bool,
})
  • फ़ीचर दस्तावेज़ीकरण :
विशेषता कक्षा आकार डीटाइप विवरण
विशेषताएं डिक्ट
इसे स्वीकार करो टेन्सर बूल
20_प्रतिशत_ट्रेन टेन्सर बूल
20_प्रतिशत_वैध टेन्सर बूल
50 प्रतिशत टेन्सर बूल
50_प्रतिशत_ट्रेन टेन्सर बूल
50_प्रतिशत_वैध टेन्सर बूल
प्रकरण_आईडी टेन्सर डोरी
क्षितिज टेन्सर int32
कदम डेटासेट
कदम / कार्रवाई टेन्सर (7,) फ्लोट64
चरण/छूट टेन्सर int32
चरण/is_first टेन्सर बूल
चरण/is_last टेन्सर बूल
कदम/is_terminal टेन्सर बूल
चरण/अवलोकन विशेषताएं डिक्ट
कदम/अवलोकन/agentview_image छवि (84, 84, 3) uint8
चरण/अवलोकन/ऑब्जेक्ट टेन्सर (14,) फ्लोट64
चरण/अवलोकन/robot0_eef_pos टेन्सर (3,) फ्लोट64 अंत-प्रभावक स्थिति
चरण/अवलोकन/robot0_eef_quat टेन्सर (4,) फ्लोट64 अंत-प्रभावक अभिविन्यास
चरण/अवलोकन/robot0_eef_vel_ang टेन्सर (3,) फ्लोट64 अंत-प्रभावक कोणीय वेग
चरण/अवलोकन/robot0_eef_vel_lin टेन्सर (3,) फ्लोट64 अंत-प्रभावकार कार्टेशियन वेग
चरण/अवलोकन/robot0_eye_in_hand_image छवि (84, 84, 3) uint8
चरण/अवलोकन/robot0_gripper_qpos टेन्सर (2,) फ्लोट64 ग्रिपर की स्थिति
चरण/अवलोकन/robot0_gripper_qvel टेन्सर (2,) फ्लोट64 ग्रिपर वेग
चरण/अवलोकन/robot0_joint_pos टेन्सर (7,) फ्लोट64 7DOF संयुक्त स्थिति
चरण/अवलोकन/robot0_joint_pos_cos टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_pos_sin टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_vel टेन्सर (7,) फ्लोट64 7DOF संयुक्त वेग
कदम / इनाम टेन्सर फ्लोट64
चरण/राज्य टेन्सर (45,) फ्लोट64
रेलगाड़ी टेन्सर बूल
वैध टेन्सर बूल

Robomimic_ph/square_ph_low_dim

  • डाउनलोड आकार : 47.69 MiB

  • डेटासेट का आकार : 29.91 MiB

  • ऑटो-कैश्ड ( दस्तावेज़ीकरण ): हाँ

  • फ़ीचर संरचना :

FeaturesDict({
    '20_percent': bool,
    '20_percent_train': bool,
    '20_percent_valid': bool,
    '50_percent': bool,
    '50_percent_train': bool,
    '50_percent_valid': bool,
    'episode_id': string,
    'horizon': int32,
    'steps': Dataset({
        'action': Tensor(shape=(7,), dtype=float64),
        'discount': int32,
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': FeaturesDict({
            'object': Tensor(shape=(14,), dtype=float64),
            'robot0_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot0_eef_vel_ang': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_vel_lin': Tensor(shape=(3,), dtype=float64),
            'robot0_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot0_gripper_qvel': Tensor(shape=(2,), dtype=float64),
            'robot0_joint_pos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_cos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_sin': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_vel': Tensor(shape=(7,), dtype=float64),
        }),
        'reward': float64,
        'states': Tensor(shape=(45,), dtype=float64),
    }),
    'train': bool,
    'valid': bool,
})
  • फ़ीचर दस्तावेज़ीकरण :
विशेषता कक्षा आकार डीटाइप विवरण
विशेषताएं डिक्ट
इसे स्वीकार करो टेन्सर बूल
20_प्रतिशत_ट्रेन टेन्सर बूल
20_प्रतिशत_वैध टेन्सर बूल
50 प्रतिशत टेन्सर बूल
50_प्रतिशत_ट्रेन टेन्सर बूल
50_प्रतिशत_वैध टेन्सर बूल
प्रकरण_आईडी टेन्सर डोरी
क्षितिज टेन्सर int32
कदम डेटासेट
कदम / कार्रवाई टेन्सर (7,) फ्लोट64
चरण/छूट टेन्सर int32
चरण/is_first टेन्सर बूल
चरण/is_last टेन्सर बूल
कदम/is_terminal टेन्सर बूल
चरण/अवलोकन विशेषताएं डिक्ट
चरण/अवलोकन/ऑब्जेक्ट टेन्सर (14,) फ्लोट64
चरण/अवलोकन/robot0_eef_pos टेन्सर (3,) फ्लोट64 अंत-प्रभावक स्थिति
चरण/अवलोकन/robot0_eef_quat टेन्सर (4,) फ्लोट64 अंत-प्रभावक अभिविन्यास
चरण/अवलोकन/robot0_eef_vel_ang टेन्सर (3,) फ्लोट64 अंत-प्रभावक कोणीय वेग
चरण/अवलोकन/robot0_eef_vel_lin टेन्सर (3,) फ्लोट64 अंत-प्रभावकार कार्टेशियन वेग
चरण/अवलोकन/robot0_gripper_qpos टेन्सर (2,) फ्लोट64 ग्रिपर की स्थिति
चरण/अवलोकन/robot0_gripper_qvel टेन्सर (2,) फ्लोट64 ग्रिपर वेग
चरण/अवलोकन/robot0_joint_pos टेन्सर (7,) फ्लोट64 7DOF संयुक्त स्थिति
चरण/अवलोकन/robot0_joint_pos_cos टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_pos_sin टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_vel टेन्सर (7,) फ्लोट64 7DOF संयुक्त वेग
कदम / इनाम टेन्सर फ्लोट64
चरण/राज्य टेन्सर (45,) फ्लोट64
रेलगाड़ी टेन्सर बूल
वैध टेन्सर बूल

Robomimic_ph/transport_ph_image

  • डाउनलोड का आकार : 15.07 GiB

  • डेटासेट का आकार : 3.64 GiB

  • ऑटो-कैश्ड ( दस्तावेज़ीकरण ): नहीं

  • फ़ीचर संरचना :

FeaturesDict({
    '20_percent': bool,
    '20_percent_train': bool,
    '20_percent_valid': bool,
    '50_percent': bool,
    '50_percent_train': bool,
    '50_percent_valid': bool,
    'episode_id': string,
    'horizon': int32,
    'steps': Dataset({
        'action': Tensor(shape=(14,), dtype=float64),
        'discount': int32,
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': FeaturesDict({
            'object': Tensor(shape=(41,), dtype=float64),
            'robot0_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot0_eef_vel_ang': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_vel_lin': Tensor(shape=(3,), dtype=float64),
            'robot0_eye_in_hand_image': Image(shape=(84, 84, 3), dtype=uint8),
            'robot0_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot0_gripper_qvel': Tensor(shape=(2,), dtype=float64),
            'robot0_joint_pos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_cos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_sin': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_vel': Tensor(shape=(7,), dtype=float64),
            'robot1_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot1_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot1_eef_vel_ang': Tensor(shape=(3,), dtype=float64),
            'robot1_eef_vel_lin': Tensor(shape=(3,), dtype=float64),
            'robot1_eye_in_hand_image': Image(shape=(84, 84, 3), dtype=uint8),
            'robot1_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot1_gripper_qvel': Tensor(shape=(2,), dtype=float64),
            'robot1_joint_pos': Tensor(shape=(7,), dtype=float64),
            'robot1_joint_pos_cos': Tensor(shape=(7,), dtype=float64),
            'robot1_joint_pos_sin': Tensor(shape=(7,), dtype=float64),
            'robot1_joint_vel': Tensor(shape=(7,), dtype=float64),
            'shouldercamera0_image': Image(shape=(84, 84, 3), dtype=uint8),
            'shouldercamera1_image': Image(shape=(84, 84, 3), dtype=uint8),
        }),
        'reward': float64,
        'states': Tensor(shape=(115,), dtype=float64),
    }),
    'train': bool,
    'valid': bool,
})
  • फ़ीचर दस्तावेज़ीकरण :
विशेषता कक्षा आकार डीटाइप विवरण
विशेषताएं डिक्ट
इसे स्वीकार करो टेन्सर बूल
20_प्रतिशत_ट्रेन टेन्सर बूल
20_प्रतिशत_वैध टेन्सर बूल
50 प्रतिशत टेन्सर बूल
50_प्रतिशत_ट्रेन टेन्सर बूल
50_प्रतिशत_वैध टेन्सर बूल
प्रकरण_आईडी टेन्सर डोरी
क्षितिज टेन्सर int32
कदम डेटासेट
कदम / कार्रवाई टेन्सर (14,) फ्लोट64
चरण/छूट टेन्सर int32
चरण/is_first टेन्सर बूल
चरण/is_last टेन्सर बूल
कदम/is_terminal टेन्सर बूल
चरण/अवलोकन विशेषताएं डिक्ट
चरण/अवलोकन/ऑब्जेक्ट टेन्सर (41,) फ्लोट64
चरण/अवलोकन/robot0_eef_pos टेन्सर (3,) फ्लोट64 अंत-प्रभावक स्थिति
चरण/अवलोकन/robot0_eef_quat टेन्सर (4,) फ्लोट64 अंत-प्रभावक अभिविन्यास
चरण/अवलोकन/robot0_eef_vel_ang टेन्सर (3,) फ्लोट64 अंत-प्रभावक कोणीय वेग
चरण/अवलोकन/robot0_eef_vel_lin टेन्सर (3,) फ्लोट64 अंत-प्रभावकार कार्टेशियन वेग
चरण/अवलोकन/robot0_eye_in_hand_image छवि (84, 84, 3) uint8
चरण/अवलोकन/robot0_gripper_qpos टेन्सर (2,) फ्लोट64 ग्रिपर की स्थिति
चरण/अवलोकन/robot0_gripper_qvel टेन्सर (2,) फ्लोट64 ग्रिपर वेग
चरण/अवलोकन/robot0_joint_pos टेन्सर (7,) फ्लोट64 7DOF संयुक्त स्थिति
चरण/अवलोकन/robot0_joint_pos_cos टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_pos_sin टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_vel टेन्सर (7,) फ्लोट64 7DOF संयुक्त वेग
चरण/अवलोकन/robot1_eef_pos टेन्सर (3,) फ्लोट64 अंत-प्रभावक स्थिति
चरण/अवलोकन/robot1_eef_quat टेन्सर (4,) फ्लोट64 अंत-प्रभावक अभिविन्यास
चरण/अवलोकन/robot1_eef_vel_ang टेन्सर (3,) फ्लोट64 अंत-प्रभावक कोणीय वेग
चरण/अवलोकन/robot1_eef_vel_lin टेन्सर (3,) फ्लोट64 अंत-प्रभावकार कार्टेशियन वेग
चरण/अवलोकन/robot1_eye_in_hand_image छवि (84, 84, 3) uint8
चरण/अवलोकन/robot1_gripper_qpos टेन्सर (2,) फ्लोट64 ग्रिपर की स्थिति
चरण/अवलोकन/robot1_gripper_qvel टेन्सर (2,) फ्लोट64 ग्रिपर वेग
चरण/अवलोकन/robot1_joint_pos टेन्सर (7,) फ्लोट64 7DOF संयुक्त स्थिति
चरण/अवलोकन/robot1_joint_pos_cos टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot1_joint_pos_sin टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot1_joint_vel टेन्सर (7,) फ्लोट64 7DOF संयुक्त वेग
स्टेप्स/ऑब्जर्वेशन/शोल्डरकैमरा0_इमेज छवि (84, 84, 3) uint8
स्टेप्स/ऑब्जर्वेशन/शोल्डरकैमरा1_इमेज छवि (84, 84, 3) uint8
कदम / इनाम टेन्सर फ्लोट64
चरण/राज्य टेन्सर (115,) फ्लोट64
रेलगाड़ी टेन्सर बूल
वैध टेन्सर बूल

Robomimic_ph/transport_ph_low_dim

  • डाउनलोड आकार : 294.70 MiB

  • डेटासेट का आकार : 208.05 MiB

  • ऑटो-कैश्ड ( दस्तावेज़ीकरण ): केवल जब shuffle_files=False (ट्रेन)

  • फ़ीचर संरचना :

FeaturesDict({
    '20_percent': bool,
    '20_percent_train': bool,
    '20_percent_valid': bool,
    '50_percent': bool,
    '50_percent_train': bool,
    '50_percent_valid': bool,
    'episode_id': string,
    'horizon': int32,
    'steps': Dataset({
        'action': Tensor(shape=(14,), dtype=float64),
        'discount': int32,
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': FeaturesDict({
            'object': Tensor(shape=(41,), dtype=float64),
            'robot0_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot0_eef_vel_ang': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_vel_lin': Tensor(shape=(3,), dtype=float64),
            'robot0_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot0_gripper_qvel': Tensor(shape=(2,), dtype=float64),
            'robot0_joint_pos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_cos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_sin': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_vel': Tensor(shape=(7,), dtype=float64),
            'robot1_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot1_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot1_eef_vel_ang': Tensor(shape=(3,), dtype=float64),
            'robot1_eef_vel_lin': Tensor(shape=(3,), dtype=float64),
            'robot1_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot1_gripper_qvel': Tensor(shape=(2,), dtype=float64),
            'robot1_joint_pos': Tensor(shape=(7,), dtype=float64),
            'robot1_joint_pos_cos': Tensor(shape=(7,), dtype=float64),
            'robot1_joint_pos_sin': Tensor(shape=(7,), dtype=float64),
            'robot1_joint_vel': Tensor(shape=(7,), dtype=float64),
        }),
        'reward': float64,
        'states': Tensor(shape=(115,), dtype=float64),
    }),
    'train': bool,
    'valid': bool,
})
  • फ़ीचर दस्तावेज़ीकरण :
विशेषता कक्षा आकार डीटाइप विवरण
विशेषताएं डिक्ट
इसे स्वीकार करो टेन्सर बूल
20_प्रतिशत_ट्रेन टेन्सर बूल
20_प्रतिशत_वैध टेन्सर बूल
50 प्रतिशत टेन्सर बूल
50_प्रतिशत_ट्रेन टेन्सर बूल
50_प्रतिशत_वैध टेन्सर बूल
प्रकरण_आईडी टेन्सर डोरी
क्षितिज टेन्सर int32
कदम डेटासेट
कदम / कार्रवाई टेन्सर (14,) फ्लोट64
चरण/छूट टेन्सर int32
चरण/is_first टेन्सर बूल
चरण/is_last टेन्सर बूल
कदम/is_terminal टेन्सर बूल
चरण/अवलोकन विशेषताएं डिक्ट
चरण/अवलोकन/ऑब्जेक्ट टेन्सर (41,) फ्लोट64
चरण/अवलोकन/robot0_eef_pos टेन्सर (3,) फ्लोट64 अंत-प्रभावक स्थिति
चरण/अवलोकन/robot0_eef_quat टेन्सर (4,) फ्लोट64 अंत-प्रभावक अभिविन्यास
चरण/अवलोकन/robot0_eef_vel_ang टेन्सर (3,) फ्लोट64 अंत-प्रभावक कोणीय वेग
चरण/अवलोकन/robot0_eef_vel_lin टेन्सर (3,) फ्लोट64 अंत-प्रभावकार कार्टेशियन वेग
चरण/अवलोकन/robot0_gripper_qpos टेन्सर (2,) फ्लोट64 ग्रिपर की स्थिति
चरण/अवलोकन/robot0_gripper_qvel टेन्सर (2,) फ्लोट64 ग्रिपर वेग
चरण/अवलोकन/robot0_joint_pos टेन्सर (7,) फ्लोट64 7DOF संयुक्त स्थिति
चरण/अवलोकन/robot0_joint_pos_cos टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_pos_sin टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot0_joint_vel टेन्सर (7,) फ्लोट64 7DOF संयुक्त वेग
चरण/अवलोकन/robot1_eef_pos टेन्सर (3,) फ्लोट64 अंत-प्रभावक स्थिति
चरण/अवलोकन/robot1_eef_quat टेन्सर (4,) फ्लोट64 अंत-प्रभावक अभिविन्यास
चरण/अवलोकन/robot1_eef_vel_ang टेन्सर (3,) फ्लोट64 अंत-प्रभावक कोणीय वेग
चरण/अवलोकन/robot1_eef_vel_lin टेन्सर (3,) फ्लोट64 अंत-प्रभावकार कार्टेशियन वेग
चरण/अवलोकन/robot1_gripper_qpos टेन्सर (2,) फ्लोट64 ग्रिपर की स्थिति
चरण/अवलोकन/robot1_gripper_qvel टेन्सर (2,) फ्लोट64 ग्रिपर वेग
चरण/अवलोकन/robot1_joint_pos टेन्सर (7,) फ्लोट64 7DOF संयुक्त स्थिति
चरण/अवलोकन/robot1_joint_pos_cos टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot1_joint_pos_sin टेन्सर (7,) फ्लोट64
चरण/अवलोकन/robot1_joint_vel टेन्सर (7,) फ्लोट64 7DOF संयुक्त वेग
कदम / इनाम टेन्सर फ्लोट64
चरण/राज्य टेन्सर (115,) फ्लोट64
रेलगाड़ी टेन्सर बूल
वैध टेन्सर बूल

Robomimic_ph/tool_hang_ph_image

  • डाउनलोड का आकार : 61.96 GiB

  • डेटासेट का आकार : 9.10 GiB

  • ऑटो-कैश्ड ( दस्तावेज़ीकरण ): नहीं

  • फ़ीचर संरचना :

FeaturesDict({
    'episode_id': string,
    'horizon': int32,
    'steps': Dataset({
        'action': Tensor(shape=(7,), dtype=float64),
        'discount': int32,
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': FeaturesDict({
            'object': Tensor(shape=(44,), dtype=float64),
            'robot0_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot0_eef_vel_ang': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_vel_lin': Tensor(shape=(3,), dtype=float64),
            'robot0_eye_in_hand_image': Image(shape=(240, 240, 3), dtype=uint8),
            'robot0_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot0_gripper_qvel': Tensor(shape=(2,), dtype=float64),
            'robot0_joint_pos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_cos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_sin': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_vel': Tensor(shape=(7,), dtype=float64),
            'sideview_image': Image(shape=(240, 240, 3), dtype=uint8),
        }),
        'reward': float64,
        'states': Tensor(shape=(58,), dtype=float64),
    }),
    'train': bool,
    'valid': bool,
})
  • फ़ीचर दस्तावेज़ीकरण :
विशेषता कक्षा आकार डीटाइप विवरण
विशेषताएं डिक्ट
प्रकरण_आईडी टेन्सर डोरी
क्षितिज टेन्सर int32
कदम डेटासेट
कदम / कार्रवाई टेन्सर (7,) फ्लोट64
चरण/छूट टेन्सर int32
चरण/is_first टेन्सर बूल
चरण/is_last Tensor bool
steps/is_terminal Tensor bool
steps/observation FeaturesDict
steps/observation/object Tensor (44,) float64
steps/observation/robot0_eef_pos Tensor (3,) float64 End-effector position
steps/observation/robot0_eef_quat Tensor (4,) float64 End-effector orientation
steps/observation/robot0_eef_vel_ang Tensor (3,) float64 End-effector angular velocity
steps/observation/robot0_eef_vel_lin Tensor (3,) float64 End-effector cartesian velocity
steps/observation/robot0_eye_in_hand_image Image (240, 240, 3) uint8
steps/observation/robot0_gripper_qpos Tensor (2,) float64 Gripper position
steps/observation/robot0_gripper_qvel Tensor (2,) float64 Gripper velocity
steps/observation/robot0_joint_pos Tensor (7,) float64 7DOF joint positions
steps/observation/robot0_joint_pos_cos Tensor (7,) float64
steps/observation/robot0_joint_pos_sin Tensor (7,) float64
steps/observation/robot0_joint_vel Tensor (7,) float64 7DOF joint velocities
steps/observation/sideview_image Image (240, 240, 3) uint8
steps/reward Tensor float64
steps/states Tensor (58,) float64
train Tensor bool
valid Tensor bool

robomimic_ph/tool_hang_ph_low_dim

  • Download size : 192.29 MiB

  • Dataset size : 121.77 MiB

  • Auto-cached ( documentation ): Yes

  • Feature structure :

FeaturesDict({
    'episode_id': string,
    'horizon': int32,
    'steps': Dataset({
        'action': Tensor(shape=(7,), dtype=float64),
        'discount': int32,
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': FeaturesDict({
            'object': Tensor(shape=(44,), dtype=float64),
            'robot0_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot0_eef_vel_ang': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_vel_lin': Tensor(shape=(3,), dtype=float64),
            'robot0_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot0_gripper_qvel': Tensor(shape=(2,), dtype=float64),
            'robot0_joint_pos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_cos': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_pos_sin': Tensor(shape=(7,), dtype=float64),
            'robot0_joint_vel': Tensor(shape=(7,), dtype=float64),
        }),
        'reward': float64,
        'states': Tensor(shape=(58,), dtype=float64),
    }),
    'train': bool,
    'valid': bool,
})
  • Feature documentation :
Feature Class Shape Dtype Description
FeaturesDict
episode_id Tensor string
horizon Tensor int32
steps Dataset
steps/action Tensor (7,) float64
steps/discount Tensor int32
steps/is_first Tensor bool
steps/is_last Tensor bool
steps/is_terminal Tensor bool
steps/observation FeaturesDict
steps/observation/object Tensor (44,) float64
steps/observation/robot0_eef_pos Tensor (3,) float64 End-effector position
steps/observation/robot0_eef_quat Tensor (4,) float64 End-effector orientation
steps/observation/robot0_eef_vel_ang Tensor (3,) float64 End-effector angular velocity
steps/observation/robot0_eef_vel_lin Tensor (3,) float64 End-effector cartesian velocity
steps/observation/robot0_gripper_qpos Tensor (2,) float64 Gripper position
steps/observation/robot0_gripper_qvel Tensor (2,) float64 Gripper velocity
steps/observation/robot0_joint_pos Tensor (7,) float64 7DOF joint positions
steps/observation/robot0_joint_pos_cos Tensor (7,) float64
steps/observation/robot0_joint_pos_sin Tensor (7,) float64
steps/observation/robot0_joint_vel Tensor (7,) float64 7DOF joint velocities
steps/reward Tensor float64
steps/states Tensor (58,) float64
train Tensor bool
valid Tensor bool