bair_robot_pushing_small

  • Description:

Cet ensemble de données contient environ 44 000 exemples de mouvements de poussée de robot, y compris un ensemble d'entraînement (train) et deux ensembles de tests 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},
}