- Описание :
Этот набор данных содержит примерно 44 000 примеров толкающих движений робота, включая один обучающий набор (поезд) и два тестовых набора ранее увиденных (тестовые) и невидимых (тестовые новые) объектов. Это маленькая версия 64x64.
Домашняя страница : https://sites.google.com/view/sna-visual-mpc/
Исходный код :
tfds.datasets.bair_robot_pushing_small.BuilderВерсии :
-
2.0.0(по умолчанию): новый раздельный API ( https://tensorflow.org/datasets/splits )
-
Размер загрузки :
30.06 GiBРазмер набора данных :
20.80 GiBАвтоматическое кэширование ( документация ): Нет
Сплиты :
| Расколоть | Примеры |
|---|---|
'test' | 256 |
'train' | 43 264 |
- Структура функции :
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),
})
- Документация по функциям :
| Особенность | Учебный класс | Форма | Dтип | Описание |
|---|---|---|---|---|
| Последовательность | ||||
| действие | Тензор | (4,) | поплавок32 | |
| endefector_pos | Тензор | (3,) | поплавок32 | |
| image_aux1 | Изображение | (64, 64, 3) | uint8 | |
| image_main | Изображение | (64, 64, 3) | uint8 |
Ключи под наблюдением (см . документ
as_supervised):NoneРисунок ( tfds.show_examples ): не поддерживается.
Примеры ( tfds.as_dataframe ):
- Цитата :
@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},
}