bair_robot_pushing_small

  • Описание :

Этот набор данных содержит примерно 44 000 примеров толкающих движений робота, включая один обучающий набор (поезд) и два тестовых набора ранее увиденных (тестовые) и невидимых (тестовые новые) объектов. Это маленькая версия 64x64.

Расколоть Примеры
'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
  • Цитата :
@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},
}