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.

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ự mô tả
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
  • 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},
}