robo_set

  • شرح :

مجموعه داده واقعی یک بازوی روباتی که 12 مهارت دستکاری غیر پیش پا افتاده را در 38 کار، 7500 مسیر نشان می دهد.

شکاف مثال ها
'train' 18250
  • ساختار ویژگی :
FeaturesDict({
    'episode_metadata': FeaturesDict({
        'file_path': string,
        'trial_id': string,
    }),
    'steps': Dataset({
        'action': Tensor(shape=(8,), dtype=float32),
        'discount': Scalar(shape=(), dtype=float32),
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'language_instruction': string,
        'observation': FeaturesDict({
            'image_left': Image(shape=(240, 424, 3), dtype=uint8),
            'image_right': Image(shape=(240, 424, 3), dtype=uint8),
            'image_top': Image(shape=(240, 424, 3), dtype=uint8),
            'image_wrist': Image(shape=(240, 424, 3), dtype=uint8),
            'state': Tensor(shape=(8,), dtype=float32),
            'state_velocity': Tensor(shape=(8,), dtype=float32),
        }),
        'reward': Scalar(shape=(), dtype=float32),
    }),
})
  • مستندات ویژگی :
ویژگی کلاس شکل نوع D شرح
FeaturesDict
episode_metadata FeaturesDict
episode_metadata/file_path تانسور رشته
episode_metadata/trial_id تانسور رشته
مراحل مجموعه داده
مراحل/عمل تانسور (8،) float32
مراحل/تخفیف اسکالر float32
Steps/is_first تانسور بوول
Steps/is_last تانسور بوول
Steps/is_terminal تانسور بوول
مراحل/زبان_آموزش تانسور رشته
مراحل / مشاهده FeaturesDict
steps/observation/image_left تصویر (240، 424، 3) uint8
مراحل/مشاهده/تصویر_راست تصویر (240، 424، 3) uint8
steps/observation/image_top تصویر (240، 424، 3) uint8
steps/observation/image_wrist تصویر (240، 424، 3) uint8
مراحل / مشاهده / حالت تانسور (8،) float32
مراحل/مشاهده/سرعت_حالت تانسور (8،) float32
مراحل/پاداش اسکالر float32
@misc{bharadhwaj2023roboagent, title={RoboAgent: Generalization and Efficiency in Robot Manipulation via Semantic Augmentations and Action Chunking}, author={Homanga Bharadhwaj and Jay Vakil and Mohit Sharma and Abhinav Gupta and Shubham Tulsiani and Vikash Kumar},  year={2023}, eprint={2309.01918}, archivePrefix={arXiv}, primaryClass={cs.RO} }