├── .flake8 ├── .gitignore ├── .pre-commit-config.yaml ├── .pylintrc ├── README.md ├── assets ├── overview.gif ├── scaling-devices.png └── scaling-envs.png ├── benchmark ├── __init__.py ├── benchmark.py ├── benchmark_soil_mechanics_debug.py └── plot_benchmark.py ├── config └── env_generation_config.yaml ├── data └── custom │ ├── dumpability │ ├── custom_1.png │ ├── custom_10.png │ ├── custom_11.png │ ├── custom_12.png │ ├── custom_2.png │ ├── custom_3.png │ ├── custom_4.png │ ├── custom_5.png │ ├── custom_6.png │ ├── custom_7.png │ ├── custom_8.png │ └── custom_9.png │ ├── images │ ├── custom_1.png │ ├── custom_10.png │ ├── custom_11.png │ ├── custom_12.png │ ├── custom_2.png │ ├── custom_3.png │ ├── custom_4.png │ ├── custom_5.png │ ├── custom_6.png │ ├── custom_7.png │ ├── custom_8.png │ └── custom_9.png │ └── occupancy │ ├── custom_1.png │ ├── custom_10.png │ ├── custom_11.png │ ├── custom_12.png │ ├── custom_2.png │ ├── custom_3.png │ ├── custom_4.png │ ├── custom_5.png │ ├── custom_6.png │ ├── custom_7.png │ ├── custom_8.png │ └── custom_9.png ├── environment.yml ├── pyproject.toml ├── requirements-dev.txt ├── requirements.txt ├── setup.py ├── terra ├── __init__.py ├── actions.py ├── agent.py ├── config.py ├── curriculum.py ├── env.py ├── env_generation │ ├── README.md │ ├── __init__.py │ ├── convert_to_terra.py │ ├── create_train_data.py │ ├── generate_dataset.py │ ├── generate_foundations.py │ ├── generate_relocations.py │ ├── openstreet.py │ ├── procedural_data.py │ ├── terrain_generation.py │ ├── utils.py │ └── visualize_maps.py ├── map.py ├── map_generator.py ├── map_utils │ ├── jax_terrain_generation.py │ ├── openstreet_dataset_generator.py │ ├── openstreet_plugin.py │ ├── terrain_generation.py │ ├── test_openstreet_generator.py │ └── test_openstreet_plugin.py ├── maps_buffer.py ├── settings.py ├── state.py ├── utils.py ├── viz │ ├── __init__.py │ ├── game │ │ ├── agent.py │ │ ├── game.py │ │ ├── settings.py │ │ ├── utils.py │ │ └── world.py │ ├── main_manual.py │ └── play.py └── wrappers.py └── tests ├── __init__.py └── unit └── __init__.py /.flake8: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/.flake8 -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.pylintrc: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/.pylintrc -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/README.md -------------------------------------------------------------------------------- /assets/overview.gif: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/assets/overview.gif -------------------------------------------------------------------------------- /assets/scaling-devices.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/assets/scaling-devices.png -------------------------------------------------------------------------------- /assets/scaling-envs.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/assets/scaling-envs.png -------------------------------------------------------------------------------- /benchmark/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /benchmark/benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/benchmark/benchmark.py -------------------------------------------------------------------------------- /benchmark/benchmark_soil_mechanics_debug.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/benchmark/benchmark_soil_mechanics_debug.py -------------------------------------------------------------------------------- /benchmark/plot_benchmark.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/benchmark/plot_benchmark.py -------------------------------------------------------------------------------- /config/env_generation_config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/config/env_generation_config.yaml -------------------------------------------------------------------------------- /data/custom/dumpability/custom_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/dumpability/custom_1.png -------------------------------------------------------------------------------- /data/custom/dumpability/custom_10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/dumpability/custom_10.png -------------------------------------------------------------------------------- /data/custom/dumpability/custom_11.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/dumpability/custom_11.png -------------------------------------------------------------------------------- /data/custom/dumpability/custom_12.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/dumpability/custom_12.png -------------------------------------------------------------------------------- /data/custom/dumpability/custom_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/dumpability/custom_2.png -------------------------------------------------------------------------------- /data/custom/dumpability/custom_3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/dumpability/custom_3.png -------------------------------------------------------------------------------- /data/custom/dumpability/custom_4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/dumpability/custom_4.png -------------------------------------------------------------------------------- /data/custom/dumpability/custom_5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/dumpability/custom_5.png -------------------------------------------------------------------------------- /data/custom/dumpability/custom_6.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/dumpability/custom_6.png -------------------------------------------------------------------------------- /data/custom/dumpability/custom_7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/dumpability/custom_7.png -------------------------------------------------------------------------------- /data/custom/dumpability/custom_8.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/dumpability/custom_8.png -------------------------------------------------------------------------------- /data/custom/dumpability/custom_9.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/dumpability/custom_9.png -------------------------------------------------------------------------------- /data/custom/images/custom_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/images/custom_1.png -------------------------------------------------------------------------------- /data/custom/images/custom_10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/images/custom_10.png -------------------------------------------------------------------------------- /data/custom/images/custom_11.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/images/custom_11.png -------------------------------------------------------------------------------- /data/custom/images/custom_12.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/images/custom_12.png -------------------------------------------------------------------------------- /data/custom/images/custom_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/images/custom_2.png -------------------------------------------------------------------------------- /data/custom/images/custom_3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/images/custom_3.png -------------------------------------------------------------------------------- /data/custom/images/custom_4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/images/custom_4.png -------------------------------------------------------------------------------- /data/custom/images/custom_5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/images/custom_5.png -------------------------------------------------------------------------------- /data/custom/images/custom_6.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/images/custom_6.png -------------------------------------------------------------------------------- /data/custom/images/custom_7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/images/custom_7.png -------------------------------------------------------------------------------- /data/custom/images/custom_8.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/images/custom_8.png -------------------------------------------------------------------------------- /data/custom/images/custom_9.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/images/custom_9.png -------------------------------------------------------------------------------- /data/custom/occupancy/custom_1.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/occupancy/custom_1.png -------------------------------------------------------------------------------- /data/custom/occupancy/custom_10.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/occupancy/custom_10.png -------------------------------------------------------------------------------- /data/custom/occupancy/custom_11.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/occupancy/custom_11.png -------------------------------------------------------------------------------- /data/custom/occupancy/custom_12.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/occupancy/custom_12.png -------------------------------------------------------------------------------- /data/custom/occupancy/custom_2.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/occupancy/custom_2.png -------------------------------------------------------------------------------- /data/custom/occupancy/custom_3.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/occupancy/custom_3.png -------------------------------------------------------------------------------- /data/custom/occupancy/custom_4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/occupancy/custom_4.png -------------------------------------------------------------------------------- /data/custom/occupancy/custom_5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/occupancy/custom_5.png -------------------------------------------------------------------------------- /data/custom/occupancy/custom_6.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/occupancy/custom_6.png -------------------------------------------------------------------------------- /data/custom/occupancy/custom_7.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/occupancy/custom_7.png -------------------------------------------------------------------------------- /data/custom/occupancy/custom_8.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/occupancy/custom_8.png -------------------------------------------------------------------------------- /data/custom/occupancy/custom_9.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/data/custom/occupancy/custom_9.png -------------------------------------------------------------------------------- /environment.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/environment.yml -------------------------------------------------------------------------------- /pyproject.toml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/pyproject.toml -------------------------------------------------------------------------------- /requirements-dev.txt: -------------------------------------------------------------------------------- 1 | black 2 | -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/setup.py -------------------------------------------------------------------------------- /terra/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /terra/actions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/actions.py -------------------------------------------------------------------------------- /terra/agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/agent.py -------------------------------------------------------------------------------- /terra/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/config.py -------------------------------------------------------------------------------- /terra/curriculum.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/curriculum.py -------------------------------------------------------------------------------- /terra/env.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/env.py -------------------------------------------------------------------------------- /terra/env_generation/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/env_generation/README.md -------------------------------------------------------------------------------- /terra/env_generation/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /terra/env_generation/convert_to_terra.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/env_generation/convert_to_terra.py -------------------------------------------------------------------------------- /terra/env_generation/create_train_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/env_generation/create_train_data.py -------------------------------------------------------------------------------- /terra/env_generation/generate_dataset.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/env_generation/generate_dataset.py -------------------------------------------------------------------------------- /terra/env_generation/generate_foundations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/env_generation/generate_foundations.py -------------------------------------------------------------------------------- /terra/env_generation/generate_relocations.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/env_generation/generate_relocations.py -------------------------------------------------------------------------------- /terra/env_generation/openstreet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/env_generation/openstreet.py -------------------------------------------------------------------------------- /terra/env_generation/procedural_data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/env_generation/procedural_data.py -------------------------------------------------------------------------------- /terra/env_generation/terrain_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/env_generation/terrain_generation.py -------------------------------------------------------------------------------- /terra/env_generation/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/env_generation/utils.py -------------------------------------------------------------------------------- /terra/env_generation/visualize_maps.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/env_generation/visualize_maps.py -------------------------------------------------------------------------------- /terra/map.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/map.py -------------------------------------------------------------------------------- /terra/map_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/map_generator.py -------------------------------------------------------------------------------- /terra/map_utils/jax_terrain_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/map_utils/jax_terrain_generation.py -------------------------------------------------------------------------------- /terra/map_utils/openstreet_dataset_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/map_utils/openstreet_dataset_generator.py -------------------------------------------------------------------------------- /terra/map_utils/openstreet_plugin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/map_utils/openstreet_plugin.py -------------------------------------------------------------------------------- /terra/map_utils/terrain_generation.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/map_utils/terrain_generation.py -------------------------------------------------------------------------------- /terra/map_utils/test_openstreet_generator.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/map_utils/test_openstreet_generator.py -------------------------------------------------------------------------------- /terra/map_utils/test_openstreet_plugin.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/map_utils/test_openstreet_plugin.py -------------------------------------------------------------------------------- /terra/maps_buffer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/maps_buffer.py -------------------------------------------------------------------------------- /terra/settings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/settings.py -------------------------------------------------------------------------------- /terra/state.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/state.py -------------------------------------------------------------------------------- /terra/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/utils.py -------------------------------------------------------------------------------- /terra/viz/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /terra/viz/game/agent.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/viz/game/agent.py -------------------------------------------------------------------------------- /terra/viz/game/game.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/viz/game/game.py -------------------------------------------------------------------------------- /terra/viz/game/settings.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/viz/game/settings.py -------------------------------------------------------------------------------- /terra/viz/game/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/viz/game/utils.py -------------------------------------------------------------------------------- /terra/viz/game/world.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/viz/game/world.py -------------------------------------------------------------------------------- /terra/viz/main_manual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/viz/main_manual.py -------------------------------------------------------------------------------- /terra/viz/play.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/viz/play.py -------------------------------------------------------------------------------- /terra/wrappers.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/leggedrobotics/terra/HEAD/terra/wrappers.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/unit/__init__.py: -------------------------------------------------------------------------------- 1 | --------------------------------------------------------------------------------