Add T1 dance task config and update motion tool docs#2
Open
fanyahao1 wants to merge 3 commits intoBoosterRobotics:mainfrom
Open
Add T1 dance task config and update motion tool docs#2fanyahao1 wants to merge 3 commits intoBoosterRobotics:mainfrom
fanyahao1 wants to merge 3 commits intoBoosterRobotics:mainfrom
Conversation
Author
|
This change only covers the booster_train side. The motion file configured here in |
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Sign up for free
to join this conversation on GitHub.
Already have an account?
Sign in to comment
Add this suggestion to a batch that can be applied as a single commit.This suggestion is invalid because no changes were made to the code.Suggestions cannot be applied while the pull request is closed.Suggestions cannot be applied while viewing a subset of changes.Only one suggestion per line can be applied in a batch.Add this suggestion to a batch that can be applied as a single commit.Applying suggestions on deleted lines is not supported.You must change the existing code in this line in order to create a valid suggestion.Outdated suggestions cannot be applied.This suggestion has been applied or marked resolved.Suggestions cannot be applied from pending reviews.Suggestions cannot be applied on multi-line comments.Suggestions cannot be applied while the pull request is queued to merge.Suggestion cannot be applied right now. Please check back later.
Summary
This PR adds the first T1 dance motion-tracking task configuration to
booster_train, updates the play environment registration used by the existing K1 dance task, and refreshes the motion conversion / replay documentation so the workflow matches the current scripts.What changed
1. Add T1 dance task configuration
Introduces a new T1 dance task package under
source/booster_train/booster_train/tasks/manager_based/beyond_mimic/robots/T1/dance/.Included files:
__init__.pyBooster-T1-Dance-v0Booster-T1-Dance-v0-Playenv_cfg.pyBOOSTER_T1_CFGmotions/T1/CLIO_Dimitroula_1_stageii.npztracking_env_cfg.pyppo_cfg.py2. Add T1 robot support to motion tools
Updates the motion conversion / replay scripts so they can be used with both K1 and T1.
Changes include:
scripts/mimic/scripts/mimic/csv_to_npz.py--robot {k1,t1}--frame_range--output_fpsscripts/mimic/replay_npz.py--robot {k1,t1}.npzfile or a Weights & Biases registry artifact:latestwhen the registry alias is omitted3. Update dance play environment registration
Updates the existing K1 dance config so the play environment uses
PlayFlatWoStateEstimationEnvCfginstead of the training config, keeping the play-time setup aligned with the dedicated play behavior.4. Refresh README documentation
Expands
README.mdso the documented workflow matches the current scripts:scripts/...toscripts/mimic/...--output_file--robot,--frame_range, and--output_fpsCommit breakdown
3b51bc5—feat: add PlayFlatWoStateEstimationEnvCfg and update gym registration for play environment54165a9—docs: update motion conversion and replay tools README0216b24—feat: add T1 dance task with motion tracking environment configDiff summary
Notes