Skip to content

feat(action): add human hand pose dataset#71

Open
hans-yang-tps wants to merge 2 commits into
NVIDIA:mainfrom
hans-yang-tps:data/hand-pose-fd
Open

feat(action): add human hand pose dataset#71
hans-yang-tps wants to merge 2 commits into
NVIDIA:mainfrom
hans-yang-tps:data/hand-pose-fd

Conversation

@hans-yang-tps

@hans-yang-tps hans-yang-tps commented Jul 1, 2026

Copy link
Copy Markdown

Description

Add a human hand-pose LeRobot dataset definition for Cosmos3 forward dynamics.

Companion cookbook example: NVIDIA/cosmos#253.

The dataset converts camera, wrist, and fingertip poses into a 57-dimensional dual-hand action representation, aligns wrist poses to the camera frame, samples 30 FPS source data at 15 FPS, and applies validation-derived quantile normalization.

Changes

  • Add HumanHandPoseLeRobotDataset with multi-episode-safe frame indexing.
  • Add hand-pose normalization statistics derived from the selected validation episode.
  • Register the dataset and the 57D hand_pose action domain.
  • Add unit coverage for action conversion and episode-boundary indexing.

Validation

  • New dataset tests: 2 passed.
  • Pose utility and dataset tests: 19 passed.
  • Ruff check and formatting passed.
  • Cosmos3 Nano framework inference completed successfully on the public cookbook sample: 17 frames at 15 FPS, 832x480.
  • Both updated cookbook notebooks were executed end-to-end; all non-empty code cells completed without errors.

@hans-yang-tps

Copy link
Copy Markdown
Author

Need Michael(Zhengyu) Huang's help to review it.

@hans-yang-tps hans-yang-tps marked this pull request as draft July 1, 2026 09:58
Signed-off-by: Hans Yang <hayang@nvidia.com>
@hans-yang-tps hans-yang-tps changed the title feat(action): add Mecka hand pose dataset feat(action): add human hand pose dataset Jul 2, 2026

@lzhang57 lzhang57 left a comment

Copy link
Copy Markdown
Collaborator

Choose a reason for hiding this comment

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

LGTM!

@mli0603

mli0603 commented Jul 2, 2026

Copy link
Copy Markdown
Collaborator

one more comment. Please fix.

@hans-yang-tps

Copy link
Copy Markdown
Author

Also removed Mecka keywords.

# Conflicts:
#	cosmos_framework/data/generator/action/datasets/human_hand_pose_lerobot_dataset.py
#	cosmos_framework/data/generator/action/datasets/human_hand_pose_lerobot_dataset_test.py
#	cosmos_framework/data/generator/action/normalizer_stats/human_hand_pose_lerobot_stats.json
#	cosmos_framework/data/vfm/action/datasets/__init__.py
@hans-yang-tps hans-yang-tps marked this pull request as ready for review July 9, 2026 08:45
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.

3 participants