- বর্ণনা :
14টি দীর্ঘ দিগন্ত ম্যানিপুলেশন টাস্কের বাস্তব ডেটাসেট। মানুষের খেলার ডেটা এবং একক রোবট আর্ম ডেটার মিশ্রণ একই কাজগুলি সম্পাদন করে৷
হোমপেজ : https://mimic-play.github.io/
সোর্স কোড :
tfds.robotics.rtx.MimicPlay
সংস্করণ :
-
0.1.0
(ডিফল্ট): প্রাথমিক প্রকাশ।
-
ডাউনলোড আকার :
Unknown size
ডেটাসেটের আকার :
7.14 GiB
স্বয়ংক্রিয় ক্যাশে ( ডকুমেন্টেশন ): না
বিভাজন :
বিভক্ত | উদাহরণ |
---|---|
'train' | 378 |
- বৈশিষ্ট্য গঠন :
FeaturesDict({
'episode_metadata': FeaturesDict({
'file_path': string,
}),
'steps': Dataset({
'action': Tensor(shape=(7,), dtype=float32),
'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': FeaturesDict({
'front_image_1': Image(shape=(120, 120, 3), dtype=uint8),
'front_image_2': Image(shape=(120, 120, 3), dtype=uint8),
}),
'state': FeaturesDict({
'ee_pose': Tensor(shape=(7,), dtype=float32),
'gripper_position': float32,
'joint_positions': Tensor(shape=(7,), dtype=float32),
'joint_velocities': Tensor(shape=(7,), dtype=float32),
}),
'wrist_image': FeaturesDict({
'wrist_image': Image(shape=(120, 120, 3), dtype=uint8),
}),
}),
'reward': Scalar(shape=(), dtype=float32),
}),
})
- বৈশিষ্ট্য ডকুমেন্টেশন :
বৈশিষ্ট্য | ক্লাস | আকৃতি | ডিটাইপ | বর্ণনা |
---|---|---|---|---|
ফিচারসডিক্ট | ||||
episode_metadata | ফিচারসডিক্ট | |||
episode_metadata/file_path | টেনসর | স্ট্রিং | ||
পদক্ষেপ | ডেটাসেট | |||
পদক্ষেপ/ক্রিয়া | টেনসর | (৭,) | float32 | |
পদক্ষেপ/ছাড় | স্কেলার | float32 | ||
steps/is_first | টেনসর | bool | ||
ধাপ/শেষ_শেষ | টেনসর | bool | ||
steps/is_terminal | টেনসর | bool | ||
পদক্ষেপ/ভাষা_এম্বেডিং | টেনসর | (512,) | float32 | |
পদক্ষেপ/ভাষা_নির্দেশ | টেনসর | স্ট্রিং | ||
পদক্ষেপ/পর্যবেক্ষণ | ফিচারসডিক্ট | |||
পদক্ষেপ/পর্যবেক্ষণ/চিত্র | ফিচারসডিক্ট | |||
পদক্ষেপ/পর্যবেক্ষণ/ছবি/সামনের_চিত্র_1 | ছবি | (120, 120, 3) | uint8 | |
পদক্ষেপ/পর্যবেক্ষণ/ছবি/সামনের_ছবি_2 | ছবি | (120, 120, 3) | uint8 | |
পদক্ষেপ/পর্যবেক্ষণ/রাষ্ট্র | ফিচারসডিক্ট | |||
পদক্ষেপ/অবজারভেশন/স্টেট/ইই_পোজ | টেনসর | (৭,) | float32 | |
পদক্ষেপ/পর্যবেক্ষণ/রাষ্ট্র/গ্রিপার_পজিশন | টেনসর | float32 | ||
পদক্ষেপ/পর্যবেক্ষণ/রাষ্ট্র/জয়েন্ট_পজিশন | টেনসর | (৭,) | float32 | |
পদক্ষেপ/পর্যবেক্ষন/রাষ্ট্র/জয়েন্ট_বেগ | টেনসর | (৭,) | float32 | |
পদক্ষেপ/পর্যবেক্ষণ/কব্জি_চিত্র | ফিচারসডিক্ট | |||
পদক্ষেপ/পর্যবেক্ষণ/wrist_image/wrist_image | ছবি | (120, 120, 3) | uint8 | |
পদক্ষেপ/পুরস্কার | স্কেলার | float32 |
তত্ত্বাবধান করা কী (দেখুন
as_supervised
doc ):None
চিত্র ( tfds.show_examples ): সমর্থিত নয়।
উদাহরণ ( tfds.as_dataframe ):
- উদ্ধৃতি :
@article{wang2023mimicplay,title={Mimicplay: Long-horizon imitation learning by watching human play},author={Wang, Chen and Fan, Linxi and Sun, Jiankai and Zhang, Ruohan and Fei-Fei, Li and Xu, Danfei and Zhu, Yuke and Anandkumar, Anima},journal={arXiv preprint arXiv:2302.12422},year={2023} }