bair_robot_push_small

• Descrição :

Este conjunto de dados contém cerca de 44.000 exemplos de movimentos de empurrar robôs, incluindo um conjunto de treinamento (trem) e dois conjuntos de teste de objetos vistos anteriormente (testseen) e não vistos (testnovel). Esta é a pequena versão 64x64.

• Página inicial : https://sites.google.com/view/sna-visual-mpc/

• Código -fonte: tfds.datasets.bair_robot_pushing_small.Builder

• Versões :

  • 2.0.0 (padrão): Nova API de divisão ( https://tensorflow.org/datasets/splits )

• Tamanho do download : 30.06 GiB

• Tamanho do conjunto de dados : 20.80 GiB

• Armazenado em cache automaticamente ( documentação ): Não

• Divisões :

Dividir	Exemplos
'test'	256
'train'	43.264

• Estrutura de recursos :

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

• Documentação do recurso:

Característica	Classe	Forma	Tipo D	Descrição
	Seqüência
ação	tensor	(4,)	float32
endeffector_pos	tensor	(3,)	float32
image_aux1	Imagem	(64, 64, 3)	uint8
image_main	Imagem	(64, 64, 3)	uint8

• Chaves supervisionadas (Consulte as_supervised doc ): None

• Figura ( tfds.show_examples ): Não compatível.

• Exemplos ( tfds.as_dataframe ):

• Citação :

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