删除 uf_lerobot_eval.py 中的调试 print/exit 代码, 清理 dual_xarm6_umi_record_config.yaml 中废弃的 teleop_id 和注释配置。 TeleopConfig 新增 __post_init__ 将 cameras 初始化为空字典,避免下游空值报错。
70 lines
2.0 KiB
YAML
70 lines
2.0 KiB
YAML
robot:
|
|
type: uf::multiple_mock_robot
|
|
id: "multiple_uf_mock_robots"
|
|
robots:
|
|
left:
|
|
type: uf::mock_robot
|
|
id: "left_uf_mock_robot"
|
|
control_space: "cartesian"
|
|
gripper_type: 1
|
|
state_offset_action: 3
|
|
teleop_id: "umi_250801DR48FP26001318"
|
|
cameras:
|
|
fisheye:
|
|
type: uf::umi_camera
|
|
serial_number: "250801DR48FP26001318"
|
|
width: 1280
|
|
height: 1280
|
|
fps: 30
|
|
right:
|
|
type: uf::mock_robot
|
|
id: "right_uf_mock_robot"
|
|
control_space: "cartesian"
|
|
gripper_type: 1
|
|
state_offset_action: 3
|
|
teleop_id: "umi_250801DR48FP26001295"
|
|
cameras:
|
|
fisheye:
|
|
type: uf::umi_camera
|
|
serial_number: "250801DR48FP26001295"
|
|
width: 1280
|
|
height: 1280
|
|
fps: 30
|
|
|
|
# make sure to edit with your correct configurations!
|
|
teleop:
|
|
type: uf::multiple_umi_teleop
|
|
id: "multiple_teleops"
|
|
teleops:
|
|
left:
|
|
type: uf::umi_teleop
|
|
id: "umi_250801DR48FP26001318"
|
|
serial_number: "250801DR48FP26001318"
|
|
use_vive_tracker: False
|
|
vive_tracker_id: "LHR-555DC7BF"
|
|
use_gripper: True
|
|
tracker_to_robot_eef: [0, 0, 0, 0, 0, -90]
|
|
robot_base_pose: [400, 0, 150, -90, 0, 0]
|
|
right:
|
|
type: uf::umi_teleop
|
|
id: "umi_250801DR48FP26001295"
|
|
serial_number: "250801DR48FP26001295"
|
|
use_vive_tracker: False
|
|
vive_tracker_id: "LHR-2425BAD3"
|
|
use_gripper: True
|
|
tracker_to_robot_eef: [0, 0, 0, 0, 0, -90]
|
|
robot_base_pose: [400, 0, 150, 90, 0, 0]
|
|
|
|
dataset:
|
|
# root of local repo: /home/<user_name>/.cache/huggingface/lerobot (default)
|
|
root: "/home/uf/Data/lerobot_datas/record/ufactory/dual_umi_datas"
|
|
repo_id: "ufactory/dual_umi_datas"
|
|
single_task: "Pick up the purple grape and drop into the box on the left."
|
|
fps: 60
|
|
episode_time_s: 60 # max duration for one episode
|
|
reset_time_s: 20 # time for resetting env between episodes
|
|
num_episodes: 100
|
|
push_to_hub: False
|
|
|
|
|