- বর্ণনা :
এই ডেটাসেটগুলি রোবোসুইট রোবোটিক আর্ম সিমুলেটরের পিকপ্লেসক্যান পরিবেশের সাথে তৈরি করা হয়েছে। মানব ডেটাসেটগুলি RLDS ক্রিয়েটর এবং একটি গেমপ্যাড কন্ট্রোলার ব্যবহার করে একটি একক অপারেটর দ্বারা রেকর্ড করা হয়েছিল।
সিনথেটিক ডেটাসেটগুলি EnvLogger লাইব্রেরি ব্যবহার করে রেকর্ড করা হয়েছে।
ডেটাসেটগুলি ধাপ এবং পর্বগুলি উপস্থাপন করতে RLDS বিন্যাস অনুসরণ করে।
পর্ব 400টি ধাপ নিয়ে গঠিত। প্রতিটি পর্বে, একটি ট্যাগ যোগ করা হয় যখন কাজটি সম্পন্ন হয়, এই ট্যাগটি কাস্টম ধাপের মেটাডেটার অংশ হিসাবে সংরক্ষণ করা হয়।
মনে রাখবেন, EnvLogger নির্ভরতার কারণে, এই ডেটাসেট তৈরি করা বর্তমানে শুধুমাত্র Linux পরিবেশে সমর্থিত।
সোর্স কোড :
tfds.rlds.datasets.robosuite_panda_pick_place_can.RobosuitePandaPickPlaceCan
সংস্করণ :
-
1.0.0
(ডিফল্ট): প্রাথমিক প্রকাশ।
-
স্বয়ংক্রিয় ক্যাশে ( ডকুমেন্টেশন ): না
তত্ত্বাবধান করা কী (দেখুন
as_supervised
doc ):None
চিত্র ( tfds.show_examples ): সমর্থিত নয়।
উদ্ধৃতি :
@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 পর্ব)।
ডাউনলোড সাইজ :
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 |
- উদাহরণ ( tfds.as_dataframe ):
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 |
- উদাহরণ ( tfds.as_dataframe ):
robosuite_panda_pick_place_can/synthetic_stochastic_sac_afe13968
কনফিগারেশনের বিবরণ : SAC (200 পর্ব) এর সাথে প্রশিক্ষিত একটি স্টোকাস্টিক এজেন্ট দ্বারা তৈরি সিন্থেটিক ডেটাসেট।
ডাউনলোড সাইজ :
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 |
- উদাহরণ ( tfds.as_dataframe ):