├── .gitignore ├── LICENSE.md ├── README.md ├── asset └── teaser.gif ├── configs ├── AIST │ ├── aist_example.yaml │ └── aist_example2.yaml ├── AMASS │ ├── amass_example.yaml │ ├── amass_example2.yaml │ └── amass_example3.yaml ├── AMP │ ├── backflip.yaml │ ├── backflip_20s.yaml │ ├── backflip_20s_vanilla.yaml │ ├── cartwheel_20s.yaml │ ├── crawl_20s.yaml │ ├── dance_20s.yaml │ ├── jog_20s.yaml │ ├── jump_20s.yaml │ ├── roll.yaml │ ├── roll_20s.yaml │ ├── run_20s.yaml │ ├── sideflip_20s.yaml │ ├── spinkick_20s.yaml │ ├── walk_20s.yaml │ └── zombie_20s.yaml └── SWORDSHIELD │ └── RL_Avatar_Atk_2xCombo01_Motion.yaml ├── data ├── demo_aist │ ├── gBR_sBM_cAll_d04_mBR0_ch01.npy │ └── gBR_sFM_cAll_d04_mBR1_ch02.npy ├── demo_amass │ ├── 75_09_stageii.npy │ ├── KickHuefthoch05_stageii.npy │ └── WalkingStraightBackwards07_stageii.npy ├── demo_humanoid │ ├── backflip_fps_30_20s.npy │ ├── cartwheel_fps_30_20s.npy │ ├── crawl_fps_30_20s.npy │ ├── dance_a_fps_30_20s.npy │ ├── jog_fps_30_20s.npy │ ├── jump_fps_30_20s.npy │ ├── roll_fps_30_20s.npy │ ├── run_fps_30_20s.npy │ ├── sideflip_fps_30_20s.npy │ ├── spinkick_fps_30_20s.npy │ ├── walk_fps_30_20s.npy │ └── zombie_fps_30_20s.npy ├── demo_swordshield │ └── RL_Avatar_Atk_2xCombo01_Motion.npy └── tools │ ├── README.md │ ├── aist_converter.py │ ├── amass_converter.py │ ├── amp_converter.py │ ├── joint_utils.py │ ├── rotation_utils │ ├── conversions.py │ └── quaternion.py │ ├── sword_converter.py │ └── to_smpl.py ├── diffmimic ├── __init__.py ├── brax_lib │ ├── __init__.py │ ├── acting.py │ └── agent_diffmimic.py ├── mimic_envs │ ├── __init__.py │ ├── humanoid_mimic.py │ ├── humanoid_mimic_train.py │ ├── losses.py │ ├── pd_controller.py │ └── system_configs │ │ ├── HUMANOID.py │ │ ├── SMPL.py │ │ ├── SWORDSHIELD.py │ │ └── __init__.py └── utils │ ├── __init__.py │ ├── io.py │ └── rotation6d.py ├── mimic.py ├── requirements.txt └── visualize.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/LICENSE.md -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/README.md -------------------------------------------------------------------------------- /asset/teaser.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/asset/teaser.gif -------------------------------------------------------------------------------- /configs/AIST/aist_example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/configs/AIST/aist_example.yaml -------------------------------------------------------------------------------- /configs/AIST/aist_example2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/configs/AIST/aist_example2.yaml -------------------------------------------------------------------------------- /configs/AMASS/amass_example.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/configs/AMASS/amass_example.yaml -------------------------------------------------------------------------------- /configs/AMASS/amass_example2.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/configs/AMASS/amass_example2.yaml -------------------------------------------------------------------------------- /configs/AMASS/amass_example3.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/configs/AMASS/amass_example3.yaml -------------------------------------------------------------------------------- /configs/AMP/backflip.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/configs/AMP/backflip.yaml -------------------------------------------------------------------------------- /configs/AMP/backflip_20s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/configs/AMP/backflip_20s.yaml -------------------------------------------------------------------------------- /configs/AMP/backflip_20s_vanilla.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/configs/AMP/backflip_20s_vanilla.yaml -------------------------------------------------------------------------------- /configs/AMP/cartwheel_20s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/configs/AMP/cartwheel_20s.yaml -------------------------------------------------------------------------------- /configs/AMP/crawl_20s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/configs/AMP/crawl_20s.yaml -------------------------------------------------------------------------------- /configs/AMP/dance_20s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/configs/AMP/dance_20s.yaml -------------------------------------------------------------------------------- /configs/AMP/jog_20s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/configs/AMP/jog_20s.yaml -------------------------------------------------------------------------------- /configs/AMP/jump_20s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/configs/AMP/jump_20s.yaml -------------------------------------------------------------------------------- /configs/AMP/roll.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/configs/AMP/roll.yaml -------------------------------------------------------------------------------- /configs/AMP/roll_20s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/configs/AMP/roll_20s.yaml -------------------------------------------------------------------------------- /configs/AMP/run_20s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/configs/AMP/run_20s.yaml -------------------------------------------------------------------------------- /configs/AMP/sideflip_20s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/configs/AMP/sideflip_20s.yaml -------------------------------------------------------------------------------- /configs/AMP/spinkick_20s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/configs/AMP/spinkick_20s.yaml -------------------------------------------------------------------------------- /configs/AMP/walk_20s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/configs/AMP/walk_20s.yaml -------------------------------------------------------------------------------- /configs/AMP/zombie_20s.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/configs/AMP/zombie_20s.yaml -------------------------------------------------------------------------------- /configs/SWORDSHIELD/RL_Avatar_Atk_2xCombo01_Motion.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/configs/SWORDSHIELD/RL_Avatar_Atk_2xCombo01_Motion.yaml -------------------------------------------------------------------------------- /data/demo_aist/gBR_sBM_cAll_d04_mBR0_ch01.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/demo_aist/gBR_sBM_cAll_d04_mBR0_ch01.npy -------------------------------------------------------------------------------- /data/demo_aist/gBR_sFM_cAll_d04_mBR1_ch02.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/demo_aist/gBR_sFM_cAll_d04_mBR1_ch02.npy -------------------------------------------------------------------------------- /data/demo_amass/75_09_stageii.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/demo_amass/75_09_stageii.npy -------------------------------------------------------------------------------- /data/demo_amass/KickHuefthoch05_stageii.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/demo_amass/KickHuefthoch05_stageii.npy -------------------------------------------------------------------------------- /data/demo_amass/WalkingStraightBackwards07_stageii.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/demo_amass/WalkingStraightBackwards07_stageii.npy -------------------------------------------------------------------------------- /data/demo_humanoid/backflip_fps_30_20s.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/demo_humanoid/backflip_fps_30_20s.npy -------------------------------------------------------------------------------- /data/demo_humanoid/cartwheel_fps_30_20s.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/demo_humanoid/cartwheel_fps_30_20s.npy -------------------------------------------------------------------------------- /data/demo_humanoid/crawl_fps_30_20s.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/demo_humanoid/crawl_fps_30_20s.npy -------------------------------------------------------------------------------- /data/demo_humanoid/dance_a_fps_30_20s.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/demo_humanoid/dance_a_fps_30_20s.npy -------------------------------------------------------------------------------- /data/demo_humanoid/jog_fps_30_20s.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/demo_humanoid/jog_fps_30_20s.npy -------------------------------------------------------------------------------- /data/demo_humanoid/jump_fps_30_20s.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/demo_humanoid/jump_fps_30_20s.npy -------------------------------------------------------------------------------- /data/demo_humanoid/roll_fps_30_20s.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/demo_humanoid/roll_fps_30_20s.npy -------------------------------------------------------------------------------- /data/demo_humanoid/run_fps_30_20s.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/demo_humanoid/run_fps_30_20s.npy -------------------------------------------------------------------------------- /data/demo_humanoid/sideflip_fps_30_20s.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/demo_humanoid/sideflip_fps_30_20s.npy -------------------------------------------------------------------------------- /data/demo_humanoid/spinkick_fps_30_20s.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/demo_humanoid/spinkick_fps_30_20s.npy -------------------------------------------------------------------------------- /data/demo_humanoid/walk_fps_30_20s.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/demo_humanoid/walk_fps_30_20s.npy -------------------------------------------------------------------------------- /data/demo_humanoid/zombie_fps_30_20s.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/demo_humanoid/zombie_fps_30_20s.npy -------------------------------------------------------------------------------- /data/demo_swordshield/RL_Avatar_Atk_2xCombo01_Motion.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/demo_swordshield/RL_Avatar_Atk_2xCombo01_Motion.npy -------------------------------------------------------------------------------- /data/tools/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/tools/README.md -------------------------------------------------------------------------------- /data/tools/aist_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/tools/aist_converter.py -------------------------------------------------------------------------------- /data/tools/amass_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/tools/amass_converter.py -------------------------------------------------------------------------------- /data/tools/amp_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/tools/amp_converter.py -------------------------------------------------------------------------------- /data/tools/joint_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/tools/joint_utils.py -------------------------------------------------------------------------------- /data/tools/rotation_utils/conversions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/tools/rotation_utils/conversions.py -------------------------------------------------------------------------------- /data/tools/rotation_utils/quaternion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/tools/rotation_utils/quaternion.py -------------------------------------------------------------------------------- /data/tools/sword_converter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/tools/sword_converter.py -------------------------------------------------------------------------------- /data/tools/to_smpl.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/data/tools/to_smpl.py -------------------------------------------------------------------------------- /diffmimic/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /diffmimic/brax_lib/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /diffmimic/brax_lib/acting.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/diffmimic/brax_lib/acting.py -------------------------------------------------------------------------------- /diffmimic/brax_lib/agent_diffmimic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/diffmimic/brax_lib/agent_diffmimic.py -------------------------------------------------------------------------------- /diffmimic/mimic_envs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/diffmimic/mimic_envs/__init__.py -------------------------------------------------------------------------------- /diffmimic/mimic_envs/humanoid_mimic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/diffmimic/mimic_envs/humanoid_mimic.py -------------------------------------------------------------------------------- /diffmimic/mimic_envs/humanoid_mimic_train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/diffmimic/mimic_envs/humanoid_mimic_train.py -------------------------------------------------------------------------------- /diffmimic/mimic_envs/losses.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/diffmimic/mimic_envs/losses.py -------------------------------------------------------------------------------- /diffmimic/mimic_envs/pd_controller.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/diffmimic/mimic_envs/pd_controller.py -------------------------------------------------------------------------------- /diffmimic/mimic_envs/system_configs/HUMANOID.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/diffmimic/mimic_envs/system_configs/HUMANOID.py -------------------------------------------------------------------------------- /diffmimic/mimic_envs/system_configs/SMPL.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/diffmimic/mimic_envs/system_configs/SMPL.py -------------------------------------------------------------------------------- /diffmimic/mimic_envs/system_configs/SWORDSHIELD.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/diffmimic/mimic_envs/system_configs/SWORDSHIELD.py -------------------------------------------------------------------------------- /diffmimic/mimic_envs/system_configs/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/diffmimic/mimic_envs/system_configs/__init__.py -------------------------------------------------------------------------------- /diffmimic/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/diffmimic/utils/__init__.py -------------------------------------------------------------------------------- /diffmimic/utils/io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/diffmimic/utils/io.py -------------------------------------------------------------------------------- /diffmimic/utils/rotation6d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/diffmimic/utils/rotation6d.py -------------------------------------------------------------------------------- /mimic.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/mimic.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/requirements.txt -------------------------------------------------------------------------------- /visualize.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/jiawei-ren/diffmimic/HEAD/visualize.py --------------------------------------------------------------------------------