Update follow and master key mismatch in compute_norm_stats.py (#86)
Co-authored-by: Jerry Chen <jerry@x2robot.com>
This commit is contained in:
co-authored by
Jerry Chen
parent
d18fa65fa1
commit
97406f2ab5
@@ -161,12 +161,12 @@ def main() -> None:
|
|||||||
data_root_path = "/path/to/lerobot/dataset"
|
data_root_path = "/path/to/lerobot/dataset"
|
||||||
output_stats_dir = "/path/to/save/action_stats"
|
output_stats_dir = "/path/to/save/action_stats"
|
||||||
trajectory_keys = { # your dataset keys
|
trajectory_keys = { # your dataset keys
|
||||||
"action": {
|
"propri": {
|
||||||
"follow_right_ee_cartesian_pos": [0, 3],
|
"follow_right_ee_cartesian_pos": [0, 3],
|
||||||
"follow_right_ee_rotation": [3, 6],
|
"follow_right_ee_rotation": [3, 6],
|
||||||
"follow_right_gripper": [6, 7],
|
"follow_right_gripper": [6, 7],
|
||||||
},
|
},
|
||||||
"propri": {
|
"action": {
|
||||||
"master_right_ee_cartesian_pos": [0, 3],
|
"master_right_ee_cartesian_pos": [0, 3],
|
||||||
"master_right_ee_rotation": [3, 6],
|
"master_right_ee_rotation": [3, 6],
|
||||||
"master_right_gripper": [6, 7],
|
"master_right_gripper": [6, 7],
|
||||||
|
|||||||
Reference in New Issue
Block a user