bair_robot_pushing_small

  • Descriptif :

Cet ensemble de données contient environ 44 000 exemples de mouvements de poussée de robot, dont un ensemble d'entraînement (train) et deux ensembles de test d'objets déjà vus (testseen) et invisibles (testnovel). Il s'agit de la petite version 64x64.

Diviser Exemples
'test' 256
'train' 43 264
  • Caractéristiques :
Sequence({
    'action': Tensor(shape=(4,), dtype=tf.float32),
    'endeffector_pos': Tensor(shape=(3,), dtype=tf.float32),
    'image_aux1': Image(shape=(64, 64, 3), dtype=tf.uint8),
    'image_main': Image(shape=(64, 64, 3), dtype=tf.uint8),
})
  • Citation :
@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},
}