python - 如何在 Open AI 健身游戏中映射自定义按键？

Question

我试图让 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

所以我想知道如何做这个自定义 Action 映射？播放代码中的注释说这是一个映射 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

这是一个来自 pygame 的例子。我还在评论中说明了如何玩其他健身房环境。

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)