bair_robot_pushing_small

• Description:

This data set contains roughly 44,000 examples of robot pushing motions, including one training set (train) and two test sets of previously seen (testseen) and unseen (testnovel) objects. This is the small 64x64 version.

• Homepage: https://sites.google.com/view/sna-visual-mpc/

• Source code: tfds.datasets.bair_robot_pushing_small.Builder

• Versions:

  • 2.0.0 (default): New split API (https://tensorflow.org/datasets/splits)

• Download size: 30.06 GiB

• Dataset size: 20.80 GiB

• Auto-cached (documentation): No

• Splits:

Split	Examples
'test'	256
'train'	43,264

• Feature structure:

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

• Feature documentation:

Feature	Class	Shape	Dtype	Description
	Sequence
action	Tensor	(4,)	float32
endeffector_pos	Tensor	(3,)	float32
image_aux1	Image	(64, 64, 3)	uint8
image_main	Image	(64, 64, 3)	uint8

• Supervised keys (See as_supervised doc): None

• Figure (tfds.show_examples): Not supported.

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