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 :

Diviser	Exemples
'test'	256
'train'	43 264

• 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 :

Caractéristique	Classer	Forme	Dtype	La description
	Séquence
action	Tenseur	(4,)	float32
endeffector_pos	Tenseur	(3,)	float32
image_aux1	Image	(64, 64, 3)	uint8
image_main	Image	(64, 64, 3)	uint8

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