plex_robosuite

Tanım :

Robosuite'in Door, Stack, PickPlaceMilk, PickPlaceBread, PickPlaceCereal ve NutAssemblyRound görevleri için yüksek kaliteli gösterim yörüngelerinden oluşan bir veri kümesi, her biri 75 gösterim.

Ana sayfa : https://microsoft.github.io/PLEX/
Kaynak kodu : tfds.robotics.rtx.PlexRobosuite
Sürümler :
- 0.1.0 (varsayılan): İlk sürüm.
İndirme boyutu : Unknown size
Veri kümesi boyutu : 1.27 GiB
Otomatik önbelleğe alınmış ( belgeler ): Hayır
Bölünmeler :

Bölmek	Örnekler
`'train'`	402
`'val'`	48

Özellik yapısı :

FeaturesDict({
    'episode_metadata': FeaturesDict({
        'episode_id': string,
    }),
    'steps': Dataset({
        'action': Tensor(shape=(7,), dtype=float64),
        'discount': Scalar(shape=(), dtype=float32),
        'is_first': bool,
        'is_last': bool,
        'is_terminal': bool,
        'language_embedding': Tensor(shape=(512,), dtype=float32),
        'language_instruction': string,
        'observation': FeaturesDict({
            'image': Image(shape=(128, 128, 3), dtype=uint8),
            'state': Tensor(shape=(32,), dtype=float64),
            'wrist_image': Image(shape=(128, 128, 3), dtype=uint8),
        }),
        'reward': Scalar(shape=(), dtype=float32),
    }),
})

Özellik belgeleri :

Özellik	Sınıf	Şekil	Dtipi
	ÖzelliklerDict
bölüm_meta verileri	ÖzelliklerDict
bölüm_metadata/episode_id	Tensör		sicim
adımlar	Veri kümesi
adımlar/eylem	Tensör	(7,)	kayan nokta64
adımlar/indirim	Skaler		kayan nokta32
adımlar/is_first	Tensör		bool
adımlar/is_last	Tensör		bool
adımlar/is_terminal	Tensör		bool
adımlar/dil_embedding	Tensör	(512,)	kayan nokta32
adımlar/language_instruction	Tensör		sicim
adımlar/gözlem	ÖzelliklerDict
adımlar/gözlem/görüntü	Resim	(128, 128, 3)	uint8
adımlar/gözlem/durum	Tensör	(32,)	kayan nokta64
adımlar/gözlem/wrist_image	Resim	(128, 128, 3)	uint8
adımlar/ödül	Skaler		kayan nokta32

Denetlenen anahtarlar ( as_supervised belgesine bakın): None
Şekil ( tfds.show_examples ): Desteklenmiyor.
Örnekler ( tfds.as_dataframe ): Eksik.
Alıntı :

https://doi.org/10.48550/arXiv.2303.08789