Skip to content

Add MicroAGI embodiment and data configs:#496

Draft
ElmoPA wants to merge 2 commits into
mainfrom
testing-fork
Draft

Add MicroAGI embodiment and data configs:#496
ElmoPA wants to merge 2 commits into
mainfrom
testing-fork

Conversation

@ElmoPA

@ElmoPA ElmoPA commented Jun 12, 2026

Copy link
Copy Markdown
Contributor

Add MicroAGI embodiment and data configs:

  • Microagi Human-sibling embodiment with keymaps, transform list,
    intrinsics fallback, MediaPipe / MANO keypoint connectivity
    -microagi_bimanual / right_arm / left_arm embodiment IDs + hydra data/viz configs
  • CONTRIBUTING_DATA.md embodiment table updates

Co-authored-by: Aristotelis-Sib aristotelis98@gmail.com

viz_language: resolve short image keys against dotted batch keys

The evaluator/viz configs (keypoints.yaml, keypoints_wrist.yaml) specify
image_key as a flat name like 'front_img_1' because they were written for
the trainer's EvalVideo path, where HPT flattens dotted batch keys to
their last segment before viz. viz_language operates on raw dataloader
batches, where the same image is keyed 'observations.images.front_img_1',
so those configs raised KeyError on every batch and produced zero videos.

Resolve a missing image_key by matching batch keys on their last dotted
segment — the inverse of HPT's flattening. Only the image key is resolved
(wholesale flattening would collide, e.g. left/right.obs_keypoints), and
only when the exact key is absent, so configs using full dotted keys
(cotrain_lang.yaml) are unaffected.

marcopepunkt added 2 commits June 11, 2026 21:08
  - Microagi Human-sibling embodiment with keymaps, transform list,
    intrinsics fallback, MediaPipe / MANO keypoint connectivity
  -microagi_bimanual / right_arm / left_arm embodiment IDs + hydra data/viz configs
  - CONTRIBUTING_DATA.md embodiment table updates

  Co-authored-by: Aristotelis-Sib <aristotelis98@gmail.com>
The evaluator/viz configs (keypoints.yaml, keypoints_wrist.yaml) specify
image_key as a flat name like 'front_img_1' because they were written for
the trainer's EvalVideo path, where HPT flattens dotted batch keys to
their last segment before viz. viz_language operates on raw dataloader
batches, where the same image is keyed 'observations.images.front_img_1',
so those configs raised KeyError on every batch and produced zero videos.

Resolve a missing image_key by matching batch keys on their last dotted
segment — the inverse of HPT's flattening. Only the image key is resolved
(wholesale flattening would collide, e.g. left/right.obs_keypoints), and
only when the exact key is absent, so configs using full dotted keys
(cotrain_lang.yaml) are unaffected.

ElmoPA commented Jun 12, 2026

Copy link
Copy Markdown
Contributor Author

This stack of pull requests is managed by Graphite. Learn more about stacking.

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.

1 participant