- 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.
Página de inicio: https://sites.google.com/view/sna-visual-mpc/
Código fuente :
tfds.datasets.bair_robot_pushing_small.Builder
Versiones :
-
2.0.0
(predeterminado): nueva API dividida ( https://tensorflow.org/datasets/splits )
-
Tamaño de descarga :
30.06 GiB
Tamaño del conjunto de datos :
20.80 GiB
Almacenamiento automático en caché ( documentación ): No
Divisiones :
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 |
Claves supervisadas (Ver
as_supervised
doc ):None
Figura ( tfds.show_examples ): no compatible.
Ejemplos ( tfds.as_dataframe ):
- 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},
}