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.

• Page d' accueil : https://sites.google.com/view/sna-visual-mpc/

• Code source : tfds.datasets.bair_robot_pushing_small.Builder

• Versions :

  • 2.0.0 (par défaut) : nouvelle API fractionnée ( https://tensorflow.org/datasets/splits )

• Taille du téléchargement : 30.06 GiB

• Taille du jeu de données : 20.80 GiB

• Mise en cache automatique ( documentation ): Non

• Fractionnements :

• Structure des fonctionnalités :

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),
})

• Documentation des fonctionnalités :

• Clés supervisées (Voir as_supervised doc ): None

• Figure ( tfds.show_examples ) : non pris en charge.

• Exemples ( tfds.as_dataframe ):

• 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},
}