如何在Open AI健身房游戏中映射自定义键?

时间:2018-04-09 02:04:13

标签: python openai-gym

我试图让用户使用自定义键来启用CarRacing-v0环境我认为我可以使用utils.play这样做:

import gym
from gym.utils.play import *

play(gym.make("CarRacing-v0"))

它适用于atari环境但在这种情况下我得到了

---------------------------------------------------------------------------
AssertionError                            Traceback (most recent call last)
<ipython-input-3-080385c697d2> in <module>()
      8 # play.keys_to_action = KEYWORD_TO_KEY
      9 
---> 10 play(gym.make("CarRacing-v0"))
     11 
     12 # mapping of keys

~/Documents/openai/gym/gym/utils/play.py in play(env, transpose, fps, zoom, callback, keys_to_action)
     92         else:
     93             assert False, env.spec.id + " does not have explicit key to action mapping, " + \
---> 94                           "please specify one manually"
     95     relevant_keys = set(sum(map(list, keys_to_action.keys()),[]))
     96 

AssertionError: CarRacing-v0 does not have explicit key to action mapping, please specify one manually

所以我想知道如何进行自定义动作映射? 播放代码中的注释表示它是映射keys_to_action:dict:tuple(int) - &gt; INT

atari_env这样做:

KEYWORD_TO_KEY = {
            'UP':      ord('w'),
            'DOWN':    ord('s'),
            'LEFT':    ord('a'),
            'RIGHT':   ord('d'),
            'FIRE':    ord(' '),
        }

我知道car_racing脚本通过捕获按下3元素数组的键并将该值传递给env.step来完成此操作。所以我在这里尝试了类似的方式:

KEYWORD_TO_KEY = {'STEER':ord('a'),'GAS':ord('w'),'BREAK':ord('s')}
play.keys_to_action = KEYWORD_TO_KEY

它不起作用。我知道转向是错的,但我以为我至少会让车辆转向一个方向。 然后我检查了将关键字重新映射到其中一个Atari游戏的自定义组合。 游戏有效但键映射是原始的而不是我的修改。

你们有没有想过如何正确地进行这种自定义键映射?

0 个答案:

没有答案