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.

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