mantle2048 / gym_custom_terrain

custom gym env

gym_custom_terrain

This package will help you customize the terrain in MuJoCo. Terrain information can be specified as grayscale images.

How to install

pip install git+https://github.com/wataru0/gym_custom_terrain.git

How to use

from gym_custom_terrain import custom_make


env = custom_make("CustomTerrainAnt-v0", "images/terrain.png")

env.reset()
done = False

while True:
    env.render()
    if done:
        break
    action = env.action_space.sample()
    _, _, done, _ = env.step(action)

Demo

Input Image

Simulation Environment