bair_robot_pushing_small

  • توضیحات :

این مجموعه داده شامل تقریباً 44000 نمونه از حرکات هل دادن ربات، از جمله یک مجموعه آموزشی (قطار) و دو مجموعه آزمایشی از اشیاء قبلاً دیده شده (تست دیده شده) و دیده نشده (تست نوول) است. این نسخه کوچک 64x64 است.

شکاف مثال ها
'test' 256
'train' 43,264
  • ساختار ویژگی :
Sequence({
   
'action': Tensor(shape=(4,), dtype=float32),
   
'endeffector_pos': Tensor(shape=(3,), dtype=float32),
   
'image_aux1': Image(shape=(64, 64, 3), dtype=uint8),
   
'image_main': Image(shape=(64, 64, 3), dtype=uint8),
})
  • مستندات ویژگی :
ویژگی کلاس شکل نوع D شرح
توالی
عمل تانسور (4،) float32
endeffector_pos تانسور (3،) float32
image_aux1 تصویر (64، 64، 3) uint8
image_main تصویر (64، 64، 3) uint8
  • نقل قول :
@misc{1710.05268,
 
Author = {Frederik Ebert and Chelsea Finn and Alex X. Lee and Sergey Levine},
 
Title = {Self-Supervised Visual Planning with Temporal Skip Connections},
 
Year = {2017},
 
Eprint = {arXiv:1710.05268},
}