bair_robot_pushing_small

  • 説明

このデータセットには、ロボットを押す動作の約44,000の例が含まれています。これには、1つのトレーニングセット(train)と、以前に表示された(testseen)オブジェクトと表示されていない(testnovel)オブジェクトの2つのテストセットが含まれます。これは小さな64x64バージョンです。

スプリット
'test' 256
'train' 43,264
  • 特長
Sequence({
    'action': Tensor(shape=(4,), dtype=tf.float32),
    'endeffector_pos': Tensor(shape=(3,), dtype=tf.float32),
    'image_aux1': Image(shape=(64, 64, 3), dtype=tf.uint8),
    'image_main': Image(shape=(64, 64, 3), dtype=tf.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},
}