plex_robosuite

  • विवरण :

रोबोसुइट्स डोर, स्टैक, पिकप्लेसमिल्क, पिकप्लेसब्रेड, पिकप्लेससेरियल और नटअसेंबलीराउंड कार्यों के लिए उच्च गुणवत्ता वाले प्रदर्शन प्रक्षेप पथ का एक डेटासेट, प्रत्येक के लिए 75 प्रदर्शन।

विभाजित करना उदाहरण
'train' 402
'val' 48
  • फ़ीचर संरचना :
FeaturesDict({
    'episode_metadata': FeaturesDict({
        'episode_id': string,
    }),
    'steps': Dataset({
        'action': Tensor(shape=(7,), dtype=float64),
        'discount': Scalar(shape=(), dtype=float32),
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'language_embedding': Tensor(shape=(512,), dtype=float32),
        'language_instruction': string,
        'observation': FeaturesDict({
            'image': Image(shape=(128, 128, 3), dtype=uint8),
            'state': Tensor(shape=(32,), dtype=float64),
            'wrist_image': Image(shape=(128, 128, 3), dtype=uint8),
        }),
        'reward': Scalar(shape=(), dtype=float32),
    }),
})
  • फ़ीचर दस्तावेज़ीकरण :
विशेषता कक्षा आकार डीप्रकार विवरण
फीचर्सडिक्ट
एपिसोड_मेटाडेटा फीचर्सडिक्ट
एपिसोड_मेटाडेटा/एपिसोड_आईडी टेन्सर डोरी
कदम डेटासेट
कदम/कार्रवाई टेन्सर (7,) फ्लोट64
कदम/छूट अदिश फ्लोट32
चरण/पहला है टेन्सर बूल
चरण/अंतिम है टेन्सर बूल
चरण/is_terminal टेन्सर बूल
चरण/भाषा_एम्बेडिंग टेन्सर (512,) फ्लोट32
चरण/भाषा_निर्देश टेन्सर डोरी
चरण/अवलोकन फीचर्सडिक्ट
चरण/अवलोकन/छवि छवि (128, 128, 3) uint8
चरण/अवलोकन/स्थिति टेन्सर (32,) फ्लोट64
चरण/अवलोकन/कलाई_छवि छवि (128, 128, 3) uint8
कदम/इनाम अदिश फ्लोट32
  • उद्धरण :
https://doi.org/10.48550/arXiv.2303.08789