bair_robot_pushing_small

Mô tả :

Tập dữ liệu này chứa khoảng 44.000 ví dụ về chuyển động đẩy của rô-bốt, bao gồm một tập huấn luyện (đào tạo) và hai tập kiểm tra gồm các đối tượng đã nhìn thấy trước đó (kiểm tra) và chưa nhìn thấy (tiểu thuyết thử nghiệm). Đây là phiên bản nhỏ 64x64.

Trang chủ : https://sites.google.com/view/sna-visual-mpc/
Mã nguồn : tfds.datasets.bair_robot_pushing_small.Builder
Phiên bản :
- 2.0.0 (mặc định): API phân tách mới ( https://tensorflow.org/datasets/splits )
Kích thước tải xuống : 30.06 GiB
Kích thước tập dữ liệu : 20.80 GiB
Tự động lưu vào bộ nhớ cache ( tài liệu ): Không
Chia tách :

Tách ra	ví dụ
`'test'`	256
`'train'`	43,264

Cấu trúc tính năng :

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),
})

Tài liệu tính năng :

Tính năng	Lớp	Hình dạng	Dtype
	Sự phối hợp
hoạt động	tenxơ	(4,)	phao32
endeffector_pos	tenxơ	(3,)	phao32
image_aux1	Hình ảnh	(64, 64, 3)	uint8
image_main	Hình ảnh	(64, 64, 3)	uint8

Các khóa được giám sát (Xem as_supervised doc ): None
Hình ( tfds.show_examples ): Không được hỗ trợ.
Ví dụ ( tfds.as_dataframe ):

trích dẫn :

@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},
}