Skip to content

Add support for qpos/pose/rot6d conversions - #64

Merged
kou merged 4 commits into
enactic:mainfrom
k1000dai:pose_qpos_covnersion
Jul 27, 2026
Merged

Add support for qpos/pose/rot6d conversions#64
kou merged 4 commits into
enactic:mainfrom
k1000dai:pose_qpos_covnersion

Conversation

@k1000dai

@k1000dai k1000dai commented Jul 22, 2026

Copy link
Copy Markdown
Contributor

Fix #32

Summary

OpenArm Dataset stores only the raw recorded arm state (qpos from joint
teleop, pose from Cartesian teleop). This PR adds on-the-fly conversion
between state representations, so any dataset can be consumed — or exported
to LeRobot — in the representation a policy needs, without ever rewriting
the raw data.

  • qpospose via forward kinematics, poseqpos via inverse
    kinematics, using openarm_control
    (MuJoCo/mink) as a new core dependency
  • rot6d (position + first two rotation-matrix columns + gripper, 10-dim,
    Zhou et al.'s continuous 6D rotation representation) derived from the pose
  • LeRobot v2.1 / v3.0 / GR00T exports always emit one uniform representation,
    qpos by default

Copilot AI review requested due to automatic review settings July 22, 2026 06:35

Copilot AI left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Copilot was unable to review this pull request because the user who requested the review has reached their quota limit.

@kou kou changed the title Pose qpos covnersion Pose qpos conversion Jul 22, 2026
Copilot AI review requested due to automatic review settings July 24, 2026 06:15

Copilot AI left a comment

Copy link
Copy Markdown
Contributor

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Copilot was unable to review this pull request because the user who requested the review has reached their quota limit.

@k1000dai

Copy link
Copy Markdown
Contributor Author

enactic/openarm_control#18
pose / qpos conversion needs to match this convention. pose origin is arm_origin in mjcf.

@kou kou changed the title Pose qpos conversion Add support for qpos/pose/rot6d conversions Jul 27, 2026
@kou
kou merged commit d80fc34 into enactic:main Jul 27, 2026
7 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

Support EEF space conversion

3 participants