bair_robot_pushing_small

説明:

このデータセットには、ロボットのプッシュモーションの約 44,000 の例が含まれています。これには、1 つのトレーニングセット (train) と、以前に見た (testseen) オブジェクトと目に見えない (testnovel) オブジェクトの 2 つのテストセットが含まれます。これは小さい 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),
})

機能のドキュメント:

特徴	クラス	形	Dtype
	順序
アクション	テンソル	(4)	float32
endeffector_pos	テンソル	(3)	float32
image_aux1	画像	(64, 64, 3)	uint8
image_main	画像	(64, 64, 3)	uint8

監視されたキー( as_supervised docを参照): 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},
}