API для обучения роботов с подкреплением в стиле Isaac Lab, работающий на MuJoCo Warp с GPU-ускорением.
robfiras/loco-mujoco
Бенчмарк имитационного обучения для управления всем телом с окружениями MuJoCo и MJX для гуманоидов, четвероногих роботов и биомеханических моделей человека.
- Звёзды
- 1,472
- Форки
- 152
- Открытые issues
- 43
- Последний коммит
- позавчера
Что это?
LocoMuJoCo — бенчмарк имитационного обучения для управления всем телом. Он включает набор окружений для передвижения гуманоидов, четвероногих роботов и биомеханических моделей человека, каждое из которых связано с наборами данных захвата движений, такими как AMASS, LAFAN1 и собственный датасет LocoMuJoCo.
Окружения работают на MuJoCo для симуляции отдельных экземпляров либо на MJX и MJWarp для параллельной симуляции на совместимых GPU. В проект входят реализации PPO, GAIL, AMP и DeepMimic на JAX, каждая в одном файле; также поддерживаются пользовательские функции вознаграждения для обучения с подкреплением. Кроме того, проект предоставляет интерфейс Gymnasium.
Помимо обучения, проект предлагает ретаргетинг движений между роботами, метрики сравнения траекторий, такие как динамическое выравнивание по времени и дискретное расстояние Фреше, а также встроенную рандомизацию домена и ландшафта. Типы наблюдений, функции вознаграждения, обработчики завершения эпизодов и компоненты рандомизации можно заменять и повторно использовать.
Ключевые возможности
- 01Окружения для гуманоидов, четвероногих роботов и биомеханических моделей человека на базе MuJoCo
- 02Параллельная симуляция с бэкендами MJX и MJWarp
- 03Реализации PPO, GAIL, AMP и DeepMimic на JAX, каждая в одном файле
- 04Ретаргетинг движений между роботами для существующих наборов данных
- 05Встроенная рандомизация домена и ландшафта
- 06Интерфейс Gymnasium
Сценарии использования
- Сравнение алгоритмов имитационного обучения на задачах передвижения
- Обучение политик ходьбы гуманоидов на данных захвата движений
- Обучение с подкреплением и пользовательскими функциями вознаграждения на моделях шагающих роботов
- Сравнение траекторий движений с помощью метрик DTW или расстояния Фреше
Быстрый старт
- 1
Клонируйте репозиторий
- 2
Установите пакет в редактируемом режиме из папки репозитория
pip install -e . - 3
При необходимости установите JAX с поддержкой GPU
pip install jax["cuda12"]
Что важно знать
- Согласно README, необходимо клонировать репозиторий, а не устанавливать пакет напрямую из PyPI.
- Для использования GPU с JAX требуется отдельно установить CUDA.
- Для работы среды MyoSkeleton нужно принять лицензию и отдельно скачать модель.
- Из-за условий лицензирования наборы данных AMASS необходимо отдельно скачать и установить.
В подборках
Темы
Частые вопросы
- Что такое loco-mujoco?
- Бенчмарк имитационного обучения для управления всем телом с окружениями MuJoCo и MJX для гуманоидов, четвероногих роботов и биомеханических моделей человека. LocoMuJoCo — бенчмарк имитационного обучения для управления всем телом. Он включает набор окружений для передвижения гуманоидов, четвероногих роботов и биомеханических моделей человека, каждое из которых связано с наборами данных захвата движений, такими как AMASS, LAFAN1 и собственный датасет LocoMuJoCo.
- loco-mujoco можно использовать бесплатно? Какая у него лицензия?
- loco-mujoco распространяется с открытым исходным кодом под лицензией MIT. Перед использованием в коммерческих продуктах проверьте условия лицензии.
- loco-mujoco всё ещё поддерживается?
- Активно поддерживается. Последний коммит в loco-mujoco — 5 нояб. 2026 г..
- Для чего можно использовать loco-mujoco?
- Сравнение алгоритмов имитационного обучения на задачах передвижения Обучение политик ходьбы гуманоидов на данных захвата движений Обучение с подкреплением и пользовательскими функциями вознаграждения на моделях шагающих роботов Сравнение траекторий движений с помощью метрик DTW или расстояния Фреше
- Как установить loco-mujoco?
- Клонируйте репозиторий Установите пакет в редактируемом режиме из папки репозитория: pip install -e . При необходимости установите JAX с поддержкой GPU: pip install jax["cuda12"]
- Где скачать loco-mujoco?
- Исходный код и инструкции по установке — на GitHub: https://github.com/robfiras/loco-mujoco
Описание создано ИИ на основе README проекта. Статистика GitHub обновляется ежедневно.
Похожие проекты
Открытые VLA-модели для робототехники от Physical Intelligence: предобученные чекпойнты и примеры дообучения.
Библиотека PyTorch с моделями, датасетами и инструментами для робототехники в реальных условиях: управление оборудованием, сбор данных, обучение и оценка политик.
Среды обучения с подкреплением для робота-бипеда Microduck высотой около 25 см: в симуляции MuJoCo он учится ходить, вставать и выполнять трюки.
Набор робототехнических сред для обучения с подкреплением на базе MuJoCo, реализованных через API Gymnasium.
Плагин для Unreal Engine 5, который запускает физику MuJoCo в редакторе и игре, сочетая точную динамику контактов с рендерингом Unreal.