- Descripción :
UR5 realiza tareas de selección/colocación/rotación de mesa
Página de inicio : https://link.springer.com/article/10.1007/s10514-023-10129-1
Código fuente :
tfds.robotics.rtx.AsuTableTopConvertedExternallyToRlds
Versiones :
-
0.1.0
(predeterminado): versión inicial.
-
Tamaño de descarga :
Unknown size
Tamaño del conjunto de datos :
737.60 MiB
Almacenamiento en caché automático ( documentación ): No
Divisiones :
Dividir | Ejemplos |
---|---|
'train' | 110 |
- Estructura de características :
FeaturesDict({
'episode_metadata': FeaturesDict({
'file_path': Text(shape=(), dtype=string),
}),
'steps': Dataset({
'action': Tensor(shape=(7,), dtype=float32),
'action_delta': Tensor(shape=(7,), dtype=float32),
'action_inst': Text(shape=(), dtype=string),
'discount': Scalar(shape=(), dtype=float32),
'goal_object': Text(shape=(), dtype=string),
'ground_truth_states': FeaturesDict({
'EE': Tensor(shape=(6,), dtype=float32),
'bottle': Tensor(shape=(6,), dtype=float32),
'bread': Tensor(shape=(6,), dtype=float32),
'coke': Tensor(shape=(6,), dtype=float32),
'cube': Tensor(shape=(6,), dtype=float32),
'milk': Tensor(shape=(6,), dtype=float32),
'pepsi': Tensor(shape=(6,), dtype=float32),
}),
'is_first': bool,
'is_last': bool,
'is_terminal': bool,
'language_embedding': Tensor(shape=(512,), dtype=float32),
'language_instruction': Text(shape=(), dtype=string),
'observation': FeaturesDict({
'image': Image(shape=(224, 224, 3), dtype=uint8),
'state': Tensor(shape=(7,), dtype=float32),
'state_vel': Tensor(shape=(7,), dtype=float32),
}),
'reward': Scalar(shape=(), dtype=float32),
}),
})
- Documentación de funciones :
Característica | Clase | Forma | tipo D | Descripción |
---|---|---|---|---|
FuncionesDict | ||||
episodio_metadatos | FuncionesDict | |||
metadatos_episodio/ruta_archivo | Texto | cadena | Ruta al archivo de datos original. | |
pasos | Conjunto de datos | |||
pasos/acción | Tensor | (7,) | flotador32 | La acción del robot consta de [7x velocidades de las articulaciones, 2x velocidades de la pinza, 1x episodio final]. |
pasos/action_delta | Tensor | (7,) | flotador32 | La acción delta del robot consta de [7 velocidades de las articulaciones, 2 velocidades de la pinza, 1 episodio de terminación]. |
pasos/action_inst | Texto | cadena | Acción a realizar. | |
pasos/descuento | Escalar | flotador32 | Descuento si se proporciona, el valor predeterminado es 1. | |
pasos/objeto_objetivo | Texto | cadena | Objeto a manipular. | |
pasos/estados_de_verdad_terreno | FuncionesDict | |||
pasos/ground_truth_states/EE | Tensor | (6,) | flotador32 | xyzrpy |
pasos/estados_de_verdad_terreno/botella | Tensor | (6,) | flotador32 | xyzrpy |
pasos/estados_de_verdad_terreno/pan | Tensor | (6,) | flotador32 | xyzrpy |
pasos/estados_de_verdad_terreno/coca | Tensor | (6,) | flotador32 | xyzrpy |
pasos/estados_de_verdad_terreno/cubo | Tensor | (6,) | flotador32 | xyzrpy |
pasos/estados_de_la_verdad_terreno/leche | Tensor | (6,) | flotador32 | xyzrpy |
pasos/estados_de_verdad_terreno/pepsi | Tensor | (6,) | flotador32 | xyzrpy |
pasos/es_primero | Tensor | booleano | ||
pasos/es_último | Tensor | booleano | ||
pasos/es_terminal | Tensor | booleano | ||
pasos/idioma_incrustación | Tensor | (512,) | flotador32 | Incorporación del lenguaje Kona. Consulte https://tfhub.dev/google/universal-sentence-encoder-large/5 |
pasos/instrucción_idioma | Texto | cadena | Instrucción de idiomas. | |
pasos/observación | FuncionesDict | |||
pasos/observación/imagen | Imagen | (224, 224, 3) | uint8 | Observación RGB de la cámara principal. |
pasos/observación/estado | Tensor | (7,) | flotador32 | El estado del robot consta de [6 ángulos de articulación del robot, 1 posición de la pinza]. |
pasos/observación/state_vel | Tensor | (7,) | flotador32 | La velocidad de la articulación del robot consta de [6x ángulos de la articulación del robot, 1x posición de la pinza]. |
pasos/recompensa | Escalar | flotador32 | Recompensa si se proporciona, 1 en el paso final para demostraciones. |
Claves supervisadas (ver documento
as_supervised
):None
Figura ( tfds.show_examples ): no compatible.
Ejemplos ( tfds.as_dataframe ):
- Cita :
@inproceedings{zhou2023modularity,
title={Modularity through Attention: Efficient Training and Transfer of Language-Conditioned Policies for Robot Manipulation},
author={Zhou, Yifan and Sonawani, Shubham and Phielipp, Mariano and Stepputtis, Simon and Amor, Heni},
booktitle={Conference on Robot Learning},
pages={1684--1695},
year={2023},
organization={PMLR}
}
@article{zhou2023learning,
title={Learning modular language-conditioned robot policies through attention},
author={Zhou, Yifan and Sonawani, Shubham and Phielipp, Mariano and Ben Amor, Heni and Stepputtis, Simon},
journal={Autonomous Robots},
pages={1--21},
year={2023},
publisher={Springer}
}