robosuite_panda_pick_place_can

  • বর্ণনা :

এই ডেটাসেটগুলি রোবোসুইট রোবোটিক আর্ম সিমুলেটরের পিকপ্লেসক্যান পরিবেশের সাথে তৈরি করা হয়েছে। মানব ডেটাসেটগুলি RLDS ক্রিয়েটর এবং একটি গেমপ্যাড কন্ট্রোলার ব্যবহার করে একটি একক অপারেটর দ্বারা রেকর্ড করা হয়েছিল।

সিনথেটিক ডেটাসেটগুলি EnvLogger লাইব্রেরি ব্যবহার করে রেকর্ড করা হয়েছে।

ডেটাসেটগুলি ধাপ এবং পর্বগুলি উপস্থাপন করতে RLDS বিন্যাস অনুসরণ করে।

পর্ব 400টি ধাপ নিয়ে গঠিত। প্রতিটি পর্বে, একটি ট্যাগ যোগ করা হয় যখন কাজটি সম্পন্ন হয়, এই ট্যাগটি কাস্টম ধাপের মেটাডেটার অংশ হিসাবে সংরক্ষণ করা হয়।

মনে রাখবেন, EnvLogger নির্ভরতার কারণে, এই ডেটাসেট তৈরি করা বর্তমানে শুধুমাত্র Linux পরিবেশে সমর্থিত।

@misc{ramos2021rlds,
      title={RLDS: an Ecosystem to Generate, Share and Use Datasets in Reinforcement Learning},
      author={Sabela Ramos and Sertan Girgin and Léonard Hussenot and Damien Vincent and Hanna Yakubovich and Daniel Toyama and Anita Gergely and Piotr Stanczyk and Raphael Marinier and Jeremiah Harmsen and Olivier Pietquin and Nikola Momchev},
      year={2021},
      eprint={2111.02767},
      archivePrefix={arXiv},
      primaryClass={cs.LG}
}

robosuite_panda_pick_place_can/human_dc29b40a (ডিফল্ট কনফিগারেশন)

  • কনফিগারেশনের বিবরণ : মানব উৎপন্ন ডেটাসেট (50 পর্ব)।

  • হোমপেজ : https://github.com/google-research/rlds

  • ডাউনলোড সাইজ : 96.67 MiB

  • ডেটাসেটের আকার : 407.24 MiB

  • বিভাজন :

বিভক্ত উদাহরণ
'train' 50
  • বৈশিষ্ট্য গঠন :
FeaturesDict({
    'agent_id': string,
    'episode_id': string,
    'episode_index': int32,
    'steps': Dataset({
        'action': Tensor(shape=(7,), dtype=float64),
        'discount': float64,
        'image': Image(shape=(None, None, 3), dtype=uint8),
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': FeaturesDict({
            'Can_pos': Tensor(shape=(3,), dtype=float64),
            'Can_quat': Tensor(shape=(4,), dtype=float64),
            'Can_to_robot0_eef_pos': Tensor(shape=(3,), dtype=float64),
            'Can_to_robot0_eef_quat': Tensor(shape=(4,), dtype=float32),
            'object-state': Tensor(shape=(14,), dtype=float64),
            'robot0_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot0_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot0_gripper_qvel': Tensor(shape=(2,), 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),
            'robot0_proprio-state': Tensor(shape=(32,), dtype=float64),
        }),
        'reward': float64,
        'tag:placed': bool,
    }),
})
  • বৈশিষ্ট্য ডকুমেন্টেশন :
বৈশিষ্ট্য ক্লাস আকৃতি ডিটাইপ বর্ণনা
ফিচারসডিক্ট
agent_id টেনসর স্ট্রিং
episode_id টেনসর স্ট্রিং
episode_index টেনসর int32
পদক্ষেপ ডেটাসেট
পদক্ষেপ/ক্রিয়া টেনসর (৭,) float64
পদক্ষেপ/ছাড় টেনসর float64
পদক্ষেপ/চিত্র ছবি (কোনটিই নয়, 3) uint8
steps/is_first টেনসর bool
ধাপ/শেষ_শেষ টেনসর bool
steps/is_terminal টেনসর bool
পদক্ষেপ/পর্যবেক্ষণ ফিচারসডিক্ট
পদক্ষেপ/পর্যবেক্ষণ/ক্যান_পোস টেনসর (৩,) float64
পদক্ষেপ/পর্যবেক্ষণ/ক্যান_কোয়াট টেনসর (৪,) float64
পদক্ষেপ/পর্যবেক্ষণ/ক্যান_টু_রোবট0_eef_pos টেনসর (৩,) float64
পদক্ষেপ/পর্যবেক্ষণ/Can_to_robot0_eef_quat টেনসর (৪,) float32
পদক্ষেপ/পর্যবেক্ষণ/বস্তু-স্থিতি টেনসর (14,) float64
পদক্ষেপ/পর্যবেক্ষণ/robot0_eef_pos টেনসর (৩,) float64
পদক্ষেপ/পর্যবেক্ষণ/robot0_eef_quat টেনসর (৪,) float64
পদক্ষেপ/পর্যবেক্ষণ/robot0_gripper_qpos টেনসর (2,) float64
পদক্ষেপ/পর্যবেক্ষণ/robot0_gripper_qvel টেনসর (2,) float64
পদক্ষেপ/পর্যবেক্ষণ/robot0_joint_pos_cos টেনসর (৭,) float64
পদক্ষেপ/পর্যবেক্ষণ/robot0_joint_pos_sin টেনসর (৭,) float64
পদক্ষেপ/পর্যবেক্ষণ/robot0_joint_vel টেনসর (৭,) float64
পদক্ষেপ/পর্যবেক্ষণ/robot0_proprio-state টেনসর (৩২,) float64
পদক্ষেপ/পুরস্কার টেনসর float64
পদক্ষেপ/ট্যাগ: স্থাপন করা হয়েছে টেনসর bool

robosuite_panda_pick_place_can/human_images_dc29b40a

  • কনফিগারেশনের বিবরণ : পর্যবেক্ষণে বিভিন্ন ক্যামেরা অ্যাঙ্গেল সহ ছবি সহ মানব উৎপন্ন ডেটাসেট। মনে রাখবেন এটি তৈরি হতে কিছু সময় লাগতে পারে।

  • হোমপেজ : https://www.tensorflow.org/datasets/catalog/robosuite_panda_pick_place_can

  • ডাউনলোড সাইজ : 10.95 GiB

  • ডেটাসেটের আকার : 7.53 GiB

  • বিভাজন :

বিভক্ত উদাহরণ
'train' 50
  • বৈশিষ্ট্য গঠন :
FeaturesDict({
    'agent_id': string,
    'episode_id': string,
    'episode_index': int32,
    'steps': Dataset({
        'action': Tensor(shape=(7,), dtype=float64),
        'discount': float64,
        'image': Image(shape=(None, None, 3), dtype=uint8),
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': FeaturesDict({
            'Can_pos': Tensor(shape=(3,), dtype=float64),
            'Can_quat': Tensor(shape=(4,), dtype=float64),
            'Can_to_robot0_eef_pos': Tensor(shape=(3,), dtype=float64),
            'Can_to_robot0_eef_quat': Tensor(shape=(4,), dtype=float32),
            'agentview_image': Image(shape=(256, 256, 3), dtype=uint8),
            'birdview_image': Image(shape=(256, 256, 3), dtype=uint8),
            'object-state': Tensor(shape=(14,), dtype=float64),
            'robot0_eef_pos': Tensor(shape=(3,), dtype=float64),
            'robot0_eef_quat': Tensor(shape=(4,), dtype=float64),
            'robot0_eye_in_hand_image': Image(shape=(256, 256, 3), dtype=uint8),
            'robot0_gripper_qpos': Tensor(shape=(2,), dtype=float64),
            'robot0_gripper_qvel': Tensor(shape=(2,), 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),
            'robot0_proprio-state': Tensor(shape=(32,), dtype=float64),
            'robot0_robotview_image': Image(shape=(256, 256, 3), dtype=uint8),
        }),
        'reward': float64,
        'tag:placed': bool,
    }),
})
  • বৈশিষ্ট্য ডকুমেন্টেশন :
বৈশিষ্ট্য ক্লাস আকৃতি ডিটাইপ বর্ণনা
ফিচারসডিক্ট
agent_id টেনসর স্ট্রিং
episode_id টেনসর স্ট্রিং
episode_index টেনসর int32
পদক্ষেপ ডেটাসেট
পদক্ষেপ/ক্রিয়া টেনসর (৭,) float64
পদক্ষেপ/ছাড় টেনসর float64
পদক্ষেপ/চিত্র ছবি (কোনটিই নয়, 3) uint8
steps/is_first টেনসর bool
ধাপ/শেষ_শেষ টেনসর bool
steps/is_terminal টেনসর bool
পদক্ষেপ/পর্যবেক্ষণ ফিচারসডিক্ট
পদক্ষেপ/পর্যবেক্ষণ/ক্যান_পোস টেনসর (৩,) float64
পদক্ষেপ/পর্যবেক্ষণ/ক্যান_কোয়াট টেনসর (৪,) float64
পদক্ষেপ/পর্যবেক্ষণ/ক্যান_টু_রোবট0_eef_pos টেনসর (৩,) float64
পদক্ষেপ/পর্যবেক্ষণ/Can_to_robot0_eef_quat টেনসর (৪,) float32
পদক্ষেপ/পর্যবেক্ষণ/এজেন্টভিউ_ইমেজ ছবি (256, 256, 3) uint8
পদক্ষেপ/পর্যবেক্ষণ/পাখি দেখার_চিত্র ছবি (256, 256, 3) uint8
পদক্ষেপ/পর্যবেক্ষণ/বস্তু-স্থিতি টেনসর (14,) float64
পদক্ষেপ/পর্যবেক্ষণ/robot0_eef_pos টেনসর (৩,) float64
পদক্ষেপ/পর্যবেক্ষণ/robot0_eef_quat টেনসর (৪,) float64
পদক্ষেপ/পর্যবেক্ষণ/robot0_eye_in_hand_image ছবি (256, 256, 3) uint8
পদক্ষেপ/পর্যবেক্ষণ/robot0_gripper_qpos টেনসর (2,) float64
পদক্ষেপ/পর্যবেক্ষণ/robot0_gripper_qvel টেনসর (2,) float64
পদক্ষেপ/পর্যবেক্ষণ/robot0_joint_pos_cos টেনসর (৭,) float64
পদক্ষেপ/পর্যবেক্ষণ/robot0_joint_pos_sin টেনসর (৭,) float64
পদক্ষেপ/পর্যবেক্ষণ/robot0_joint_vel টেনসর (৭,) float64
পদক্ষেপ/পর্যবেক্ষণ/robot0_proprio-state টেনসর (৩২,) float64
পদক্ষেপ/পর্যবেক্ষণ/robot0_robotview_image ছবি (256, 256, 3) uint8
পদক্ষেপ/পুরস্কার টেনসর float64
পদক্ষেপ/ট্যাগ: স্থাপন করা হয়েছে টেনসর bool

robosuite_panda_pick_place_can/synthetic_stochastic_sac_afe13968

  • কনফিগারেশনের বিবরণ : SAC (200 পর্ব) এর সাথে প্রশিক্ষিত একটি স্টোকাস্টিক এজেন্ট দ্বারা তৈরি সিন্থেটিক ডেটাসেট।

  • হোমপেজ : https://github.com/google-research/rlds

  • ডাউনলোড সাইজ : 144.44 MiB

  • ডেটাসেটের আকার : 622.86 MiB

  • বিভাজন :

বিভক্ত উদাহরণ
'train' 200
  • বৈশিষ্ট্য গঠন :
FeaturesDict({
    'agent_id': string,
    'episode_id': string,
    'episode_index': int32,
    'steps': Dataset({
        'action': Tensor(shape=(7,), dtype=float32),
        'discount': float64,
        'image': Image(shape=(None, None, 3), dtype=uint8),
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'observation': FeaturesDict({
            'Can_pos': Tensor(shape=(3,), dtype=float32),
            'Can_quat': Tensor(shape=(4,), dtype=float32),
            'Can_to_robot0_eef_pos': Tensor(shape=(3,), dtype=float32),
            'Can_to_robot0_eef_quat': Tensor(shape=(4,), dtype=float32),
            'object-state': Tensor(shape=(14,), dtype=float32),
            'robot0_eef_pos': Tensor(shape=(3,), dtype=float32),
            'robot0_eef_quat': Tensor(shape=(4,), dtype=float32),
            'robot0_gripper_qpos': Tensor(shape=(2,), dtype=float32),
            'robot0_gripper_qvel': Tensor(shape=(2,), dtype=float32),
            'robot0_joint_pos_cos': Tensor(shape=(7,), dtype=float32),
            'robot0_joint_pos_sin': Tensor(shape=(7,), dtype=float32),
            'robot0_joint_vel': Tensor(shape=(7,), dtype=float32),
            'robot0_proprio-state': Tensor(shape=(32,), dtype=float32),
        }),
        'reward': float64,
        'tag:placed': bool,
    }),
})
  • বৈশিষ্ট্য ডকুমেন্টেশন :
বৈশিষ্ট্য ক্লাস আকৃতি ডিটাইপ বর্ণনা
ফিচারসডিক্ট
agent_id টেনসর স্ট্রিং
episode_id টেনসর স্ট্রিং
episode_index টেনসর int32
পদক্ষেপ ডেটাসেট
পদক্ষেপ/ক্রিয়া টেনসর (৭,) float32
পদক্ষেপ/ছাড় টেনসর float64
পদক্ষেপ/চিত্র ছবি (কোনটিই নয়, 3) uint8
steps/is_first টেনসর bool
ধাপ/শেষ_শেষ টেনসর bool
steps/is_terminal টেনসর bool
পদক্ষেপ/পর্যবেক্ষণ ফিচারসডিক্ট
পদক্ষেপ/পর্যবেক্ষণ/ক্যান_পোস টেনসর (৩,) float32
পদক্ষেপ/পর্যবেক্ষণ/ক্যান_কোয়াট টেনসর (৪,) float32
পদক্ষেপ/পর্যবেক্ষণ/ক্যান_টু_রোবট0_eef_pos টেনসর (৩,) float32
পদক্ষেপ/পর্যবেক্ষণ/Can_to_robot0_eef_quat টেনসর (৪,) float32
পদক্ষেপ/পর্যবেক্ষণ/বস্তু-স্থিতি টেনসর (14,) float32
পদক্ষেপ/পর্যবেক্ষণ/robot0_eef_pos টেনসর (৩,) float32
পদক্ষেপ/পর্যবেক্ষণ/robot0_eef_quat টেনসর (৪,) float32
পদক্ষেপ/পর্যবেক্ষণ/robot0_gripper_qpos টেনসর (2,) float32
পদক্ষেপ/পর্যবেক্ষণ/robot0_gripper_qvel টেনসর (2,) float32
পদক্ষেপ/পর্যবেক্ষণ/robot0_joint_pos_cos টেনসর (৭,) float32
পদক্ষেপ/পর্যবেক্ষণ/robot0_joint_pos_sin টেনসর (৭,) float32
পদক্ষেপ/পর্যবেক্ষণ/robot0_joint_vel টেনসর (৭,) float32
পদক্ষেপ/পর্যবেক্ষণ/robot0_proprio-state টেনসর (৩২,) float32
পদক্ষেপ/পুরস্কার টেনসর float64
পদক্ষেপ/ট্যাগ: স্থাপন করা হয়েছে টেনসর bool