如何映射开放AI健身房游戏中的自定义键？

Question

我试图让用户使用自定义键来播放赛车-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

所以我想知道如何做这个自定义的动作映射？play代码中的注释说它是一个映射keys_to_action: dict: tuple( int ) -> 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游戏中的一个自定义组合。游戏成功了，但关键的映射是原始的，而不是我的修改。

你们知道如何正确地进行这个自定义键映射吗？

Answer 1

下面是一个游戏游戏的例子。我还在评论中包括了其他健身房运动员的比赛方式。

import pygame
import gym
from gym.utils.play import play        
import numpy as np 
import warnings
warnings.filterwarnings('ignore')

# mapping = {(pygame.K_LEFT,): 0, (pygame.K_RIGHT,): 1}
# play(gym.make("CartPole-v0"), keys_to_action=mapping)

# mapping = {(pygame.K_LEFT,): 0, (pygame.K_RIGHT,): 2}
# play(gym.make("MountainCar-v0"), keys_to_action=mapping, noop=0)

mapping = {"w": np.array([0, 0.7, 0]),
            "a": np.array([-1, 0, 0]),
            "s": np.array([0, 0, 1]),
            "d": np.array([1, 0, 0]),
            "wa": np.array([-1, 0.7, 0]),
            "dw": np.array([1, 0.7, 0]),
            "ds": np.array([1, 0, 1]),
            "as": np.array([-1, 0, 1]),
            }
default_action = np.array([0,0,0])
play(gym.make("CarRacing-v2"), keys_to_action=mapping, noop=default_action)