anybody tried to convert bridge dataset v2 to lerobot format with this code?

[kokun66]

anybody tried to convert bridge dataset v2 to lerobot format with this code?

[Tavish9]

anybody tried to convert bridge dataset v2 to lerobot format with this code?

Hi, bridge dataset v2 is currently supported in openx2lerobot.

use bridge_orig or bridge_dataset to convert the dataset

any4lerobot/openx2lerobot/oxe_utils/configs.py

Lines 118 to 135 in 37f856f

	"bridge_orig": { # Original version of Bridge V2 from project website
	"image_obs_keys": {"primary": "image_0", "secondary": "image_1", "wrist": None},
	"depth_obs_keys": {"primary": None, "secondary": None, "wrist": None},
	"state_obs_keys": ["EEF_state", None, "gripper_state"],
	"state_encoding": StateEncoding.POS_EULER,
	"action_encoding": ActionEncoding.EEF_POS,
	"control_frequency": 5,
	"robot_type": "WidowX",
	},
	"bridge_dataset": { # Original version of Bridge V2 from project website
	"image_obs_keys": {"primary": "image_0", "secondary": "image_1", "wrist": None},
	"depth_obs_keys": {"primary": None, "secondary": None, "wrist": None},
	"state_obs_keys": ["EEF_state", None, "gripper_state"],
	"state_encoding": StateEncoding.POS_EULER,
	"action_encoding": ActionEncoding.EEF_POS,
	"control_frequency": 5,
	"robot_type": "WidowX",
	},