bair_robot_empujando_pequeño

  • Descripción :

Este conjunto de datos contiene aproximadamente 44 000 ejemplos de movimientos de empuje de robots, incluido un conjunto de entrenamiento (tren) y dos conjuntos de prueba de objetos vistos previamente (testseen) y no vistos (testnovel). Esta es la versión pequeña de 64x64.

Separar Ejemplos
'test' 256
'train' 43,264
  • Estructura de características :
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),
})
  • Documentación de características :
Rasgo Clase Forma Tipo D Descripción
Secuencia
acción Tensor (4,) flotar32
endeffector_pos Tensor (3,) flotar32
imagen_aux1 Imagen (64, 64, 3) uint8
imagen_principal Imagen (64, 64, 3) uint8
  • Cita :
@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},
}