bair_robot_pushing_small

Описание :

Этот набор данных содержит примерно 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},
}