├── .gitignore ├── .vscode ├── .gitignore ├── extensions.json ├── tasks.json └── tools │ ├── launch.template.json │ ├── settings.template.json │ └── setup_vscode.py ├── README.md ├── assets ├── __init__.py ├── motions │ ├── amp_humanoid_jog.npy │ ├── amp_humanoid_run.npy │ └── amp_humanoid_walk.npy ├── pretrained │ ├── jog.pth │ ├── run.pth │ └── walk.pth └── robots │ ├── __init__.py │ ├── amp_humanoid2-instanceable.usd │ ├── amp_humanoid2.py │ ├── amp_humanoid2.usda │ └── mjcf │ ├── amp_humanoid.xml │ └── amp_humanoid2.xml ├── media └── demo.gif ├── scripts ├── isaacgym_to_sim_convert.py ├── motion_viewer.py ├── play.py ├── rl_games_wrapper.py └── train.py └── source └── amp_rlg ├── __init__.py ├── learning ├── __init__.py ├── amp │ ├── amp_agent.py │ ├── amp_models.py │ ├── amp_network_builder.py │ └── amp_players.py ├── amp_datasets.py ├── ppo │ ├── ppo_agent.py │ ├── ppo_models.py │ ├── ppo_network_builder.py │ └── ppo_players.py └── replay_buffer.py ├── poselib ├── README.md └── poselib │ ├── __init__.py │ ├── core │ ├── __init__.py │ ├── backend │ │ ├── __init__.py │ │ ├── abstract.py │ │ └── logger.py │ ├── my_rotation3d.py │ └── tensor_utils.py │ └── skeleton │ ├── __init__.py │ ├── backend │ ├── __init__.py │ └── fbx │ │ ├── __init__.py │ │ ├── fbx_backend.py │ │ └── fbx_read_wrapper.py │ └── my_skeleton3d.py ├── tasks ├── __init__.py ├── amp │ ├── __init__.py │ ├── agents │ │ ├── __init__.py │ │ └── rl_games_amp_ppo_cfg.yaml │ ├── amp_env.py │ └── amp_env_cfg.py └── utils │ ├── __init__.py │ ├── hydra.py │ ├── importer.py │ └── parse_cfg.py └── utils ├── molib.py └── rotations.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/.gitignore -------------------------------------------------------------------------------- /.vscode/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/.vscode/.gitignore -------------------------------------------------------------------------------- /.vscode/extensions.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/.vscode/extensions.json -------------------------------------------------------------------------------- /.vscode/tasks.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/.vscode/tasks.json -------------------------------------------------------------------------------- /.vscode/tools/launch.template.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/.vscode/tools/launch.template.json -------------------------------------------------------------------------------- /.vscode/tools/settings.template.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/.vscode/tools/settings.template.json -------------------------------------------------------------------------------- /.vscode/tools/setup_vscode.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/.vscode/tools/setup_vscode.py -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/README.md -------------------------------------------------------------------------------- /assets/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /assets/motions/amp_humanoid_jog.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/assets/motions/amp_humanoid_jog.npy -------------------------------------------------------------------------------- /assets/motions/amp_humanoid_run.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/assets/motions/amp_humanoid_run.npy -------------------------------------------------------------------------------- /assets/motions/amp_humanoid_walk.npy: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/assets/motions/amp_humanoid_walk.npy -------------------------------------------------------------------------------- /assets/pretrained/jog.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/assets/pretrained/jog.pth -------------------------------------------------------------------------------- /assets/pretrained/run.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/assets/pretrained/run.pth -------------------------------------------------------------------------------- /assets/pretrained/walk.pth: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/assets/pretrained/walk.pth -------------------------------------------------------------------------------- /assets/robots/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /assets/robots/amp_humanoid2-instanceable.usd: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/assets/robots/amp_humanoid2-instanceable.usd -------------------------------------------------------------------------------- /assets/robots/amp_humanoid2.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/assets/robots/amp_humanoid2.py -------------------------------------------------------------------------------- /assets/robots/amp_humanoid2.usda: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/assets/robots/amp_humanoid2.usda -------------------------------------------------------------------------------- /assets/robots/mjcf/amp_humanoid.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/assets/robots/mjcf/amp_humanoid.xml -------------------------------------------------------------------------------- /assets/robots/mjcf/amp_humanoid2.xml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/assets/robots/mjcf/amp_humanoid2.xml -------------------------------------------------------------------------------- /media/demo.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/media/demo.gif -------------------------------------------------------------------------------- /scripts/isaacgym_to_sim_convert.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/scripts/isaacgym_to_sim_convert.py -------------------------------------------------------------------------------- /scripts/motion_viewer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/scripts/motion_viewer.py -------------------------------------------------------------------------------- /scripts/play.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/scripts/play.py -------------------------------------------------------------------------------- /scripts/rl_games_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/scripts/rl_games_wrapper.py -------------------------------------------------------------------------------- /scripts/train.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/scripts/train.py -------------------------------------------------------------------------------- /source/amp_rlg/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/__init__.py -------------------------------------------------------------------------------- /source/amp_rlg/learning/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /source/amp_rlg/learning/amp/amp_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/learning/amp/amp_agent.py -------------------------------------------------------------------------------- /source/amp_rlg/learning/amp/amp_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/learning/amp/amp_models.py -------------------------------------------------------------------------------- /source/amp_rlg/learning/amp/amp_network_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/learning/amp/amp_network_builder.py -------------------------------------------------------------------------------- /source/amp_rlg/learning/amp/amp_players.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/learning/amp/amp_players.py -------------------------------------------------------------------------------- /source/amp_rlg/learning/amp_datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/learning/amp_datasets.py -------------------------------------------------------------------------------- /source/amp_rlg/learning/ppo/ppo_agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/learning/ppo/ppo_agent.py -------------------------------------------------------------------------------- /source/amp_rlg/learning/ppo/ppo_models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/learning/ppo/ppo_models.py -------------------------------------------------------------------------------- /source/amp_rlg/learning/ppo/ppo_network_builder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/learning/ppo/ppo_network_builder.py -------------------------------------------------------------------------------- /source/amp_rlg/learning/ppo/ppo_players.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/learning/ppo/ppo_players.py -------------------------------------------------------------------------------- /source/amp_rlg/learning/replay_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/learning/replay_buffer.py -------------------------------------------------------------------------------- /source/amp_rlg/poselib/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/poselib/README.md -------------------------------------------------------------------------------- /source/amp_rlg/poselib/poselib/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/poselib/poselib/__init__.py -------------------------------------------------------------------------------- /source/amp_rlg/poselib/poselib/core/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/poselib/poselib/core/__init__.py -------------------------------------------------------------------------------- /source/amp_rlg/poselib/poselib/core/backend/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/poselib/poselib/core/backend/__init__.py -------------------------------------------------------------------------------- /source/amp_rlg/poselib/poselib/core/backend/abstract.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/poselib/poselib/core/backend/abstract.py -------------------------------------------------------------------------------- /source/amp_rlg/poselib/poselib/core/backend/logger.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/poselib/poselib/core/backend/logger.py -------------------------------------------------------------------------------- /source/amp_rlg/poselib/poselib/core/my_rotation3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/poselib/poselib/core/my_rotation3d.py -------------------------------------------------------------------------------- /source/amp_rlg/poselib/poselib/core/tensor_utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/poselib/poselib/core/tensor_utils.py -------------------------------------------------------------------------------- /source/amp_rlg/poselib/poselib/skeleton/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/poselib/poselib/skeleton/__init__.py -------------------------------------------------------------------------------- /source/amp_rlg/poselib/poselib/skeleton/backend/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /source/amp_rlg/poselib/poselib/skeleton/backend/fbx/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/poselib/poselib/skeleton/backend/fbx/__init__.py -------------------------------------------------------------------------------- /source/amp_rlg/poselib/poselib/skeleton/backend/fbx/fbx_backend.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/poselib/poselib/skeleton/backend/fbx/fbx_backend.py -------------------------------------------------------------------------------- /source/amp_rlg/poselib/poselib/skeleton/backend/fbx/fbx_read_wrapper.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/poselib/poselib/skeleton/backend/fbx/fbx_read_wrapper.py -------------------------------------------------------------------------------- /source/amp_rlg/poselib/poselib/skeleton/my_skeleton3d.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/poselib/poselib/skeleton/my_skeleton3d.py -------------------------------------------------------------------------------- /source/amp_rlg/tasks/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/tasks/__init__.py -------------------------------------------------------------------------------- /source/amp_rlg/tasks/amp/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/tasks/amp/__init__.py -------------------------------------------------------------------------------- /source/amp_rlg/tasks/amp/agents/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /source/amp_rlg/tasks/amp/agents/rl_games_amp_ppo_cfg.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/tasks/amp/agents/rl_games_amp_ppo_cfg.yaml -------------------------------------------------------------------------------- /source/amp_rlg/tasks/amp/amp_env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/tasks/amp/amp_env.py -------------------------------------------------------------------------------- /source/amp_rlg/tasks/amp/amp_env_cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/tasks/amp/amp_env_cfg.py -------------------------------------------------------------------------------- /source/amp_rlg/tasks/utils/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/tasks/utils/__init__.py -------------------------------------------------------------------------------- /source/amp_rlg/tasks/utils/hydra.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/tasks/utils/hydra.py -------------------------------------------------------------------------------- /source/amp_rlg/tasks/utils/importer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/tasks/utils/importer.py -------------------------------------------------------------------------------- /source/amp_rlg/tasks/utils/parse_cfg.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/tasks/utils/parse_cfg.py -------------------------------------------------------------------------------- /source/amp_rlg/utils/molib.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/utils/molib.py -------------------------------------------------------------------------------- /source/amp_rlg/utils/rotations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Gudegi/IsaacLab_AMP_rl-games/HEAD/source/amp_rlg/utils/rotations.py --------------------------------------------------------------------------------